driftdetect-core 0.1.3 → 0.4.0

package/dist/parsers/tree-sitter/pydantic/constraint-parser.js

@@ -0,0 +1,321 @@
+/**
+ * Pydantic Constraint Parser
+ *
+ * Parses Field() constraints from function call arguments.
+ * Handles numeric, string, and custom constraints.
+ *
+ * @module pydantic/constraint-parser
+ */
+// ============================================
+// Constraint Parser Class
+// ============================================
+/**
+ * Parses Pydantic Field() constraints.
+ *
+ * Handles:
+ * - Numeric: ge, le, gt, lt, multiple_of
+ * - String: min_length, max_length, pattern, regex
+ * - Other: frozen, alias, title, description
+ */
+export class ConstraintParser {
+    /**
+     * Parse constraints from Field() arguments.
+     *
+     * @param args - Field() call arguments
+     * @returns Parsed constraints
+     */
+    parseConstraints(args) {
+        const constraints = {
+            ge: null,
+            le: null,
+            gt: null,
+            lt: null,
+            minLength: null,
+            maxLength: null,
+            pattern: null,
+            multipleOf: null,
+            frozen: false,
+            validators: [],
+            jsonSchemaExtra: null,
+        };
+        for (const arg of args) {
+            if (!arg.isKeyword || !arg.name)
+                continue;
+            this.parseConstraintArg(arg.name, arg.value, constraints);
+        }
+        return constraints;
+    }
+    /**
+     * Parse a single constraint argument.
+     */
+    parseConstraintArg(name, value, constraints) {
+        switch (name) {
+            // Numeric constraints
+            case 'ge':
+                constraints.ge = this.parseNumber(value);
+                break;
+            case 'le':
+                constraints.le = this.parseNumber(value);
+                break;
+            case 'gt':
+                constraints.gt = this.parseNumber(value);
+                break;
+            case 'lt':
+                constraints.lt = this.parseNumber(value);
+                break;
+            case 'multiple_of':
+                constraints.multipleOf = this.parseNumber(value);
+                break;
+            // String constraints
+            case 'min_length':
+                constraints.minLength = this.parseNumber(value);
+                break;
+            case 'max_length':
+                constraints.maxLength = this.parseNumber(value);
+                break;
+            case 'pattern':
+            case 'regex':
+                constraints.pattern = this.parseString(value);
+                break;
+            // Boolean constraints
+            case 'frozen':
+                constraints.frozen = this.parseBoolean(value);
+                break;
+            // JSON schema extra
+            case 'json_schema_extra':
+                constraints.jsonSchemaExtra = this.parseJsonSchemaExtra(value);
+                break;
+            // Validators (custom validation functions)
+            case 'validator':
+            case 'validators':
+                const validators = this.parseValidators(value);
+                constraints.validators.push(...validators);
+                break;
+        }
+    }
+    // ============================================
+    // Value Parsers
+    // ============================================
+    /**
+     * Parse a numeric value.
+     */
+    parseNumber(value) {
+        const trimmed = value.trim();
+        // Handle negative numbers
+        const num = parseFloat(trimmed);
+        if (!isNaN(num)) {
+            return num;
+        }
+        return null;
+    }
+    /**
+     * Parse a string value (remove quotes).
+     */
+    parseString(value) {
+        const trimmed = value.trim();
+        // Handle raw strings: r'pattern' or r"pattern"
+        if (trimmed.startsWith('r"') || trimmed.startsWith("r'")) {
+            return trimmed.slice(2, -1);
+        }
+        // Handle regular strings
+        if ((trimmed.startsWith('"') && trimmed.endsWith('"')) ||
+            (trimmed.startsWith("'") && trimmed.endsWith("'"))) {
+            return trimmed.slice(1, -1);
+        }
+        // Handle triple-quoted strings
+        if (trimmed.startsWith('"""') || trimmed.startsWith("'''")) {
+            return trimmed.slice(3, -3);
+        }
+        return trimmed;
+    }
+    /**
+     * Parse a boolean value.
+     */
+    parseBoolean(value) {
+        const trimmed = value.trim().toLowerCase();
+        return trimmed === 'true';
+    }
+    /**
+     * Parse json_schema_extra value.
+     */
+    parseJsonSchemaExtra(value) {
+        // This is typically a dict literal or function call
+        // For now, we'll store the raw value and mark it as complex
+        try {
+            // Simple dict parsing for common cases
+            if (value.startsWith('{') && value.endsWith('}')) {
+                return this.parseSimpleDict(value);
+            }
+        }
+        catch {
+            // Fall through to return null
+        }
+        return null;
+    }
+    /**
+     * Parse a simple Python dict literal.
+     */
+    parseSimpleDict(value) {
+        // Remove braces
+        const inner = value.slice(1, -1).trim();
+        if (!inner)
+            return {};
+        const result = {};
+        const pairs = this.splitDictPairs(inner);
+        for (const pair of pairs) {
+            const colonIndex = pair.indexOf(':');
+            if (colonIndex === -1)
+                continue;
+            const key = this.parseString(pair.slice(0, colonIndex).trim());
+            const val = pair.slice(colonIndex + 1).trim();
+            if (key) {
+                result[key] = this.parseValue(val);
+            }
+        }
+        return result;
+    }
+    /**
+     * Split dict pairs respecting nested structures.
+     */
+    splitDictPairs(inner) {
+        const pairs = [];
+        let current = '';
+        let depth = 0;
+        let inString = false;
+        let stringChar = '';
+        for (let i = 0; i < inner.length; i++) {
+            const char = inner[i];
+            const prevChar = inner[i - 1];
+            // Handle string boundaries
+            if ((char === '"' || char === "'") && prevChar !== '\\') {
+                if (!inString) {
+                    inString = true;
+                    stringChar = char;
+                }
+                else if (char === stringChar) {
+                    inString = false;
+                }
+            }
+            if (!inString) {
+                if (char === '{' || char === '[' || char === '(')
+                    depth++;
+                else if (char === '}' || char === ']' || char === ')')
+                    depth--;
+                else if (char === ',' && depth === 0) {
+                    pairs.push(current.trim());
+                    current = '';
+                    continue;
+                }
+            }
+            current += char;
+        }
+        if (current.trim()) {
+            pairs.push(current.trim());
+        }
+        return pairs;
+    }
+    /**
+     * Parse a generic Python value.
+     */
+    parseValue(value) {
+        const trimmed = value.trim();
+        // Boolean
+        if (trimmed === 'True')
+            return true;
+        if (trimmed === 'False')
+            return false;
+        if (trimmed === 'None')
+            return null;
+        // Number
+        const num = parseFloat(trimmed);
+        if (!isNaN(num) && /^-?\d+\.?\d*$/.test(trimmed)) {
+            return num;
+        }
+        // String
+        const str = this.parseString(trimmed);
+        if (str !== trimmed) {
+            return str;
+        }
+        // List
+        if (trimmed.startsWith('[') && trimmed.endsWith(']')) {
+            return this.parseSimpleList(trimmed);
+        }
+        // Dict
+        if (trimmed.startsWith('{') && trimmed.endsWith('}')) {
+            return this.parseSimpleDict(trimmed);
+        }
+        // Return as-is for complex expressions
+        return trimmed;
+    }
+    /**
+     * Parse a simple Python list literal.
+     */
+    parseSimpleList(value) {
+        const inner = value.slice(1, -1).trim();
+        if (!inner)
+            return [];
+        const items = this.splitDictPairs(inner); // Same splitting logic works
+        return items.map((item) => this.parseValue(item));
+    }
+    /**
+     * Parse validator references.
+     */
+    parseValidators(value) {
+        // Could be a single validator or list
+        if (value.startsWith('[') && value.endsWith(']')) {
+            const inner = value.slice(1, -1);
+            return inner.split(',').map((v) => v.trim()).filter(Boolean);
+        }
+        return [value.trim()];
+    }
+}
+// ============================================
+// Constraint Utilities
+// ============================================
+/**
+ * Check if constraints have any numeric constraints.
+ */
+export function hasNumericConstraints(constraints) {
+    return (constraints.ge !== null ||
+        constraints.le !== null ||
+        constraints.gt !== null ||
+        constraints.lt !== null ||
+        constraints.multipleOf !== null);
+}
+/**
+ * Check if constraints have any string constraints.
+ */
+export function hasStringConstraints(constraints) {
+    return (constraints.minLength !== null ||
+        constraints.maxLength !== null ||
+        constraints.pattern !== null);
+}
+/**
+ * Check if constraints are empty.
+ */
+export function hasAnyConstraints(constraints) {
+    return (hasNumericConstraints(constraints) ||
+        hasStringConstraints(constraints) ||
+        constraints.frozen ||
+        constraints.validators.length > 0 ||
+        constraints.jsonSchemaExtra !== null);
+}
+/**
+ * Merge two constraint objects.
+ */
+export function mergeConstraints(base, override) {
+    return {
+        ge: override.ge ?? base.ge,
+        le: override.le ?? base.le,
+        gt: override.gt ?? base.gt,
+        lt: override.lt ?? base.lt,
+        minLength: override.minLength ?? base.minLength,
+        maxLength: override.maxLength ?? base.maxLength,
+        pattern: override.pattern ?? base.pattern,
+        multipleOf: override.multipleOf ?? base.multipleOf,
+        frozen: override.frozen ?? base.frozen,
+        validators: [...base.validators, ...(override.validators ?? [])],
+        jsonSchemaExtra: override.jsonSchemaExtra ?? base.jsonSchemaExtra,
+    };
+}
+//# sourceMappingURL=constraint-parser.js.map

package/dist/parsers/tree-sitter/pydantic/constraint-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constraint-parser.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/constraint-parser.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,+CAA+C;AAC/C,0BAA0B;AAC1B,+CAA+C;AAE/C;;;;;;;GAOG;AACH,MAAM,OAAO,gBAAgB;IAC3B;;;;;OAKG;IACH,gBAAgB,CAAC,IAAqB;QACpC,MAAM,WAAW,GAAqB;YACpC,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;QAEF,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC,GAAG,CAAC,SAAS,IAAI,CAAC,GAAG,CAAC,IAAI;gBAAE,SAAS;YAE1C,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,IAAY,EACZ,KAAa,EACb,WAA6B;QAE7B,QAAQ,IAAI,EAAE,CAAC;YACb,sBAAsB;YACtB,KAAK,IAAI;gBACP,WAAW,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,IAAI;gBACP,WAAW,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,IAAI;gBACP,WAAW,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,IAAI;gBACP,WAAW,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACzC,MAAM;YACR,KAAK,aAAa;gBAChB,WAAW,CAAC,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBACjD,MAAM;YAER,qBAAqB;YACrB,KAAK,YAAY;gBACf,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM;YACR,KAAK,YAAY;gBACf,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAChD,MAAM;YACR,KAAK,SAAS,CAAC;YACf,KAAK,OAAO;gBACV,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;gBAC9C,MAAM;YAER,sBAAsB;YACtB,KAAK,QAAQ;gBACX,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC9C,MAAM;YAER,oBAAoB;YACpB,KAAK,mBAAmB;gBACtB,WAAW,CAAC,eAAe,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBAC/D,MAAM;YAER,2CAA2C;YAC3C,KAAK,WAAW,CAAC;YACjB,KAAK,YAAY;gBACf,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC/C,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;gBAC3C,MAAM;QACV,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,gBAAgB;IAChB,+CAA+C;IAE/C;;OAEG;IACK,WAAW,CAAC,KAAa;QAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAE7B,0BAA0B;QAC1B,MAAM,GAAG,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YAChB,OAAO,GAAG,CAAC;QACb,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAa;QAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAE7B,+CAA+C;QAC/C,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACzD,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC;QAED,yBAAyB;QACzB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAClD,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YACvD,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC;QAED,+BAA+B;QAC/B,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3D,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,KAAa;QAChC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,OAAO,OAAO,KAAK,MAAM,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,KAAa;QACxC,oDAAoD;QACpD,4DAA4D;QAC5D,IAAI,CAAC;YACH,uCAAuC;YACvC,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACjD,OAAO,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8BAA8B;QAChC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,KAAa;QACnC,gBAAgB;QAChB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,OAAO,EAAE,CAAC;QAEtB,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACrC,IAAI,UAAU,KAAK,CAAC,CAAC;gBAAE,SAAS;YAEhC,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAE9C,IAAI,GAAG,EAAE,CAAC;gBACR,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,KAAa;QAClC,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,UAAU,GAAG,EAAE,CAAC;QAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAE9B,2BAA2B;YAC3B,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;gBACxD,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,GAAG,IAAI,CAAC;oBAChB,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;qBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC/B,QAAQ,GAAG,KAAK,CAAC;gBACnB,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;oBAAE,KAAK,EAAE,CAAC;qBACrD,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;oBAAE,KAAK,EAAE,CAAC;qBAC1D,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;oBACrC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO,GAAG,EAAE,CAAC;oBACb,SAAS;gBACX,CAAC;YACH,CAAC;YAED,OAAO,IAAI,IAAI,CAAC;QAClB,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,KAAa;QAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAE7B,UAAU;QACV,IAAI,OAAO,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC;QACpC,IAAI,OAAO,KAAK,OAAO;YAAE,OAAO,KAAK,CAAC;QACtC,IAAI,OAAO,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC;QAEpC,SAAS;QACT,MAAM,GAAG,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACjD,OAAO,GAAG,CAAC;QACb,CAAC;QAED,SAAS;QACT,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;YACpB,OAAO,GAAG,CAAC;QACb,CAAC;QAED,OAAO;QACP,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACrD,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAED,OAAO;QACP,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACrD,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAED,uCAAuC;QACvC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,KAAa;QACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,OAAO,EAAE,CAAC;QAEtB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,6BAA6B;QACvE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,KAAa;QACnC,sCAAsC;QACtC,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACjC,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/D,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACxB,CAAC;CACF;AAED,+CAA+C;AAC/C,uBAAuB;AACvB,+CAA+C;AAE/C;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,WAA6B;IACjE,OAAO,CACL,WAAW,CAAC,EAAE,KAAK,IAAI;QACvB,WAAW,CAAC,EAAE,KAAK,IAAI;QACvB,WAAW,CAAC,EAAE,KAAK,IAAI;QACvB,WAAW,CAAC,EAAE,KAAK,IAAI;QACvB,WAAW,CAAC,UAAU,KAAK,IAAI,CAChC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,WAA6B;IAChE,OAAO,CACL,WAAW,CAAC,SAAS,KAAK,IAAI;QAC9B,WAAW,CAAC,SAAS,KAAK,IAAI;QAC9B,WAAW,CAAC,OAAO,KAAK,IAAI,CAC7B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,WAA6B;IAC7D,OAAO,CACL,qBAAqB,CAAC,WAAW,CAAC;QAClC,oBAAoB,CAAC,WAAW,CAAC;QACjC,WAAW,CAAC,MAAM;QAClB,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;QACjC,WAAW,CAAC,eAAe,KAAK,IAAI,CACrC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAsB,EACtB,QAAmC;IAEnC,OAAO;QACL,EAAE,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE;QAC1B,EAAE,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE;QAC1B,EAAE,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE;QAC1B,EAAE,EAAE,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE;QAC1B,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;QAC/C,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS;QAC/C,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;QACzC,UAAU,EAAE,QAAQ,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU;QAClD,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;QACtC,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;QAChE,eAAe,EAAE,QAAQ,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe;KAClE,CAAC;AACJ,CAAC"}

package/dist/parsers/tree-sitter/pydantic/field-extractor.d.ts

@@ -0,0 +1,74 @@
+/**
+ * Pydantic Field Extractor
+ *
+ * Extracts field definitions from Pydantic model class bodies.
+ * Handles annotated assignments, Field() calls, and default values.
+ *
+ * @module pydantic/field-extractor
+ */
+import type { TreeSitterNode } from '../types.js';
+import type { PydanticFieldInfo } from '../types.js';
+import type { ExtractionContext } from './types.js';
+/**
+ * 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 declare class FieldExtractor {
+    private readonly typeResolver;
+    private readonly constraintParser;
+    constructor();
+    /**
+     * 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: TreeSitterNode, context: ExtractionContext): PydanticFieldInfo[];
+    /**
+     * 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: TreeSitterNode, context: ExtractionContext): PydanticFieldInfo | null;
+    /**
+     * Extract raw field data from an assignment node.
+     */
+    private extractRawFieldData;
+    /**
+     * Parse a Field() call and extract arguments.
+     */
+    private parseFieldCall;
+    /**
+     * Extract arguments from a call's argument_list.
+     */
+    private extractCallArguments;
+    /**
+     * Process raw field data into PydanticFieldInfo.
+     */
+    private processRawField;
+    /**
+     * Determine if a field is required.
+     */
+    private isFieldRequired;
+    /**
+     * Extract the actual default value (not Field() call).
+     */
+    private extractDefaultValue;
+    /**
+     * Extract a string constraint value.
+     */
+    private extractStringConstraint;
+    /**
+     * Create empty constraints object.
+     */
+    private createEmptyConstraints;
+}
+//# sourceMappingURL=field-extractor.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"field-extractor.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/field-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,iBAAiB,EAAoB,MAAM,aAAa,CAAC;AACvE,OAAO,KAAK,EAAE,iBAAiB,EAA+B,MAAM,YAAY,CAAC;AASjF;;;;;;;;GAQG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAC5C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAmB;;IAOpD;;;;;;OAMG;IACH,aAAa,CACX,QAAQ,EAAE,cAAc,EACxB,OAAO,EAAE,iBAAiB,GACzB,iBAAiB,EAAE;IAatB;;;;;;OAMG;IACH,YAAY,CACV,IAAI,EAAE,cAAc,EACpB,OAAO,EAAE,iBAAiB,GACzB,iBAAiB,GAAG,IAAI;IAyB3B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAyC3B;;OAEG;IACH,OAAO,CAAC,cAAc;IA+BtB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAiC5B;;OAEG;IACH,OAAO,CAAC,eAAe;IA2CvB;;OAEG;IACH,OAAO,CAAC,eAAe;IAsBvB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAoB3B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAgB/B;;OAEG;IACH,OAAO,CAAC,sBAAsB;CAe/B"}