@zzzen/pyright-internal 1.2.0-dev.20260802 → 1.2.0-dev.20260809

package/dist/analyzer/checker.js

@@ -3247,30 +3247,49 @@ class Checker extends parseTreeWalker_1.ParseTreeWalker {
             }
         });
     }
-    // Verifies the rules specified in PEP 589 about TypedDict classes.
-    // They cannot have statements other than type annotations, doc
-    // strings, and "pass" statements or ellipses.
+    // Verifies the rules specified for TypedDict class bodies.
+    // They cannot have statements other than type annotations, doc strings,
+    // "pass" statements, ellipses, and statically evaluable if statements.
     _validateTypedDictClassSuite(suiteNode) {
         const emitBadStatementError = (node) => {
             this._evaluator.addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.typedDictBadVar(), node);
         };
-        suiteNode.d.statements.forEach((statement) => {
-            if (!AnalyzerNodeInfo.isCodeUnreachable(statement)) {
-                if (statement.nodeType === 47 /* ParseNodeType.StatementList */) {
-                    for (const substatement of statement.d.statements) {
-                        if (substatement.nodeType !== 54 /* ParseNodeType.TypeAnnotation */ &&
-                            substatement.nodeType !== 21 /* ParseNodeType.Ellipsis */ &&
-                            substatement.nodeType !== 48 /* ParseNodeType.StringList */ &&
-                            substatement.nodeType !== 42 /* ParseNodeType.Pass */) {
-                            emitBadStatementError(substatement);
-                        }
+        function validateStatement(statement) {
+            if (AnalyzerNodeInfo.isCodeUnreachable(statement)) {
+                return;
+            }
+            if (statement.nodeType === 47 /* ParseNodeType.StatementList */) {
+                for (const substatement of statement.d.statements) {
+                    if (substatement.nodeType !== 54 /* ParseNodeType.TypeAnnotation */ &&
+                        substatement.nodeType !== 21 /* ParseNodeType.Ellipsis */ &&
+                        substatement.nodeType !== 48 /* ParseNodeType.StringList */ &&
+                        substatement.nodeType !== 42 /* ParseNodeType.Pass */) {
+                        emitBadStatementError(substatement);
                     }
                 }
-                else {
+                return;
+            }
+            if (statement.nodeType === 22 /* ParseNodeType.If */) {
+                const conditionValue = AnalyzerNodeInfo.getStaticConditionValue(statement);
+                if (conditionValue === undefined) {
                     emitBadStatementError(statement);
+                    return;
+                }
+                const reachableSuite = conditionValue ? statement.d.ifSuite : statement.d.elseSuite;
+                if (reachableSuite?.nodeType === 22 /* ParseNodeType.If */) {
+                    validateStatement(reachableSuite);
                 }
+                else if (reachableSuite) {
+                    validateSuite(reachableSuite);
+                }
+                return;
             }
-        });
+            emitBadStatementError(statement);
+        }
+        function validateSuite(suite) {
+            suite.d.statements.forEach((statement) => validateStatement(statement));
+        }
+        validateSuite(suiteNode);
     }
     _validateTypeGuardFunction(node, functionType, isMethod) {
         const returnType = functionType.shared.declaredReturnType;