@zzzen/pyright-internal 1.2.0-dev.20261101 → 1.2.0-dev.20261108

package/dist/analyzer/typeEvaluator/classConstruction.d.ts

@@ -0,0 +1,25 @@
+import { ExpressionNode, ParseNode } from '../../parser/parseNodes';
+import * as AnalyzerNodeInfo from '../analyzerNodeInfo';
+import { Arg, ExpectedTypeOptions, PrefetchedTypes, TypeEvaluator, TypeResult } from '../typeEvaluatorTypes';
+import { ClassType, Type, TypeVarType, UnknownType } from '../types';
+import { InferenceContext } from '../typeUtils';
+interface Context {
+    readonly addDiagnostic: TypeEvaluator['addDiagnostic'];
+    readonly assignType: TypeEvaluator['assignType'];
+    readonly getBaseClassesFromTypeCall: (baseTupleType: ClassType) => Type[] | undefined;
+    readonly getBuiltInType: (node: ParseNode, name: string) => Type;
+    readonly getNoneType: () => Type;
+    readonly getTypeOfArg: (arg: Arg, inferenceContext: InferenceContext | undefined) => TypeResult;
+    readonly getTypeOfArgExpectingType: (arg: Arg, options?: ExpectedTypeOptions) => TypeResult;
+    readonly nodeInfo: AnalyzerNodeInfo.AnalyzerNodeInfoAccessor;
+    readonly prefetched: Partial<PrefetchedTypes> | undefined;
+    readonly printType: TypeEvaluator['printType'];
+}
+export declare function createClassConstruction(context: Context): {
+    createNewType: (errorNode: ExpressionNode, argList: Arg[]) => ClassType | undefined;
+    createClassFromMetaclass: (errorNode: ExpressionNode, argList: Arg[], metaclass: ClassType) => ClassType | undefined;
+    createSubclass: (errorNode: ExpressionNode, type1: ClassType, type2: ClassType) => ClassType;
+    buildTypeParamsFromTypeArgs: (classType: ClassType) => TypeVarType[];
+    computeEffectiveMetaclass: (classType: ClassType, errorNode: ParseNode) => UnknownType | ClassType;
+};
+export {};

package/dist/analyzer/typeEvaluator/classConstruction.js

@@ -0,0 +1,251 @@
+"use strict";
+/*
+ * classConstruction.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Class synthesis and effective metaclass selection.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createClassConstruction = createClassConstruction;
+const debug_1 = require("../../common/debug");
+const diagnostic_1 = require("../../common/diagnostic");
+const diagnosticRules_1 = require("../../common/diagnosticRules");
+const localize_1 = require("../../localization/localize");
+const ParseTreeUtils = __importStar(require("../parseTreeUtils"));
+const symbol_1 = require("../symbol");
+const types_1 = require("../types");
+const typeUtils_1 = require("../typeUtils");
+// Synthesizes classes and resolves effective metaclasses, delegating evaluation and diagnostics.
+function createClassConstruction(context) {
+    const { addDiagnostic, assignType, getBaseClassesFromTypeCall, getBuiltInType, getNoneType, getTypeOfArg, getTypeOfArgExpectingType, nodeInfo, printType, } = context;
+    function createNewType(errorNode, argList) {
+        const fileInfo = nodeInfo.getFileInfo(errorNode);
+        let className = '';
+        if (argList.length !== 2) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportCallIssue, localize_1.LocMessage.newTypeParamCount(), errorNode);
+            return undefined;
+        }
+        const nameArg = argList[0];
+        if (nameArg.argCategory === 0 /* ArgCategory.Simple */ &&
+            nameArg.valueExpression &&
+            nameArg.valueExpression.nodeType === 48 /* ParseNodeType.StringList */) {
+            className = nameArg.valueExpression.d.strings.map((s) => s.d.value).join('');
+        }
+        if (!className) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportArgumentType, localize_1.LocMessage.newTypeBadName(), argList[0].node ?? errorNode);
+            return undefined;
+        }
+        if (errorNode.parent?.nodeType === 3 /* ParseNodeType.Assignment */ &&
+            errorNode.parent.d.leftExpr.nodeType === 38 /* ParseNodeType.Name */ &&
+            errorNode.parent.d.leftExpr.d.value !== className) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeNameMismatch(), errorNode.parent.d.leftExpr);
+            return undefined;
+        }
+        let baseClass = getTypeOfArgExpectingType(argList[1]).type;
+        let isBaseClassAny = false;
+        if ((0, types_1.isAnyOrUnknown)(baseClass)) {
+            baseClass = context.prefetched?.objectClass ?? types_1.UnknownType.create();
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeAnyOrUnknown(), argList[1].node ?? errorNode);
+            isBaseClassAny = true;
+        }
+        // Specifically disallow Annotated.
+        if (baseClass.props?.specialForm &&
+            (0, types_1.isClassInstance)(baseClass.props.specialForm) &&
+            types_1.ClassType.isBuiltIn(baseClass.props.specialForm, 'Annotated')) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeNotAClass(), argList[1].node || errorNode);
+            return undefined;
+        }
+        if (!(0, types_1.isInstantiableClass)(baseClass)) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeNotAClass(), argList[1].node || errorNode);
+            return undefined;
+        }
+        if (types_1.ClassType.isProtocolClass(baseClass) || types_1.ClassType.isTypedDictClass(baseClass)) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeProtocolClass(), argList[1].node || errorNode);
+        }
+        else if (baseClass.priv.literalValue !== undefined) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.newTypeLiteral(), argList[1].node || errorNode);
+        }
+        const classType = types_1.ClassType.createInstantiable(className, ParseTreeUtils.getClassFullName(errorNode, fileInfo.moduleName, className), fileInfo.moduleName, fileInfo.fileUri, 256 /* ClassTypeFlags.Final */ | 2097152 /* ClassTypeFlags.NewTypeClass */ | 4194304 /* ClassTypeFlags.ValidTypeAliasClass */, ParseTreeUtils.getTypeSourceId(errorNode), 
+        /* declaredMetaclass */ undefined, baseClass.shared.effectiveMetaclass);
+        classType.shared.baseClasses.push(isBaseClassAny ? types_1.AnyType.create() : baseClass);
+        (0, typeUtils_1.computeMroLinearization)(classType);
+        if (!isBaseClassAny) {
+            // Synthesize an __init__ method that accepts only the specified type.
+            const initType = types_1.FunctionType.createSynthesizedInstance('__init__');
+            types_1.FunctionType.addParam(initType, types_1.FunctionParam.create(0 /* ParamCategory.Simple */, types_1.AnyType.create(), types_1.FunctionParamFlags.TypeDeclared, 'self'));
+            types_1.FunctionType.addParam(initType, types_1.FunctionParam.create(0 /* ParamCategory.Simple */, types_1.ClassType.cloneAsInstance(baseClass), types_1.FunctionParamFlags.TypeDeclared, '_x'));
+            initType.shared.declaredReturnType = getNoneType();
+            types_1.ClassType.getSymbolTable(classType).set('__init__', symbol_1.Symbol.createWithType(4 /* SymbolFlags.ClassMember */, initType));
+            // Synthesize a trivial __new__ method.
+            const newType = types_1.FunctionType.createSynthesizedInstance('__new__', 1 /* FunctionTypeFlags.ConstructorMethod */);
+            types_1.FunctionType.addParam(newType, types_1.FunctionParam.create(0 /* ParamCategory.Simple */, types_1.AnyType.create(), types_1.FunctionParamFlags.TypeDeclared, 'cls'));
+            types_1.FunctionType.addDefaultParams(newType);
+            newType.shared.declaredReturnType = types_1.ClassType.cloneAsInstance(classType);
+            newType.priv.constructorTypeVarScopeId = (0, typeUtils_1.getTypeVarScopeId)(classType);
+            types_1.ClassType.getSymbolTable(classType).set('__new__', symbol_1.Symbol.createWithType(4 /* SymbolFlags.ClassMember */, newType));
+        }
+        return classType;
+    }
+    function createClassFromMetaclass(errorNode, argList, metaclass) {
+        const fileInfo = nodeInfo.getFileInfo(errorNode);
+        const arg0Type = getTypeOfArg(argList[0], /* inferenceContext */ undefined).type;
+        if (!(0, types_1.isClassInstance)(arg0Type) || !types_1.ClassType.isBuiltIn(arg0Type, 'str')) {
+            return undefined;
+        }
+        const className = arg0Type.priv.literalValue || '_';
+        const baseTupleType = getTypeOfArg(argList[1], /* inferenceContext */ undefined).type;
+        if (!(0, types_1.isClassInstance)(baseTupleType)) {
+            return undefined;
+        }
+        const baseClasses = getBaseClassesFromTypeCall(baseTupleType);
+        if (!baseClasses) {
+            return undefined;
+        }
+        const classType = types_1.ClassType.createInstantiable(className, ParseTreeUtils.getClassFullName(errorNode, fileInfo.moduleName, className), fileInfo.moduleName, fileInfo.fileUri, 4194304 /* ClassTypeFlags.ValidTypeAliasClass */, ParseTreeUtils.getTypeSourceId(errorNode), metaclass, baseTupleType.shared.effectiveMetaclass);
+        classType.shared.baseClasses.push(...baseClasses);
+        if (!(0, typeUtils_1.computeMroLinearization)(classType)) {
+            addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.methodOrdering(), errorNode);
+        }
+        return classType;
+    }
+    function createSubclass(errorNode, type1, type2) {
+        (0, debug_1.assert)((0, types_1.isInstantiableClass)(type1) && (0, types_1.isInstantiableClass)(type2));
+        // If both classes are class objects (type[A] and type[B]), create a new
+        // class object (type[A & B]) rather than "type[A] & type[B]".
+        let createClassObject = false;
+        if (types_1.TypeBase.getInstantiableDepth(type1) > 0 && types_1.TypeBase.getInstantiableDepth(type2) > 0) {
+            type1 = types_1.ClassType.cloneAsInstance(type1);
+            type2 = types_1.ClassType.cloneAsInstance(type2);
+            createClassObject = true;
+        }
+        const className = `<subclass of ${printType((0, typeUtils_1.convertToInstance)(type1), {
+            omitTypeArgsIfUnknown: true,
+        })} and ${printType((0, typeUtils_1.convertToInstance)(type2), { omitTypeArgsIfUnknown: true })}>`;
+        const fileInfo = nodeInfo.getFileInfo(errorNode);
+        // The effective metaclass of the intersection is the narrower of the two metaclasses.
+        let effectiveMetaclass = type1.shared.effectiveMetaclass;
+        if (type2.shared.effectiveMetaclass) {
+            if (!effectiveMetaclass || assignType(effectiveMetaclass, type2.shared.effectiveMetaclass)) {
+                effectiveMetaclass = type2.shared.effectiveMetaclass;
+            }
+        }
+        let newClassType = types_1.ClassType.createInstantiable(className, ParseTreeUtils.getClassFullName(errorNode, fileInfo.moduleName, className), fileInfo.moduleName, fileInfo.fileUri, 0 /* ClassTypeFlags.None */, ParseTreeUtils.getTypeSourceId(errorNode), 
+        /* declaredMetaclass */ undefined, effectiveMetaclass, type1.shared.docString);
+        newClassType.shared.baseClasses = [type1, type2];
+        (0, typeUtils_1.computeMroLinearization)(newClassType);
+        newClassType = (0, typeUtils_1.addConditionToType)(newClassType, type1.props?.condition);
+        newClassType = (0, typeUtils_1.addConditionToType)(newClassType, type2.props?.condition);
+        if (createClassObject) {
+            newClassType = types_1.ClassType.cloneAsInstantiable(newClassType);
+        }
+        return newClassType;
+    }
+    function buildTypeParamsFromTypeArgs(classType) {
+        const typeParams = [];
+        const typeArgs = classType.priv.typeArgs ?? [];
+        typeArgs.forEach((typeArg, index) => {
+            if ((0, types_1.isTypeVar)(typeArg)) {
+                typeParams.push(typeArg);
+                return;
+            }
+            // Synthesize a dummy type parameter.
+            const typeVar = types_1.TypeVarType.createInstance(`__P${index}`);
+            typeVar.shared.isSynthesized = true;
+            typeParams.push(typeVar);
+        });
+        return typeParams;
+    }
+    function computeEffectiveMetaclass(classType, errorNode) {
+        let effectiveMetaclass = classType.shared.declaredMetaclass;
+        let reportedMetaclassConflict = false;
+        if (!effectiveMetaclass || (0, types_1.isInstantiableClass)(effectiveMetaclass)) {
+            for (const baseClass of classType.shared.baseClasses) {
+                if ((0, types_1.isInstantiableClass)(baseClass)) {
+                    const baseClassMeta = baseClass.shared.effectiveMetaclass ?? context.prefetched?.typeClass;
+                    if (baseClassMeta && (0, types_1.isInstantiableClass)(baseClassMeta)) {
+                        // Make sure there is no metaclass conflict.
+                        if (!effectiveMetaclass) {
+                            effectiveMetaclass = baseClassMeta;
+                        }
+                        else if ((0, typeUtils_1.derivesFromClassRecursive)(baseClassMeta, effectiveMetaclass, /* ignoreUnknown */ false)) {
+                            effectiveMetaclass = baseClassMeta;
+                        }
+                        else if (!(0, typeUtils_1.derivesFromClassRecursive)(effectiveMetaclass, baseClassMeta, /* ignoreUnknown */ false)) {
+                            if (!reportedMetaclassConflict) {
+                                const diag = new diagnostic_1.DiagnosticAddendum();
+                                diag.addMessage(localize_1.LocAddendum.metaclassConflict().format({
+                                    metaclass1: printType((0, typeUtils_1.convertToInstance)(effectiveMetaclass)),
+                                    metaclass2: printType((0, typeUtils_1.convertToInstance)(baseClassMeta)),
+                                }));
+                                addDiagnostic(diagnosticRules_1.DiagnosticRule.reportGeneralTypeIssues, localize_1.LocMessage.metaclassConflict() + diag.getString(), errorNode);
+                                // Don't report more than once.
+                                reportedMetaclassConflict = true;
+                            }
+                        }
+                    }
+                    else {
+                        effectiveMetaclass = baseClassMeta ? types_1.UnknownType.create() : undefined;
+                        break;
+                    }
+                }
+                else {
+                    // If one of the base classes is unknown, then the effective
+                    // metaclass is also unknowable.
+                    effectiveMetaclass = types_1.UnknownType.create();
+                    break;
+                }
+            }
+        }
+        // If we haven't found an effective metaclass, assume "type", which
+        // is the metaclass for "object".
+        if (!effectiveMetaclass) {
+            const typeMetaclass = getBuiltInType(errorNode, 'type');
+            effectiveMetaclass =
+                typeMetaclass && (0, types_1.isInstantiableClass)(typeMetaclass) ? typeMetaclass : types_1.UnknownType.create();
+        }
+        classType.shared.effectiveMetaclass = effectiveMetaclass;
+        return effectiveMetaclass;
+    }
+    return {
+        createNewType,
+        createClassFromMetaclass,
+        createSubclass,
+        buildTypeParamsFromTypeArgs,
+        computeEffectiveMetaclass,
+    };
+}
+//# sourceMappingURL=classConstruction.js.map

package/dist/analyzer/typeEvaluator/classConstruction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"classConstruction.js","sourceRoot":"","sources":["../../../../../../src/analyzer/typeEvaluator/classConstruction.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDH,0DA4UC;AA7XD,8CAA4C;AAC5C,wDAA6D;AAC7D,kEAA8D;AAC9D,0DAAsE;AAGtE,kEAAoD;AACpD,sCAAgD;AAEhD,oCAgBkB;AAClB,4CAOsB;AAetB,iGAAiG;AACjG,SAAgB,uBAAuB,CAAC,OAAgB;IACpD,MAAM,EACF,aAAa,EACb,UAAU,EACV,0BAA0B,EAC1B,cAAc,EACd,WAAW,EACX,YAAY,EACZ,yBAAyB,EACzB,QAAQ,EACR,SAAS,GACZ,GAAG,OAAO,CAAC;IAEZ,SAAS,aAAa,CAAC,SAAyB,EAAE,OAAc;QAC5D,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,SAAS,GAAG,EAAE,CAAC;QAEnB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,aAAa,CAAC,gCAAc,CAAC,eAAe,EAAE,qBAAU,CAAC,iBAAiB,EAAE,EAAE,SAAS,CAAC,CAAC;YACzF,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC3B,IACI,OAAO,CAAC,WAAW,+BAAuB;YAC1C,OAAO,CAAC,eAAe;YACvB,OAAO,CAAC,eAAe,CAAC,QAAQ,sCAA6B,EAC/D,CAAC;YACC,SAAS,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,aAAa,CAAC,gCAAc,CAAC,kBAAkB,EAAE,qBAAU,CAAC,cAAc,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC;YAC5G,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,IACI,SAAS,CAAC,MAAM,EAAE,QAAQ,qCAA6B;YACvD,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,gCAAuB;YAC3D,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,EACnD,CAAC;YACC,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,mBAAmB,EAAE,EAChC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAC9B,CAAC;YACF,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,IAAI,SAAS,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3D,IAAI,cAAc,GAAG,KAAK,CAAC;QAE3B,IAAI,IAAA,sBAAc,EAAC,SAAS,CAAC,EAAE,CAAC;YAC5B,SAAS,GAAG,OAAO,CAAC,UAAU,EAAE,WAAW,IAAI,mBAAW,CAAC,MAAM,EAAE,CAAC;YAEpE,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,mBAAmB,EAAE,EAChC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAC/B,CAAC;YAEF,cAAc,GAAG,IAAI,CAAC;QAC1B,CAAC;QAED,mCAAmC;QACnC,IACI,SAAS,CAAC,KAAK,EAAE,WAAW;YAC5B,IAAA,uBAAe,EAAC,SAAS,CAAC,KAAK,CAAC,WAAW,CAAC;YAC5C,iBAAS,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,EAAE,WAAW,CAAC,EAC/D,CAAC;YACC,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,gBAAgB,EAAE,EAC7B,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAC/B,CAAC;YACF,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,IAAA,2BAAmB,EAAC,SAAS,CAAC,EAAE,CAAC;YAClC,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,gBAAgB,EAAE,EAC7B,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAC/B,CAAC;YACF,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,IAAI,iBAAS,CAAC,eAAe,CAAC,SAAS,CAAC,IAAI,iBAAS,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,CAAC;YAChF,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,oBAAoB,EAAE,EACjC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAC/B,CAAC;QACN,CAAC;aAAM,IAAI,SAAS,CAAC,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACnD,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,cAAc,EAAE,EAC3B,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAC/B,CAAC;QACN,CAAC;QAED,MAAM,SAAS,GAAG,iBAAS,CAAC,kBAAkB,CAC1C,SAAS,EACT,cAAc,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,SAAS,CAAC,EAC1E,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,OAAO,EAChB,0EAAkD,mDAAqC,EACvF,cAAc,CAAC,eAAe,CAAC,SAAS,CAAC;QACzC,uBAAuB,CAAC,SAAS,EACjC,SAAS,CAAC,MAAM,CAAC,kBAAkB,CACtC,CAAC;QACF,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,eAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACjF,IAAA,mCAAuB,EAAC,SAAS,CAAC,CAAC;QAEnC,IAAI,CAAC,cAAc,EAAE,CAAC;YAClB,sEAAsE;YACtE,MAAM,QAAQ,GAAG,oBAAY,CAAC,yBAAyB,CAAC,UAAU,CAAC,CAAC;YACpE,oBAAY,CAAC,QAAQ,CACjB,QAAQ,EACR,qBAAa,CAAC,MAAM,+BAAuB,eAAO,CAAC,MAAM,EAAE,EAAE,0BAAkB,CAAC,YAAY,EAAE,MAAM,CAAC,CACxG,CAAC;YACF,oBAAY,CAAC,QAAQ,CACjB,QAAQ,EACR,qBAAa,CAAC,MAAM,+BAEhB,iBAAS,CAAC,eAAe,CAAC,SAAS,CAAC,EACpC,0BAAkB,CAAC,YAAY,EAC/B,IAAI,CACP,CACJ,CAAC;YACF,QAAQ,CAAC,MAAM,CAAC,kBAAkB,GAAG,WAAW,EAAE,CAAC;YACnD,iBAAS,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,GAAG,CACnC,UAAU,EACV,eAAM,CAAC,cAAc,kCAA0B,QAAQ,CAAC,CAC3D,CAAC;YAEF,uCAAuC;YACvC,MAAM,OAAO,GAAG,oBAAY,CAAC,yBAAyB,CAAC,SAAS,8CAAsC,CAAC;YACvG,oBAAY,CAAC,QAAQ,CACjB,OAAO,EACP,qBAAa,CAAC,MAAM,+BAAuB,eAAO,CAAC,MAAM,EAAE,EAAE,0BAAkB,CAAC,YAAY,EAAE,KAAK,CAAC,CACvG,CAAC;YACF,oBAAY,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YACvC,OAAO,CAAC,MAAM,CAAC,kBAAkB,GAAG,iBAAS,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACzE,OAAO,CAAC,IAAI,CAAC,yBAAyB,GAAG,IAAA,6BAAiB,EAAC,SAAS,CAAC,CAAC;YACtE,iBAAS,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,SAAS,EAAE,eAAM,CAAC,cAAc,kCAA0B,OAAO,CAAC,CAAC,CAAC;QAChH,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,SAAS,wBAAwB,CAC7B,SAAyB,EACzB,OAAc,EACd,SAAoB;QAEpB,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC;QACjF,IAAI,CAAC,IAAA,uBAAe,EAAC,QAAQ,CAAC,IAAI,CAAC,iBAAS,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC;YACtE,OAAO,SAAS,CAAC;QACrB,CAAC;QACD,MAAM,SAAS,GAAI,QAAQ,CAAC,IAAI,CAAC,YAAuB,IAAI,GAAG,CAAC;QAEhE,MAAM,aAAa,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC;QACtF,IAAI,CAAC,IAAA,uBAAe,EAAC,aAAa,CAAC,EAAE,CAAC;YAClC,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,MAAM,WAAW,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;QAC9D,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,MAAM,SAAS,GAAG,iBAAS,CAAC,kBAAkB,CAC1C,SAAS,EACT,cAAc,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,SAAS,CAAC,EAC1E,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,OAAO,oDAEhB,cAAc,CAAC,eAAe,CAAC,SAAS,CAAC,EACzC,SAAS,EACT,aAAa,CAAC,MAAM,CAAC,kBAAkB,CAC1C,CAAC;QACF,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;QAElD,IAAI,CAAC,IAAA,mCAAuB,EAAC,SAAS,CAAC,EAAE,CAAC;YACtC,aAAa,CAAC,gCAAc,CAAC,uBAAuB,EAAE,qBAAU,CAAC,cAAc,EAAE,EAAE,SAAS,CAAC,CAAC;QAClG,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,SAAS,cAAc,CAAC,SAAyB,EAAE,KAAgB,EAAE,KAAgB;QACjF,IAAA,cAAM,EAAC,IAAA,2BAAmB,EAAC,KAAK,CAAC,IAAI,IAAA,2BAAmB,EAAC,KAAK,CAAC,CAAC,CAAC;QAEjE,wEAAwE;QACxE,8DAA8D;QAC9D,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,IAAI,gBAAQ,CAAC,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,gBAAQ,CAAC,oBAAoB,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACvF,KAAK,GAAG,iBAAS,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACzC,KAAK,GAAG,iBAAS,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACzC,iBAAiB,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,MAAM,SAAS,GAAG,gBAAgB,SAAS,CAAC,IAAA,6BAAiB,EAAC,KAAK,CAAC,EAAE;YAClE,qBAAqB,EAAE,IAAI;SAC9B,CAAC,QAAQ,SAAS,CAAC,IAAA,6BAAiB,EAAC,KAAK,CAAC,EAAE,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC;QAClF,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAEjD,sFAAsF;QACtF,IAAI,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC;QACzD,IAAI,KAAK,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAClC,IAAI,CAAC,kBAAkB,IAAI,UAAU,CAAC,kBAAkB,EAAE,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,CAAC;gBACzF,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC,kBAAkB,CAAC;YACzD,CAAC;QACL,CAAC;QAED,IAAI,YAAY,GAAG,iBAAS,CAAC,kBAAkB,CAC3C,SAAS,EACT,cAAc,CAAC,gBAAgB,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,SAAS,CAAC,EAC1E,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,OAAO,+BAEhB,cAAc,CAAC,eAAe,CAAC,SAAS,CAAC;QACzC,uBAAuB,CAAC,SAAS,EACjC,kBAAkB,EAClB,KAAK,CAAC,MAAM,CAAC,SAAS,CACzB,CAAC;QAEF,YAAY,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACjD,IAAA,mCAAuB,EAAC,YAAY,CAAC,CAAC;QAEtC,YAAY,GAAG,IAAA,8BAAkB,EAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACxE,YAAY,GAAG,IAAA,8BAAkB,EAAC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAExE,IAAI,iBAAiB,EAAE,CAAC;YACpB,YAAY,GAAG,iBAAS,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;QAC/D,CAAC;QAED,OAAO,YAAY,CAAC;IACxB,CAAC;IAED,SAAS,2BAA2B,CAAC,SAAoB;QACrD,MAAM,UAAU,GAAkB,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QAE/C,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YAChC,IAAI,IAAA,iBAAS,EAAC,OAAO,CAAC,EAAE,CAAC;gBACrB,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzB,OAAO;YACX,CAAC;YAED,qCAAqC;YACrC,MAAM,OAAO,GAAG,mBAAW,CAAC,cAAc,CAAC,MAAM,KAAK,EAAE,CAAC,CAAC;YAC1D,OAAO,CAAC,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC;YACpC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,SAAS,yBAAyB,CAAC,SAAoB,EAAE,SAAoB;QACzE,IAAI,kBAAkB,GAAG,SAAS,CAAC,MAAM,CAAC,iBAAiB,CAAC;QAC5D,IAAI,yBAAyB,GAAG,KAAK,CAAC;QAEtC,IAAI,CAAC,kBAAkB,IAAI,IAAA,2BAAmB,EAAC,kBAAkB,CAAC,EAAE,CAAC;YACjE,KAAK,MAAM,SAAS,IAAI,SAAS,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBACnD,IAAI,IAAA,2BAAmB,EAAC,SAAS,CAAC,EAAE,CAAC;oBACjC,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC,kBAAkB,IAAI,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC;oBAC3F,IAAI,aAAa,IAAI,IAAA,2BAAmB,EAAC,aAAa,CAAC,EAAE,CAAC;wBACtD,4CAA4C;wBAC5C,IAAI,CAAC,kBAAkB,EAAE,CAAC;4BACtB,kBAAkB,GAAG,aAAa,CAAC;wBACvC,CAAC;6BAAM,IACH,IAAA,qCAAyB,EAAC,aAAa,EAAE,kBAAkB,EAAE,mBAAmB,CAAC,KAAK,CAAC,EACzF,CAAC;4BACC,kBAAkB,GAAG,aAAa,CAAC;wBACvC,CAAC;6BAAM,IACH,CAAC,IAAA,qCAAyB,EAAC,kBAAkB,EAAE,aAAa,EAAE,mBAAmB,CAAC,KAAK,CAAC,EAC1F,CAAC;4BACC,IAAI,CAAC,yBAAyB,EAAE,CAAC;gCAC7B,MAAM,IAAI,GAAG,IAAI,+BAAkB,EAAE,CAAC;gCAEtC,IAAI,CAAC,UAAU,CACX,sBAAW,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC;oCACnC,UAAU,EAAE,SAAS,CAAC,IAAA,6BAAiB,EAAC,kBAAkB,CAAC,CAAC;oCAC5D,UAAU,EAAE,SAAS,CAAC,IAAA,6BAAiB,EAAC,aAAa,CAAC,CAAC;iCAC1D,CAAC,CACL,CAAC;gCACF,aAAa,CACT,gCAAc,CAAC,uBAAuB,EACtC,qBAAU,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,EACjD,SAAS,CACZ,CAAC;gCAEF,+BAA+B;gCAC/B,yBAAyB,GAAG,IAAI,CAAC;4BACrC,CAAC;wBACL,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACJ,kBAAkB,GAAG,aAAa,CAAC,CAAC,CAAC,mBAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;wBACtE,MAAM;oBACV,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACJ,4DAA4D;oBAC5D,gCAAgC;oBAChC,kBAAkB,GAAG,mBAAW,CAAC,MAAM,EAAE,CAAC;oBAC1C,MAAM;gBACV,CAAC;YACL,CAAC;QACL,CAAC;QAED,mEAAmE;QACnE,iCAAiC;QACjC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACtB,MAAM,aAAa,GAAG,cAAc,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;YACxD,kBAAkB;gBACd,aAAa,IAAI,IAAA,2BAAmB,EAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,mBAAW,CAAC,MAAM,EAAE,CAAC;QACnG,CAAC;QAED,SAAS,CAAC,MAAM,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAEzD,OAAO,kBAAkB,CAAC;IAC9B,CAAC;IAED,OAAO;QACH,aAAa;QACb,wBAAwB;QACxB,cAAc;QACd,2BAA2B;QAC3B,yBAAyB;KAC5B,CAAC;AACN,CAAC"}

package/dist/analyzer/typeEvaluator/collectionInference.d.ts

@@ -0,0 +1,41 @@
+import { DiagnosticAddendum } from '../../common/diagnostic';
+import { ComprehensionForIfNode, ComprehensionNode, DictionaryNode, ExpressionNode, ListNode, ParseNode, SetNode } from '../../parser/parseNodes';
+import * as AnalyzerNodeInfo from '../analyzerNodeInfo';
+import { ConstraintTracker } from '../constraintTracker';
+import { EvalFlags, PrefetchedTypes, SolveConstraintsOptions, TypeEvaluator, TypeResult } from '../typeEvaluatorTypes';
+import { AnyType, ClassType, FunctionType, ModuleType, NeverType, OverloadedType, OverloadResultType, Type, TypeCondition, TypeVarType, UnboundType, UnionType, UnknownType } from '../types';
+import { ApplyTypeVarOptions, InferenceContext } from '../typeUtils';
+interface Context {
+    readonly addDiagnostic: TypeEvaluator['addDiagnostic'];
+    readonly assignType: TypeEvaluator['assignType'];
+    readonly assignTypeToExpression: (target: ExpressionNode, typeResult: TypeResult, srcExpr: ExpressionNode, ignoreEmptyContainers?: boolean, allowAssignmentToFinalVar?: boolean, expectedTypeDiagAddendum?: DiagnosticAddendum) => void;
+    readonly convertSpecialFormToRuntimeValue: (type: Type, flags: EvalFlags, convertModule?: boolean) => Type;
+    readonly evaluatorInterface: TypeEvaluator;
+    readonly getBuiltInObject: (node: ParseNode, name: string, typeArgs?: Type[]) => UnboundType | UnknownType | AnyType | FunctionType | OverloadedType | ClassType | ModuleType | TypeVarType | OverloadResultType | NeverType | UnionType;
+    readonly getBuiltInType: (node: ParseNode, name: string) => Type;
+    readonly getNoneType: () => Type;
+    readonly getObjectType: () => Type;
+    readonly getTupleClassType: () => ClassType | undefined;
+    readonly getTypeOfExpression: TypeEvaluator['getTypeOfExpression'];
+    readonly getTypeOfIterator: (typeResult: TypeResult, isAsync: boolean, errorNode: ExpressionNode, emitNotIterableError?: boolean) => TypeResult | undefined;
+    readonly getTypeVarScopesForNode: (node: ParseNode) => string[];
+    readonly getTypingType: (node: ParseNode, symbolName: string) => Type | undefined;
+    readonly inferVarianceForClass: (classType: ClassType) => void;
+    readonly isTypeHashable: (type: Type) => boolean;
+    readonly makeTopLevelTypeVarsConcrete: (type: Type, makeParamSpecsConcrete?: boolean, conditionFilter?: TypeCondition[]) => Type;
+    readonly maxEntriesToUseForInference: 64;
+    readonly nodeInfo: AnalyzerNodeInfo.AnalyzerNodeInfoAccessor;
+    readonly prefetched: Partial<PrefetchedTypes> | undefined;
+    readonly printType: TypeEvaluator['printType'];
+    readonly solveAndApplyConstraints: (type: Type, constraints: ConstraintTracker, applyOptions?: ApplyTypeVarOptions, solveOptions?: SolveConstraintsOptions) => Type;
+    readonly stripLiteralValue: (type: Type) => Type;
+}
+export declare function createCollectionInference(context: Context): {
+    getTypeOfDictionaryWithContext: (node: DictionaryNode, flags: EvalFlags, inferenceContext: InferenceContext, expectedDiagAddendum?: DiagnosticAddendum) => TypeResult | undefined;
+    getTypeOfDictionaryInferred: (node: DictionaryNode, flags: EvalFlags, hasExpectedType: boolean) => TypeResult;
+    getTypeOfListOrSetWithContext: (node: ListNode | SetNode, flags: EvalFlags, inferenceContext: InferenceContext) => TypeResult | undefined;
+    getTypeOfListOrSetInferred: (node: ListNode | SetNode, flags: EvalFlags, hasExpectedType: boolean) => TypeResult;
+    getTypeOfComprehension: (node: ComprehensionNode, flags: EvalFlags, inferenceContext?: InferenceContext) => TypeResult;
+    evaluateComprehensionForIf: (node: ComprehensionForIfNode) => boolean;
+};
+export {};