@zzzen/pyright-internal 1.2.0-dev.20230625 → 1.2.0-dev.20230702

package/dist/parser/parser.js

@@ -1182,7 +1182,7 @@ class Parser {
         return forNode;
     }
     // comp_iter: comp_for | comp_if
-    _tryParseListComprehension(target) {
+    _tryParseListComprehension(target, isGenerator) {
         const compFor = this._tryParseCompForStatement();
         if (!compFor) {
             return undefined;
@@ -1193,7 +1193,7 @@ class Parser {
         else if (target.nodeType === 16 /* DictionaryExpandEntry */) {
             this._addError(localize_1.Localizer.Diagnostic.dictExpandIllegalInComprehension(), target);
         }
-        const listCompNode = parseNodes_1.ListComprehensionNode.create(target);
+        const listCompNode = parseNodes_1.ListComprehensionNode.create(target, isGenerator);
         const forIfList = [compFor];
         while (true) {
             const compIter = this._tryParseCompForStatement() || this._tryParseCompIfStatement();
@@ -2965,7 +2965,7 @@ class Parser {
                 }
             }
             else {
-                const listComp = this._tryParseListComprehension(valueExpr);
+                const listComp = this._tryParseListComprehension(valueExpr, /* isGenerator */ true);
                 if (listComp) {
                     valueExpr = listComp;
                 }
@@ -3133,7 +3133,7 @@ class Parser {
             }
             return yieldExpr;
         }
-        const exprListResult = this._parseTestListWithComprehension();
+        const exprListResult = this._parseTestListWithComprehension(/* isGenerator */ true);
         const tupleOrExpression = this._makeExpressionOrTuple(exprListResult, /* enclosedInParens */ true);
         (0, parseNodes_1.extendRange)(tupleOrExpression, startParen);
         if (this._peekTokenType() !== 14 /* CloseParenthesis */) {
@@ -3150,7 +3150,7 @@ class Parser {
         var _a;
         const startBracket = this._getNextToken();
         (0, debug_1.assert)(startBracket.type === 15 /* OpenBracket */);
-        const exprListResult = this._parseTestListWithComprehension();
+        const exprListResult = this._parseTestListWithComprehension(/* isGenerator */ false);
         const closeBracket = this._peekToken();
         if (!this._consumeTokenIfType(16 /* CloseBracket */)) {
             return this._handleExpressionParseError(9 /* MissingListCloseBracket */, localize_1.Localizer.Diagnostic.expectedCloseBracket(), startBracket, (_a = exprListResult.parseError) !== null && _a !== void 0 ? _a : _createList());
@@ -3171,11 +3171,11 @@ class Parser {
             return listAtom;
         }
     }
-    _parseTestListWithComprehension() {
+    _parseTestListWithComprehension(isGenerator) {
         let sawComprehension = false;
         return this._parseExpressionListGeneric(() => {
             let expr = this._parseTestOrStarExpression(/* allowAssignmentExpression */ true);
-            const listComp = this._tryParseListComprehension(expr);
+            const listComp = this._tryParseListComprehension(expr, isGenerator);
             if (listComp) {
                 expr = listComp;
                 sawComprehension = true;
@@ -3228,7 +3228,7 @@ class Parser {
                 else {
                     const keyEntryNode = parseNodes_1.DictionaryKeyEntryNode.create(keyExpression, valueExpression);
                     let dictEntry = keyEntryNode;
-                    const listComp = this._tryParseListComprehension(keyEntryNode);
+                    const listComp = this._tryParseListComprehension(keyEntryNode, /* isGenerator */ false);
                     if (listComp) {
                         dictEntry = listComp;
                         sawListComprehension = true;
@@ -3248,7 +3248,7 @@ class Parser {
                     const listEntryNode = parseNodes_1.DictionaryExpandEntryNode.create(doubleStarExpression);
                     (0, parseNodes_1.extendRange)(listEntryNode, doubleStar);
                     let expandEntryNode = listEntryNode;
-                    const listComp = this._tryParseListComprehension(listEntryNode);
+                    const listComp = this._tryParseListComprehension(listEntryNode, /* isGenerator */ false);
                     if (listComp) {
                         expandEntryNode = listComp;
                         sawListComprehension = true;
@@ -3270,7 +3270,7 @@ class Parser {
                         this._addError(localize_1.Localizer.Diagnostic.dictKeyValuePairs(), keyExpression);
                     }
                     else {
-                        const listComp = this._tryParseListComprehension(keyExpression);
+                        const listComp = this._tryParseListComprehension(keyExpression, /* isGenerator */ false);
                         if (listComp) {
                             keyExpression = listComp;
                             sawListComprehension = true;