@zzzen/pyright-internal 1.2.0-dev.20260913 → 1.2.0-dev.20260927

package/dist/analyzer/checker.d.ts

@@ -8,6 +8,7 @@ export declare class Checker extends ParseTreeWalker {
     private _importResolver;
     private _evaluator;
     private _dependentFiles;
+    private _walkOperation?;
     private readonly _moduleNode;
     private readonly _fileInfo;
     private readonly _nodeInfo;
@@ -15,7 +16,7 @@ export declare class Checker extends ParseTreeWalker {
     private _scopedNodes;
     private _typeParamLists;
     private _multipartImports;
-    constructor(_importResolver: ImportResolver, _evaluator: TypeEvaluator, parseResults: ParserOutput, _dependentFiles: ParserOutput[] | undefined, nodeInfoReader: AnalyzerNodeInfo.AnalyzerNodeInfoReader);
+    constructor(_importResolver: ImportResolver, _evaluator: TypeEvaluator, parseResults: ParserOutput, _dependentFiles: ParserOutput[] | undefined, nodeInfoReader: AnalyzerNodeInfo.AnalyzerNodeInfoReader, _walkOperation?: ((node: ParseNode, callback: () => void) => void) | undefined);
     check(): void;
     walk(node: ParseNode): void;
     visitSuite(node: SuiteNode): boolean;

package/dist/analyzer/checker.js

@@ -80,7 +80,6 @@ const staticExpressions_1 = require("./staticExpressions");
 const SymbolNameUtils = __importStar(require("./symbolNameUtils"));
 const symbolUtils_1 = require("./symbolUtils");
 const typedDicts_1 = require("./typedDicts");
-const typeEvaluator_1 = require("./typeEvaluator");
 const typeEvaluatorTypes_1 = require("./typeEvaluatorTypes");
 const typeGuards_1 = require("./typeGuards");
 const types_1 = require("./types");
@@ -89,7 +88,7 @@ const typeUtils_1 = require("./typeUtils");
 // functions to be emitted.
 const isPrintCodeComplexityEnabled = false;
 class Checker extends parseTreeWalker_1.ParseTreeWalker {
-    constructor(_importResolver, _evaluator, parseResults, _dependentFiles, nodeInfoReader) {
+    constructor(_importResolver, _evaluator, parseResults, _dependentFiles, nodeInfoReader, _walkOperation) {
         // Forward the reader to the base walker so the structural walk expands both tier-1
         // (parser-derived) and tier-2 (evaluator-discovered, e.g. `cast("Foo", v)`) string
         // annotations. The checker runs after type evaluation, so tier-2 annotations exist by
@@ -100,6 +99,7 @@ class Checker extends parseTreeWalker_1.ParseTreeWalker {
         this._importResolver = _importResolver;
         this._evaluator = _evaluator;
         this._dependentFiles = _dependentFiles;
+        this._walkOperation = _walkOperation;
         this._isUnboundCheckSuppressed = false;
         // A list of all nodes that are defined within the module that
         // have their own scopes.
@@ -119,7 +119,7 @@ class Checker extends parseTreeWalker_1.ParseTreeWalker {
         if (isPrintCodeComplexityEnabled) {
             console.log(`Code complexity of module ${this._fileInfo.fileUri.toUserVisibleString()} is ${codeComplexity.toString()}`);
         }
-        if (codeComplexity > typeEvaluator_1.maxCodeComplexity) {
+        if (codeComplexity > this._evaluator.getMaxCodeComplexity()) {
             this._evaluator.addDiagnosticForTextRange(this._fileInfo, diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.codeTooComplexToAnalyze(), { start: 0, length: 0 });
         }
         this._walkStatementsAndReportUnreachable(this._moduleNode.d.statements);
@@ -136,6 +136,19 @@ class Checker extends parseTreeWalker_1.ParseTreeWalker {
         this._reportDuplicateImports();
     }
     walk(node) {
+        if (this._walkOperation) {
+            this._walkOperation(node, () => {
+                if (!this._nodeInfo.isCodeUnreachable(node)) {
+                    super.walk(node);
+                }
+                else {
+                    this._evaluator.suppressDiagnostics(node, () => {
+                        super.walk(node);
+                    });
+                }
+            });
+            return;
+        }
         if (!this._nodeInfo.isCodeUnreachable(node)) {
             super.walk(node);
         }
@@ -406,7 +419,7 @@ class Checker extends parseTreeWalker_1.ParseTreeWalker {
             }
         });
         const codeComplexity = this._nodeInfo.getCodeFlowComplexity(node);
-        const isTooComplexToAnalyze = codeComplexity > typeEvaluator_1.maxCodeComplexity;
+        const isTooComplexToAnalyze = codeComplexity > this._evaluator.getMaxCodeComplexity();
         if (isPrintCodeComplexityEnabled) {
             console.log(`Code complexity of function ${node.d.name.d.value} is ${codeComplexity.toString()}`);
         }