driftdetect-core 0.1.2 → 0.3.0

package/dist/parsers/tree-sitter/pydantic/field-extractor.js

@@ -0,0 +1,285 @@
+/**
+ * Pydantic Field Extractor
+ *
+ * Extracts field definitions from Pydantic model class bodies.
+ * Handles annotated assignments, Field() calls, and default values.
+ *
+ * @module pydantic/field-extractor
+ */
+import { extractPositionRange } from './types.js';
+import { TypeResolver } from './type-resolver.js';
+import { ConstraintParser } from './constraint-parser.js';
+// ============================================
+// Field Extractor Class
+// ============================================
+/**
+ * Extracts Pydantic field definitions from class bodies.
+ *
+ * Handles:
+ * - Simple annotations: `name: str`
+ * - Annotations with defaults: `name: str = "default"`
+ * - Field() calls: `name: str = Field(default="value", min_length=1)`
+ * - Optional fields: `name: Optional[str] = None`
+ */
+export class FieldExtractor {
+    typeResolver;
+    constraintParser;
+    constructor() {
+        this.typeResolver = new TypeResolver();
+        this.constraintParser = new ConstraintParser();
+    }
+    /**
+     * Extract all fields from a class body node.
+     *
+     * @param bodyNode - The class body (block) node
+     * @param context - Extraction context
+     * @returns Array of extracted field information
+     */
+    extractFields(bodyNode, context) {
+        const fields = [];
+        for (const child of bodyNode.namedChildren) {
+            const field = this.extractField(child, context);
+            if (field) {
+                fields.push(field);
+            }
+        }
+        return fields;
+    }
+    /**
+     * Extract a single field from an AST node.
+     *
+     * @param node - AST node (expression_statement or typed_assignment)
+     * @param context - Extraction context
+     * @returns Field info or null if not a field
+     */
+    extractField(node, context) {
+        // Handle expression_statement wrapper
+        if (node.type === 'expression_statement') {
+            const inner = node.namedChildren[0];
+            if (!inner)
+                return null;
+            return this.extractField(inner, context);
+        }
+        // Must be an assignment with type annotation
+        if (node.type !== 'assignment') {
+            return null;
+        }
+        const rawData = this.extractRawFieldData(node);
+        if (!rawData) {
+            return null;
+        }
+        return this.processRawField(rawData, context);
+    }
+    // ============================================
+    // Raw Data Extraction
+    // ============================================
+    /**
+     * Extract raw field data from an assignment node.
+     */
+    extractRawFieldData(node) {
+        // Get the left side (should be identifier with type annotation)
+        const left = node.childForFieldName('left');
+        if (!left)
+            return null;
+        // Check for type annotation
+        const typeNode = node.childForFieldName('type');
+        // Get field name
+        let name;
+        if (left.type === 'identifier') {
+            name = left.text;
+        }
+        else {
+            return null; // Not a simple field assignment
+        }
+        // Skip private/dunder fields
+        if (name.startsWith('_')) {
+            return null;
+        }
+        // Get type annotation
+        const typeAnnotation = typeNode?.text ?? null;
+        // Get right side (default value)
+        const right = node.childForFieldName('right');
+        const defaultValue = right?.text ?? null;
+        // Check if Field() is used
+        const { usesField, fieldArguments } = this.parseFieldCall(right);
+        return {
+            name,
+            typeAnnotation,
+            defaultValue,
+            usesField,
+            fieldArguments,
+            node,
+        };
+    }
+    /**
+     * Parse a Field() call and extract arguments.
+     */
+    parseFieldCall(node) {
+        if (!node) {
+            return { usesField: false, fieldArguments: [] };
+        }
+        // Check if this is a call to Field
+        if (node.type !== 'call') {
+            return { usesField: false, fieldArguments: [] };
+        }
+        const funcNode = node.childForFieldName('function');
+        if (!funcNode) {
+            return { usesField: false, fieldArguments: [] };
+        }
+        // Check function name
+        const funcName = funcNode.text;
+        if (funcName !== 'Field' && !funcName.endsWith('.Field')) {
+            return { usesField: false, fieldArguments: [] };
+        }
+        // Extract arguments
+        const argsNode = node.childForFieldName('arguments');
+        const fieldArguments = this.extractCallArguments(argsNode);
+        return { usesField: true, fieldArguments };
+    }
+    /**
+     * Extract arguments from a call's argument_list.
+     */
+    extractCallArguments(argsNode) {
+        if (!argsNode)
+            return [];
+        const args = [];
+        for (const child of argsNode.namedChildren) {
+            if (child.type === 'keyword_argument') {
+                const nameNode = child.childForFieldName('name');
+                const valueNode = child.childForFieldName('value');
+                if (nameNode && valueNode) {
+                    args.push({
+                        name: nameNode.text,
+                        value: valueNode.text,
+                        isKeyword: true,
+                    });
+                }
+            }
+            else {
+                // Positional argument
+                args.push({
+                    name: null,
+                    value: child.text,
+                    isKeyword: false,
+                });
+            }
+        }
+        return args;
+    }
+    // ============================================
+    // Field Processing
+    // ============================================
+    /**
+     * Process raw field data into PydanticFieldInfo.
+     */
+    processRawField(raw, context) {
+        // Resolve type
+        const typeInfo = raw.typeAnnotation
+            ? this.typeResolver.resolve(raw.typeAnnotation, context)
+            : this.typeResolver.createUnknownType();
+        // Parse constraints from Field() arguments
+        const constraints = raw.usesField
+            ? this.constraintParser.parseConstraints(raw.fieldArguments)
+            : this.createEmptyConstraints();
+        // Determine if field is required
+        const required = this.isFieldRequired(raw, typeInfo.isOptional);
+        // Extract alias and description from constraints
+        const alias = this.extractStringConstraint(raw.fieldArguments, 'alias');
+        const description = this.extractStringConstraint(raw.fieldArguments, 'description');
+        // Get default value (excluding Field() call)
+        const defaultValue = this.extractDefaultValue(raw);
+        // Build position info
+        const positions = context.includePositions
+            ? extractPositionRange(raw.node)
+            : { startPosition: { row: 0, column: 0 }, endPosition: { row: 0, column: 0 } };
+        return {
+            name: raw.name,
+            type: typeInfo,
+            defaultValue,
+            hasDefault: raw.defaultValue !== null,
+            usesField: raw.usesField,
+            constraints,
+            alias,
+            description,
+            required,
+            ...positions,
+        };
+    }
+    /**
+     * Determine if a field is required.
+     */
+    isFieldRequired(raw, isOptional) {
+        // If Optional type, not required
+        if (isOptional)
+            return false;
+        // If has default value, not required
+        if (raw.defaultValue !== null)
+            return false;
+        // If Field() with default or default_factory, not required
+        if (raw.usesField) {
+            const hasDefault = raw.fieldArguments.some((arg) => arg.name === 'default' || arg.name === 'default_factory');
+            if (hasDefault)
+                return false;
+            // Check for ... (Ellipsis) as first positional arg (required marker)
+            const firstPositional = raw.fieldArguments.find((arg) => !arg.isKeyword);
+            if (firstPositional?.value === '...')
+                return true;
+        }
+        return true;
+    }
+    /**
+     * Extract the actual default value (not Field() call).
+     */
+    extractDefaultValue(raw) {
+        if (!raw.defaultValue)
+            return null;
+        // If using Field(), extract default from arguments
+        if (raw.usesField) {
+            const defaultArg = raw.fieldArguments.find((arg) => arg.name === 'default');
+            if (defaultArg) {
+                return defaultArg.value;
+            }
+            // Check first positional argument (if not ...)
+            const firstPositional = raw.fieldArguments.find((arg) => !arg.isKeyword);
+            if (firstPositional && firstPositional.value !== '...') {
+                return firstPositional.value;
+            }
+            return null;
+        }
+        return raw.defaultValue;
+    }
+    /**
+     * Extract a string constraint value.
+     */
+    extractStringConstraint(args, name) {
+        const arg = args.find((a) => a.name === name);
+        if (!arg)
+            return null;
+        // Remove quotes from string value
+        const value = arg.value;
+        if ((value.startsWith('"') && value.endsWith('"')) ||
+            (value.startsWith("'") && value.endsWith("'"))) {
+            return value.slice(1, -1);
+        }
+        return value;
+    }
+    /**
+     * Create empty constraints object.
+     */
+    createEmptyConstraints() {
+        return {
+            ge: null,
+            le: null,
+            gt: null,
+            lt: null,
+            minLength: null,
+            maxLength: null,
+            pattern: null,
+            multipleOf: null,
+            frozen: false,
+            validators: [],
+            jsonSchemaExtra: null,
+        };
+    }
+}
+//# sourceMappingURL=field-extractor.js.map

package/dist/parsers/tree-sitter/pydantic/field-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"field-extractor.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/field-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE1D,+CAA+C;AAC/C,wBAAwB;AACxB,+CAA+C;AAE/C;;;;;;;;GAQG;AACH,MAAM,OAAO,cAAc;IACR,YAAY,CAAe;IAC3B,gBAAgB,CAAmB;IAEpD;QACE,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACvC,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;IACjD,CAAC;IAED;;;;;;OAMG;IACH,aAAa,CACX,QAAwB,EACxB,OAA0B;QAE1B,MAAM,MAAM,GAAwB,EAAE,CAAC;QAEvC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAChD,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;OAMG;IACH,YAAY,CACV,IAAoB,EACpB,OAA0B;QAE1B,sCAAsC;QACtC,IAAI,IAAI,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,CAAC,KAAK;gBAAE,OAAO,IAAI,CAAC;YACxB,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC3C,CAAC;QAED,6CAA6C;QAC7C,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,+CAA+C;IAC/C,sBAAsB;IACtB,+CAA+C;IAE/C;;OAEG;IACK,mBAAmB,CAAC,IAAoB;QAC9C,gEAAgE;QAChE,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAEvB,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAEhD,iBAAiB;QACjB,IAAI,IAAY,CAAC;QACjB,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC/B,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,CAAC,gCAAgC;QAC/C,CAAC;QAED,6BAA6B;QAC7B,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,sBAAsB;QACtB,MAAM,cAAc,GAAG,QAAQ,EAAE,IAAI,IAAI,IAAI,CAAC;QAE9C,iCAAiC;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,YAAY,GAAG,KAAK,EAAE,IAAI,IAAI,IAAI,CAAC;QAEzC,2BAA2B;QAC3B,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAEjE,OAAO;YACL,IAAI;YACJ,cAAc;YACd,YAAY;YACZ,SAAS;YACT,cAAc;YACd,IAAI;SACL,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,IAA2B;QAIhD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,mCAAmC;QACnC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,sBAAsB;QACtB,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC/B,IAAI,QAAQ,KAAK,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzD,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,oBAAoB;QACpB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE3D,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,QAA+B;QAC1D,IAAI,CAAC,QAAQ;YAAE,OAAO,EAAE,CAAC;QAEzB,MAAM,IAAI,GAAoB,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBACjD,MAAM,SAAS,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBACnD,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;oBAC1B,IAAI,CAAC,IAAI,CAAC;wBACR,IAAI,EAAE,QAAQ,CAAC,IAAI;wBACnB,KAAK,EAAE,SAAS,CAAC,IAAI;wBACrB,SAAS,EAAE,IAAI;qBAChB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,sBAAsB;gBACtB,IAAI,CAAC,IAAI,CAAC;oBACR,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,KAAK,CAAC,IAAI;oBACjB,SAAS,EAAE,KAAK;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,+CAA+C;IAC/C,mBAAmB;IACnB,+CAA+C;IAE/C;;OAEG;IACK,eAAe,CACrB,GAAiB,EACjB,OAA0B;QAE1B,eAAe;QACf,MAAM,QAAQ,GAAG,GAAG,CAAC,cAAc;YACjC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC;YACxD,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,CAAC;QAE1C,2CAA2C;QAC3C,MAAM,WAAW,GAAG,GAAG,CAAC,SAAS;YAC/B,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,GAAG,CAAC,cAAc,CAAC;YAC5D,CAAC,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAElC,iCAAiC;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QAEhE,iDAAiD;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACxE,MAAM,WAAW,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;QAEpF,6CAA6C;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAEnD,sBAAsB;QACtB,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB;YACxC,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC;YAChC,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAEjF,OAAO;YACL,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,QAAQ;YACd,YAAY;YACZ,UAAU,EAAE,GAAG,CAAC,YAAY,KAAK,IAAI;YACrC,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,WAAW;YACX,KAAK;YACL,WAAW;YACX,QAAQ;YACR,GAAG,SAAS;SACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,GAAiB,EAAE,UAAmB;QAC5D,iCAAiC;QACjC,IAAI,UAAU;YAAE,OAAO,KAAK,CAAC;QAE7B,qCAAqC;QACrC,IAAI,GAAG,CAAC,YAAY,KAAK,IAAI;YAAE,OAAO,KAAK,CAAC;QAE5C,2DAA2D;QAC3D,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CACxC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAiB,CAClE,CAAC;YACF,IAAI,UAAU;gBAAE,OAAO,KAAK,CAAC;YAE7B,qEAAqE;YACrE,MAAM,eAAe,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACzE,IAAI,eAAe,EAAE,KAAK,KAAK,KAAK;gBAAE,OAAO,IAAI,CAAC;QACpD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,GAAiB;QAC3C,IAAI,CAAC,GAAG,CAAC,YAAY;YAAE,OAAO,IAAI,CAAC;QAEnC,mDAAmD;QACnD,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAC5E,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,UAAU,CAAC,KAAK,CAAC;YAC1B,CAAC;YACD,+CAA+C;YAC/C,MAAM,eAAe,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACzE,IAAI,eAAe,IAAI,eAAe,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;gBACvD,OAAO,eAAe,CAAC,KAAK,CAAC;YAC/B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,GAAG,CAAC,YAAY,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,IAAqB,EACrB,IAAY;QAEZ,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QAEtB,kCAAkC;QAClC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAC9C,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACnD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,sBAAsB;QAC5B,OAAO;YACL,EAAE,EAAE,IAAI;YACR,EAAE,EAAE,IAAI;YACR,EAAE,EAAE,IAAI;YACR,EAAE,EAAE,IAAI;YACR,SAAS,EAAE,IAAI;YACf,SAAS,EAAE,IAAI;YACf,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,IAAI;YAChB,MAAM,EAAE,KAAK;YACb,UAAU,EAAE,EAAE;YACd,eAAe,EAAE,IAAI;SACtB,CAAC;IACJ,CAAC;CACF"}

package/dist/parsers/tree-sitter/pydantic/index.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Pydantic Module
+ *
+ * Public exports for Pydantic model extraction functionality.
+ * Provides comprehensive parsing of Pydantic v1 and v2 models.
+ *
+ * @module pydantic
+ */
+export { PydanticExtractor, modelsToContractFields, findModel, getModelNames } from './pydantic-extractor.js';
+export { FieldExtractor } from './field-extractor.js';
+export { TypeResolver } from './type-resolver.js';
+export { ConstraintParser, hasNumericConstraints, hasStringConstraints, hasAnyConstraints, mergeConstraints } from './constraint-parser.js';
+export { ValidatorExtractor } from './validator-extractor.js';
+export { ConfigExtractor } from './config-extractor.js';
+export { InheritanceResolver } from './inheritance-resolver.js';
+export type { ExtractionContext, RawFieldData, FieldArgument, ParsedType, RawValidatorData, ValidatorDecoratorType, ValidatorMode, RawConfigData, ConfigValue, BaseClassInfo, } from './types.js';
+export { createExtractionContext, PYTHON_TYPE_MAP, NUMERIC_CONSTRAINTS, STRING_CONSTRAINTS, ALL_CONSTRAINTS, PYDANTIC_BASE_CLASSES, extractPosition, extractPositionRange, } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/parsers/tree-sitter/pydantic/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAMH,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAM9G,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC5I,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAMhE,YAAY,EACV,iBAAiB,EACjB,YAAY,EACZ,aAAa,EACb,UAAU,EACV,gBAAgB,EAChB,sBAAsB,EACtB,aAAa,EACb,aAAa,EACb,WAAW,EACX,aAAa,GACd,MAAM,YAAY,CAAC;AAEpB,OAAO,EACL,uBAAuB,EACvB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,qBAAqB,EACrB,eAAe,EACf,oBAAoB,GACrB,MAAM,YAAY,CAAC"}

package/dist/parsers/tree-sitter/pydantic/index.js

@@ -0,0 +1,23 @@
+/**
+ * Pydantic Module
+ *
+ * Public exports for Pydantic model extraction functionality.
+ * Provides comprehensive parsing of Pydantic v1 and v2 models.
+ *
+ * @module pydantic
+ */
+// ============================================
+// Main Extractor
+// ============================================
+export { PydanticExtractor, modelsToContractFields, findModel, getModelNames } from './pydantic-extractor.js';
+// ============================================
+// Individual Extractors
+// ============================================
+export { FieldExtractor } from './field-extractor.js';
+export { TypeResolver } from './type-resolver.js';
+export { ConstraintParser, hasNumericConstraints, hasStringConstraints, hasAnyConstraints, mergeConstraints } from './constraint-parser.js';
+export { ValidatorExtractor } from './validator-extractor.js';
+export { ConfigExtractor } from './config-extractor.js';
+export { InheritanceResolver } from './inheritance-resolver.js';
+export { createExtractionContext, PYTHON_TYPE_MAP, NUMERIC_CONSTRAINTS, STRING_CONSTRAINTS, ALL_CONSTRAINTS, PYDANTIC_BASE_CLASSES, extractPosition, extractPositionRange, } from './types.js';
+//# sourceMappingURL=index.js.map

package/dist/parsers/tree-sitter/pydantic/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,+CAA+C;AAC/C,iBAAiB;AACjB,+CAA+C;AAE/C,OAAO,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAE9G,+CAA+C;AAC/C,wBAAwB;AACxB,+CAA+C;AAE/C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC5I,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAmBhE,OAAO,EACL,uBAAuB,EACvB,eAAe,EACf,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,qBAAqB,EACrB,eAAe,EACf,oBAAoB,GACrB,MAAM,YAAY,CAAC"}

package/dist/parsers/tree-sitter/pydantic/inheritance-resolver.d.ts

@@ -0,0 +1,70 @@
+/**
+ * Pydantic Inheritance Resolver
+ *
+ * Resolves model inheritance chains and merges fields from parent classes.
+ * Handles multiple inheritance and generic type substitution.
+ *
+ * @module pydantic/inheritance-resolver
+ */
+import type { PydanticModelInfo, PydanticFieldInfo } from '../types.js';
+import type { BaseClassInfo } from './types.js';
+/**
+ * Resolves Pydantic model inheritance.
+ *
+ * Handles:
+ * - Single inheritance: class Child(Parent)
+ * - Multiple inheritance: class Child(Parent1, Parent2)
+ * - Generic inheritance: class Child(Response[User])
+ * - Mixin classes
+ */
+export declare class InheritanceResolver {
+    /**
+     * Resolve all fields for a model including inherited fields.
+     *
+     * @param model - The model to resolve fields for
+     * @param allModels - Map of all known models by name
+     * @returns Array of all fields (inherited + own)
+     */
+    resolveFields(model: PydanticModelInfo, allModels: Map<string, PydanticModelInfo>): PydanticFieldInfo[];
+    /**
+     * Recursively resolve fields from inheritance chain.
+     */
+    private resolveFieldsRecursive;
+    /**
+     * Parse a base class string into structured info.
+     */
+    parseBaseClass(baseStr: string): BaseClassInfo;
+    /**
+     * Parse generic type arguments.
+     */
+    private parseTypeArgs;
+    /**
+     * Substitute generic type parameters in fields.
+     */
+    private substituteGenericTypes;
+    /**
+     * Find generic type parameters from a model's bases.
+     */
+    private findGenericParams;
+    /**
+     * Substitute type parameters in a TypeInfo.
+     */
+    private substituteType;
+    /**
+     * Get the full inheritance chain for a model.
+     */
+    getInheritanceChain(model: PydanticModelInfo, allModels: Map<string, PydanticModelInfo>): PydanticModelInfo[];
+    /**
+     * Recursively build inheritance chain.
+     */
+    private buildInheritanceChain;
+    /**
+     * Check if a model inherits from another.
+     */
+    inheritsFrom(model: PydanticModelInfo, ancestorName: string, allModels: Map<string, PydanticModelInfo>): boolean;
+    /**
+     * Find all models that inherit from a given model.
+     */
+    findDescendants(ancestorName: string, allModels: Map<string, PydanticModelInfo>): PydanticModelInfo[];
+}
+//# sourceMappingURL=inheritance-resolver.d.ts.map

package/dist/parsers/tree-sitter/pydantic/inheritance-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inheritance-resolver.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/inheritance-resolver.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAY,MAAM,aAAa,CAAC;AAClF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAOhD;;;;;;;;GAQG;AACH,qBAAa,mBAAmB;IAC9B;;;;;;OAMG;IACH,aAAa,CACX,KAAK,EAAE,iBAAiB,EACxB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GACxC,iBAAiB,EAAE;IAKtB;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAmE9B;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa;IA2B9C;;OAEG;IACH,OAAO,CAAC,aAAa;IA6BrB;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAyB9B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAYzB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyBtB;;OAEG;IACH,mBAAmB,CACjB,KAAK,EAAE,iBAAiB,EACxB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GACxC,iBAAiB,EAAE;IAStB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA6B7B;;OAEG;IACH,YAAY,CACV,KAAK,EAAE,iBAAiB,EACxB,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GACxC,OAAO;IAKV;;OAEG;IACH,eAAe,CACb,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GACxC,iBAAiB,EAAE;CAYvB"}

package/dist/parsers/tree-sitter/pydantic/inheritance-resolver.js

@@ -0,0 +1,251 @@
+/**
+ * Pydantic Inheritance Resolver
+ *
+ * Resolves model inheritance chains and merges fields from parent classes.
+ * Handles multiple inheritance and generic type substitution.
+ *
+ * @module pydantic/inheritance-resolver
+ */
+import { PYDANTIC_BASE_CLASSES } from './types.js';
+// ============================================
+// Inheritance Resolver Class
+// ============================================
+/**
+ * Resolves Pydantic model inheritance.
+ *
+ * Handles:
+ * - Single inheritance: class Child(Parent)
+ * - Multiple inheritance: class Child(Parent1, Parent2)
+ * - Generic inheritance: class Child(Response[User])
+ * - Mixin classes
+ */
+export class InheritanceResolver {
+    /**
+     * Resolve all fields for a model including inherited fields.
+     *
+     * @param model - The model to resolve fields for
+     * @param allModels - Map of all known models by name
+     * @returns Array of all fields (inherited + own)
+     */
+    resolveFields(model, allModels) {
+        const visited = new Set();
+        return this.resolveFieldsRecursive(model, allModels, visited);
+    }
+    /**
+     * Recursively resolve fields from inheritance chain.
+     */
+    resolveFieldsRecursive(model, allModels, visited) {
+        // Prevent infinite recursion
+        if (visited.has(model.name)) {
+            return [];
+        }
+        visited.add(model.name);
+        const inheritedFields = [];
+        // Process base classes in order (MRO)
+        for (const baseName of model.bases) {
+            const baseInfo = this.parseBaseClass(baseName);
+            // Skip Pydantic base classes
+            if (baseInfo.isPydanticBase) {
+                continue;
+            }
+            // Skip Generic[T]
+            if (baseInfo.isGeneric) {
+                continue;
+            }
+            // Find the parent model
+            const parentModel = allModels.get(baseInfo.name);
+            if (!parentModel) {
+                continue;
+            }
+            // Recursively get parent fields
+            const parentFields = this.resolveFieldsRecursive(parentModel, allModels, visited);
+            // Apply generic type substitution if needed
+            const substitutedFields = baseInfo.typeArgs.length > 0
+                ? this.substituteGenericTypes(parentFields, parentModel, baseInfo.typeArgs)
+                : parentFields;
+            // Add parent fields (avoiding duplicates)
+            for (const field of substitutedFields) {
+                if (!inheritedFields.some((f) => f.name === field.name)) {
+                    inheritedFields.push({ ...field });
+                }
+            }
+        }
+        // Add own fields (override inherited)
+        const allFields = [...inheritedFields];
+        for (const field of model.fields) {
+            const existingIndex = allFields.findIndex((f) => f.name === field.name);
+            if (existingIndex >= 0) {
+                allFields[existingIndex] = field;
+            }
+            else {
+                allFields.push(field);
+            }
+        }
+        return allFields;
+    }
+    /**
+     * Parse a base class string into structured info.
+     */
+    parseBaseClass(baseStr) {
+        // Check for generic: Parent[T, U]
+        const genericMatch = baseStr.match(/^(\w+(?:\.\w+)*)\s*\[(.+)\]$/);
+        if (genericMatch) {
+            const [, name, argsStr] = genericMatch;
+            const typeArgs = this.parseTypeArgs(argsStr ?? '');
+            return {
+                name: name ?? '',
+                fullName: null,
+                typeArgs,
+                isPydanticBase: PYDANTIC_BASE_CLASSES.has(name ?? ''),
+                isGeneric: name === 'Generic',
+            };
+        }
+        // Simple base class
+        return {
+            name: baseStr,
+            fullName: null,
+            typeArgs: [],
+            isPydanticBase: PYDANTIC_BASE_CLASSES.has(baseStr),
+            isGeneric: baseStr === 'Generic',
+        };
+    }
+    /**
+     * Parse generic type arguments.
+     */
+    parseTypeArgs(argsStr) {
+        const args = [];
+        let current = '';
+        let depth = 0;
+        for (const char of argsStr) {
+            if (char === '[' || char === '(') {
+                depth++;
+                current += char;
+            }
+            else if (char === ']' || char === ')') {
+                depth--;
+                current += char;
+            }
+            else if (char === ',' && depth === 0) {
+                if (current.trim()) {
+                    args.push(current.trim());
+                }
+                current = '';
+            }
+            else {
+                current += char;
+            }
+        }
+        if (current.trim()) {
+            args.push(current.trim());
+        }
+        return args;
+    }
+    /**
+     * Substitute generic type parameters in fields.
+     */
+    substituteGenericTypes(fields, parentModel, typeArgs) {
+        // Find generic parameters from parent (e.g., T, U from Generic[T, U])
+        const genericParams = this.findGenericParams(parentModel);
+        if (genericParams.length === 0 || genericParams.length !== typeArgs.length) {
+            return fields;
+        }
+        // Create substitution map
+        const substitutions = new Map();
+        for (let i = 0; i < genericParams.length; i++) {
+            substitutions.set(genericParams[i], typeArgs[i]);
+        }
+        // Apply substitutions to fields
+        return fields.map((field) => ({
+            ...field,
+            type: this.substituteType(field.type, substitutions),
+        }));
+    }
+    /**
+     * Find generic type parameters from a model's bases.
+     */
+    findGenericParams(model) {
+        for (const base of model.bases) {
+            if (base.startsWith('Generic[')) {
+                const match = base.match(/^Generic\[(.+)\]$/);
+                if (match) {
+                    return this.parseTypeArgs(match[1] ?? '');
+                }
+            }
+        }
+        return [];
+    }
+    /**
+     * Substitute type parameters in a TypeInfo.
+     */
+    substituteType(type, substitutions) {
+        // Check if base type should be substituted
+        const substituted = substitutions.get(type.base);
+        if (substituted) {
+            return {
+                ...type,
+                base: substituted,
+                raw: type.raw.replace(type.base, substituted),
+            };
+        }
+        // Recursively substitute in type arguments
+        if (type.args.length > 0) {
+            return {
+                ...type,
+                args: type.args.map((arg) => this.substituteType(arg, substitutions)),
+            };
+        }
+        return type;
+    }
+    /**
+     * Get the full inheritance chain for a model.
+     */
+    getInheritanceChain(model, allModels) {
+        const chain = [];
+        const visited = new Set();
+        this.buildInheritanceChain(model, allModels, chain, visited);
+        return chain;
+    }
+    /**
+     * Recursively build inheritance chain.
+     */
+    buildInheritanceChain(model, allModels, chain, visited) {
+        if (visited.has(model.name)) {
+            return;
+        }
+        visited.add(model.name);
+        // Add parents first (depth-first)
+        for (const baseName of model.bases) {
+            const baseInfo = this.parseBaseClass(baseName);
+            if (baseInfo.isPydanticBase || baseInfo.isGeneric) {
+                continue;
+            }
+            const parentModel = allModels.get(baseInfo.name);
+            if (parentModel) {
+                this.buildInheritanceChain(parentModel, allModels, chain, visited);
+            }
+        }
+        // Add this model
+        chain.push(model);
+    }
+    /**
+     * Check if a model inherits from another.
+     */
+    inheritsFrom(model, ancestorName, allModels) {
+        const chain = this.getInheritanceChain(model, allModels);
+        return chain.some((m) => m.name === ancestorName);
+    }
+    /**
+     * Find all models that inherit from a given model.
+     */
+    findDescendants(ancestorName, allModels) {
+        const descendants = [];
+        for (const model of allModels.values()) {
+            if (model.name !== ancestorName &&
+                this.inheritsFrom(model, ancestorName, allModels)) {
+                descendants.push(model);
+            }
+        }
+        return descendants;
+    }
+}
+//# sourceMappingURL=inheritance-resolver.js.map