Eliminated restriction that prevented the analysis of text files that don't end in ".py" or ".pyi".

Author: msfterictraut

packages/pyright-internal/src/analyzer/program.ts

@@ -2069,12 +2069,6 @@ export class Program {
     private _addToSourceFileListAndMap(fileInfo: SourceFileInfo) {
         const filePath = normalizePathCase(this._fs, fileInfo.sourceFile.getFilePath());
 
-        // All source files should be ".pyi" or ".py" files.
-        const fileExtension = getFileExtension(filePath).toLowerCase();
-        if (fileExtension !== '.py' && fileExtension !== '.pyi') {
-            fail(`${filePath} is not a source file`);
-        }
-
         // We should never add a file with the same path twice.
         assert(!this._sourceFileMap.has(filePath));
 

packages/pyright-internal/src/languageServerBase.ts

@@ -74,7 +74,7 @@ import {
     FileWatcherEventHandler,
     FileWatcherEventType,
 } from './common/fileSystem';
-import { containsPath, convertPathToUri, convertUriToPath, getFileExtension } from './common/pathUtils';
+import { containsPath, convertPathToUri, convertUriToPath } from './common/pathUtils';
 import { ProgressReporter, ProgressReportTracker } from './common/progressReporter';
 import { convertWorkspaceEdits } from './common/textEditUtils';
 import { DocumentRange, Position } from './common/textRange';
@@ -776,35 +776,25 @@ export abstract class LanguageServerBase implements LanguageServerInterface {
         this._connection.onDidOpenTextDocument(async (params) => {
             const filePath = convertUriToPath(params.textDocument.uri);
             const workspace = await this.getWorkspaceForFile(filePath);
-            if (this._isSupportedSourceFileType(filePath)) {
-                workspace.serviceInstance.setFileOpened(
-                    filePath,
-                    params.textDocument.version,
-                    params.textDocument.text
-                );
-            }
+            workspace.serviceInstance.setFileOpened(filePath, params.textDocument.version, params.textDocument.text);
         });
 
         this._connection.onDidChangeTextDocument(async (params) => {
             this.recordUserInteractionTime();
 
             const filePath = convertUriToPath(params.textDocument.uri);
             const workspace = await this.getWorkspaceForFile(filePath);
-            if (this._isSupportedSourceFileType(filePath)) {
-                workspace.serviceInstance.updateOpenFileContents(
-                    filePath,
-                    params.textDocument.version,
-                    params.contentChanges
-                );
-            }
+            workspace.serviceInstance.updateOpenFileContents(
+                filePath,
+                params.textDocument.version,
+                params.contentChanges
+            );
         });
 
         this._connection.onDidCloseTextDocument(async (params) => {
             const filePath = convertUriToPath(params.textDocument.uri);
             const workspace = await this.getWorkspaceForFile(filePath);
-            if (this._isSupportedSourceFileType(filePath)) {
-                workspace.serviceInstance.setFileClosed(filePath);
-            }
+            workspace.serviceInstance.setFileClosed(filePath);
         });
 
         this._connection.onDidChangeWatchedFiles((params) => {
@@ -1131,15 +1121,6 @@ export abstract class LanguageServerBase implements LanguageServerInterface {
         }
     }
 
-    private _isSupportedSourceFileType(filePath: string) {
-        // The language server assumes that all source files are either ".py"
-        // or ".pyi" files. We don't want to attempt to parse other files
-        // (like binaries, etc.). And we need to know whether to use normal
-        // or stub semantics.
-        const extension = getFileExtension(filePath).toLowerCase();
-        return extension === '.py' || extension === '.pyi';
-    }
-
     private _getCompatibleMarkupKind(clientSupportedFormats: MarkupKind[] | undefined) {
         const serverSupportedFormats = [MarkupKind.PlainText, MarkupKind.Markdown];