driftdetect-core 0.1.3 → 0.4.0

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

@@ -0,0 +1,78 @@
+/**
+ * Pydantic Config Extractor
+ *
+ * Extracts model configuration from Pydantic classes.
+ * Handles both v1 (Config class) and v2 (model_config dict).
+ *
+ * @module pydantic/config-extractor
+ */
+import type { TreeSitterNode } from '../types.js';
+import type { PydanticConfigInfo } from '../types.js';
+import type { ExtractionContext } from './types.js';
+/**
+ * Extracts Pydantic model configuration.
+ *
+ * Handles:
+ * - v1: class Config: ...
+ * - v2: model_config = ConfigDict(...)
+ * - v2: model_config = {...}
+ */
+export declare class ConfigExtractor {
+    /**
+     * Extract config from a class body.
+     *
+     * @param bodyNode - The class body (block) node
+     * @param context - Extraction context
+     * @returns Config info or null if no config found
+     */
+    extractConfig(bodyNode: TreeSitterNode, context: ExtractionContext): PydanticConfigInfo | null;
+    /**
+     * Extract v2 style model_config.
+     */
+    private extractV2Config;
+    /**
+     * Parse v2 config value (dict or ConfigDict call).
+     */
+    private parseV2ConfigValue;
+    /**
+     * Parse a dictionary literal config.
+     */
+    private parseDictConfig;
+    /**
+     * Parse a ConfigDict() call.
+     */
+    private parseConfigDictCall;
+    /**
+     * Extract v1 style Config class.
+     */
+    private extractV1Config;
+    /**
+     * Parse v1 Config class body.
+     */
+    private parseV1ConfigClass;
+    /**
+     * Parse a config value string.
+     */
+    private parseConfigValue;
+    /**
+     * Remove quotes from a string.
+     */
+    private unquote;
+    /**
+     * Process raw config data into PydanticConfigInfo.
+     */
+    private processRawConfig;
+    /**
+     * Extract extra config value.
+     */
+    private extractExtraConfig;
+    /**
+     * Extract a boolean config value.
+     */
+    private extractBooleanConfig;
+    /**
+     * Extract json_schema_extra config.
+     */
+    private extractJsonSchemaExtra;
+}
+//# sourceMappingURL=config-extractor.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"config-extractor.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/config-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACtD,OAAO,KAAK,EAAE,iBAAiB,EAA8B,MAAM,YAAY,CAAC;AAMhF;;;;;;;GAOG;AACH,qBAAa,eAAe;IAC1B;;;;;;OAMG;IACH,aAAa,CACX,QAAQ,EAAE,cAAc,EACxB,OAAO,EAAE,iBAAiB,GACzB,kBAAkB,GAAG,IAAI;IAoB5B;;OAEG;IACH,OAAO,CAAC,eAAe;IAuBvB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAqB1B;;OAEG;IACH,OAAO,CAAC,eAAe;IAiBvB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAuB3B;;OAEG;IACH,OAAO,CAAC,eAAe;IAoBvB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA+B1B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA8BxB;;OAEG;IACH,OAAO,CAAC,OAAO;IAaf;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAmBxB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAsB1B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAiB5B;;OAEG;IACH,OAAO,CAAC,sBAAsB;CAa/B"}

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

@@ -0,0 +1,278 @@
+/**
+ * Pydantic Config Extractor
+ *
+ * Extracts model configuration from Pydantic classes.
+ * Handles both v1 (Config class) and v2 (model_config dict).
+ *
+ * @module pydantic/config-extractor
+ */
+// ============================================
+// Config Extractor Class
+// ============================================
+/**
+ * Extracts Pydantic model configuration.
+ *
+ * Handles:
+ * - v1: class Config: ...
+ * - v2: model_config = ConfigDict(...)
+ * - v2: model_config = {...}
+ */
+export class ConfigExtractor {
+    /**
+     * Extract config from a class body.
+     *
+     * @param bodyNode - The class body (block) node
+     * @param context - Extraction context
+     * @returns Config info or null if no config found
+     */
+    extractConfig(bodyNode, context) {
+        // Try v2 style first (model_config)
+        const v2Config = this.extractV2Config(bodyNode, context);
+        if (v2Config) {
+            return v2Config;
+        }
+        // Try v1 style (Config class)
+        const v1Config = this.extractV1Config(bodyNode, context);
+        if (v1Config) {
+            return v1Config;
+        }
+        return null;
+    }
+    // ============================================
+    // V2 Config Extraction
+    // ============================================
+    /**
+     * Extract v2 style model_config.
+     */
+    extractV2Config(bodyNode, _context) {
+        for (const child of bodyNode.namedChildren) {
+            if (child.type !== 'expression_statement')
+                continue;
+            const inner = child.namedChildren[0];
+            if (!inner || inner.type !== 'assignment')
+                continue;
+            const left = inner.childForFieldName('left');
+            if (!left || left.text !== 'model_config')
+                continue;
+            const right = inner.childForFieldName('right');
+            if (!right)
+                continue;
+            const rawData = this.parseV2ConfigValue(right, inner);
+            return this.processRawConfig(rawData);
+        }
+        return null;
+    }
+    /**
+     * Parse v2 config value (dict or ConfigDict call).
+     */
+    parseV2ConfigValue(valueNode, assignmentNode) {
+        const values = new Map();
+        if (valueNode.type === 'dictionary') {
+            // model_config = {...}
+            this.parseDictConfig(valueNode, values);
+        }
+        else if (valueNode.type === 'call') {
+            // model_config = ConfigDict(...)
+            this.parseConfigDictCall(valueNode, values);
+        }
+        return {
+            version: 2,
+            values,
+            node: assignmentNode,
+        };
+    }
+    /**
+     * Parse a dictionary literal config.
+     */
+    parseDictConfig(dictNode, values) {
+        for (const child of dictNode.namedChildren) {
+            if (child.type === 'pair') {
+                const keyNode = child.childForFieldName('key');
+                const valueNode = child.childForFieldName('value');
+                if (keyNode && valueNode) {
+                    const key = this.unquote(keyNode.text);
+                    values.set(key, this.parseConfigValue(valueNode.text));
+                }
+            }
+        }
+    }
+    /**
+     * Parse a ConfigDict() call.
+     */
+    parseConfigDictCall(callNode, values) {
+        const argsNode = callNode.childForFieldName('arguments');
+        if (!argsNode)
+            return;
+        for (const child of argsNode.namedChildren) {
+            if (child.type === 'keyword_argument') {
+                const nameNode = child.childForFieldName('name');
+                const valueNode = child.childForFieldName('value');
+                if (nameNode && valueNode) {
+                    values.set(nameNode.text, this.parseConfigValue(valueNode.text));
+                }
+            }
+        }
+    }
+    // ============================================
+    // V1 Config Extraction
+    // ============================================
+    /**
+     * Extract v1 style Config class.
+     */
+    extractV1Config(bodyNode, _context) {
+        for (const child of bodyNode.namedChildren) {
+            if (child.type !== 'class_definition')
+                continue;
+            const nameNode = child.childForFieldName('name');
+            if (!nameNode || nameNode.text !== 'Config')
+                continue;
+            const configBody = child.childForFieldName('body');
+            if (!configBody)
+                continue;
+            const rawData = this.parseV1ConfigClass(configBody, child);
+            return this.processRawConfig(rawData);
+        }
+        return null;
+    }
+    /**
+     * Parse v1 Config class body.
+     */
+    parseV1ConfigClass(bodyNode, classNode) {
+        const values = new Map();
+        for (const child of bodyNode.namedChildren) {
+            if (child.type !== 'expression_statement')
+                continue;
+            const inner = child.namedChildren[0];
+            if (!inner || inner.type !== 'assignment')
+                continue;
+            const left = inner.childForFieldName('left');
+            const right = inner.childForFieldName('right');
+            if (left && right && left.type === 'identifier') {
+                values.set(left.text, this.parseConfigValue(right.text));
+            }
+        }
+        return {
+            version: 1,
+            values,
+            node: classNode,
+        };
+    }
+    // ============================================
+    // Value Parsing
+    // ============================================
+    /**
+     * Parse a config value string.
+     */
+    parseConfigValue(valueStr) {
+        const trimmed = valueStr.trim();
+        // Boolean
+        if (trimmed === 'True') {
+            return { raw: trimmed, parsed: true, isComplex: false };
+        }
+        if (trimmed === 'False') {
+            return { raw: trimmed, parsed: false, isComplex: false };
+        }
+        if (trimmed === 'None') {
+            return { raw: trimmed, parsed: null, isComplex: false };
+        }
+        // Number
+        const num = parseFloat(trimmed);
+        if (!isNaN(num) && /^-?\d+\.?\d*$/.test(trimmed)) {
+            return { raw: trimmed, parsed: num, isComplex: false };
+        }
+        // String
+        if ((trimmed.startsWith('"') && trimmed.endsWith('"')) ||
+            (trimmed.startsWith("'") && trimmed.endsWith("'"))) {
+            return { raw: trimmed, parsed: trimmed.slice(1, -1), isComplex: false };
+        }
+        // Complex expression
+        return { raw: trimmed, parsed: null, isComplex: true };
+    }
+    /**
+     * Remove quotes from a string.
+     */
+    unquote(value) {
+        const trimmed = value.trim();
+        if ((trimmed.startsWith('"') && trimmed.endsWith('"')) ||
+            (trimmed.startsWith("'") && trimmed.endsWith("'"))) {
+            return trimmed.slice(1, -1);
+        }
+        return trimmed;
+    }
+    // ============================================
+    // Processing
+    // ============================================
+    /**
+     * Process raw config data into PydanticConfigInfo.
+     */
+    processRawConfig(raw) {
+        const rawValues = {};
+        for (const [key, value] of raw.values) {
+            rawValues[key] = value.isComplex ? value.raw : value.parsed;
+        }
+        return {
+            extra: this.extractExtraConfig(raw.values),
+            frozen: this.extractBooleanConfig(raw.values, ['frozen', 'allow_mutation']),
+            validateAssignment: this.extractBooleanConfig(raw.values, ['validate_assignment']),
+            useEnumValues: this.extractBooleanConfig(raw.values, ['use_enum_values']),
+            validateDefault: this.extractBooleanConfig(raw.values, ['validate_default']),
+            populateByName: this.extractBooleanConfig(raw.values, ['populate_by_name', 'allow_population_by_field_name']),
+            arbitraryTypesAllowed: this.extractBooleanConfig(raw.values, ['arbitrary_types_allowed']),
+            jsonSchemaExtra: this.extractJsonSchemaExtra(raw.values),
+            raw: rawValues,
+        };
+    }
+    /**
+     * Extract extra config value.
+     */
+    extractExtraConfig(values) {
+        const value = values.get('extra');
+        if (!value)
+            return null;
+        const parsed = value.parsed;
+        if (typeof parsed === 'string') {
+            if (parsed === 'allow' || parsed === 'forbid' || parsed === 'ignore') {
+                return parsed;
+            }
+        }
+        // Handle v1 style: Extra.allow, Extra.forbid, Extra.ignore
+        const raw = value.raw;
+        if (raw.includes('allow'))
+            return 'allow';
+        if (raw.includes('forbid'))
+            return 'forbid';
+        if (raw.includes('ignore'))
+            return 'ignore';
+        return null;
+    }
+    /**
+     * Extract a boolean config value.
+     */
+    extractBooleanConfig(values, keys) {
+        for (const key of keys) {
+            const value = values.get(key);
+            if (value && typeof value.parsed === 'boolean') {
+                // Handle inverted keys (allow_mutation -> frozen)
+                if (key === 'allow_mutation') {
+                    return !value.parsed;
+                }
+                return value.parsed;
+            }
+        }
+        return null;
+    }
+    /**
+     * Extract json_schema_extra config.
+     */
+    extractJsonSchemaExtra(values) {
+        const value = values.get('json_schema_extra') || values.get('schema_extra');
+        if (!value)
+            return null;
+        // If it's a simple dict, try to parse it
+        if (!value.isComplex && typeof value.parsed === 'object' && value.parsed !== null) {
+            return value.parsed;
+        }
+        return null;
+    }
+}
+//# sourceMappingURL=config-extractor.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"config-extractor.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/config-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAMH,+CAA+C;AAC/C,yBAAyB;AACzB,+CAA+C;AAE/C;;;;;;;GAOG;AACH,MAAM,OAAO,eAAe;IAC1B;;;;;;OAMG;IACH,aAAa,CACX,QAAwB,EACxB,OAA0B;QAE1B,oCAAoC;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACzD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,8BAA8B;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACzD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,+CAA+C;IAC/C,uBAAuB;IACvB,+CAA+C;IAE/C;;OAEG;IACK,eAAe,CACrB,QAAwB,EACxB,QAA2B;QAE3B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB;gBAAE,SAAS;YAEpD,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;gBAAE,SAAS;YAEpD,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,cAAc;gBAAE,SAAS;YAEpD,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,CAAC,KAAK;gBAAE,SAAS;YAErB,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,SAAyB,EACzB,cAA8B;QAE9B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;QAE9C,IAAI,SAAS,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACpC,uBAAuB;YACvB,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACrC,iCAAiC;YACjC,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO;YACL,OAAO,EAAE,CAAC;YACV,MAAM;YACN,IAAI,EAAE,cAAc;SACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,eAAe,CACrB,QAAwB,EACxB,MAAgC;QAEhC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBAC/C,MAAM,SAAS,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBAEnD,IAAI,OAAO,IAAI,SAAS,EAAE,CAAC;oBACzB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBACvC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CACzB,QAAwB,EACxB,MAAgC;QAEhC,MAAM,QAAQ,GAAG,QAAQ,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,QAAQ;YAAE,OAAO;QAEtB,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;gBAEnD,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;oBAC1B,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,+CAA+C;IAC/C,uBAAuB;IACvB,+CAA+C;IAE/C;;OAEG;IACK,eAAe,CACrB,QAAwB,EACxB,QAA2B;QAE3B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB;gBAAE,SAAS;YAEhD,MAAM,QAAQ,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACjD,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ;gBAAE,SAAS;YAEtD,MAAM,UAAU,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU;gBAAE,SAAS;YAE1B,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC3D,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,QAAwB,EACxB,SAAyB;QAEzB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;QAE9C,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB;gBAAE,SAAS;YAEpD,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;gBAAE,SAAS;YAEpD,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE/C,IAAI,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChD,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,CAAC;YACV,MAAM;YACN,IAAI,EAAE,SAAS;SAChB,CAAC;IACJ,CAAC;IAED,+CAA+C;IAC/C,gBAAgB;IAChB,+CAA+C;IAE/C;;OAEG;IACK,gBAAgB,CAAC,QAAgB;QACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEhC,UAAU;QACV,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;YACvB,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC1D,CAAC;QACD,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;YACxB,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC3D,CAAC;QACD,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;YACvB,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC1D,CAAC;QAED,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,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QACzD,CAAC;QAED,SAAS;QACT,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,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC1E,CAAC;QAED,qBAAqB;QACrB,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IACzD,CAAC;IAED;;OAEG;IACK,OAAO,CAAC,KAAa;QAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAC7B,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;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+CAA+C;IAC/C,aAAa;IACb,+CAA+C;IAE/C;;OAEG;IACK,gBAAgB,CAAC,GAAkB;QACzC,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;YACtC,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC;QAC9D,CAAC;QAED,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1C,MAAM,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;YAC3E,kBAAkB,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,qBAAqB,CAAC,CAAC;YAClF,aAAa,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,iBAAiB,CAAC,CAAC;YACzE,eAAe,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,kBAAkB,CAAC,CAAC;YAC5E,cAAc,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,kBAAkB,EAAE,gCAAgC,CAAC,CAAC;YAC7G,qBAAqB,EAAE,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,yBAAyB,CAAC,CAAC;YACzF,eAAe,EAAE,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC;YACxD,GAAG,EAAE,SAAS;SACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,MAAgC;QAEhC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,IAAI,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;gBACrE,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAED,2DAA2D;QAC3D,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;QACtB,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QAC1C,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC5C,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;QAE5C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,MAAgC,EAChC,IAAc;QAEd,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC9B,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC/C,kDAAkD;gBAClD,IAAI,GAAG,KAAK,gBAAgB,EAAE,CAAC;oBAC7B,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvB,CAAC;gBACD,OAAO,KAAK,CAAC,MAAM,CAAC;YACtB,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,MAAgC;QAEhC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC5E,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,yCAAyC;QACzC,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YAClF,OAAO,KAAK,CAAC,MAAiC,CAAC;QACjD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/parsers/tree-sitter/pydantic/constraint-parser.d.ts

@@ -0,0 +1,84 @@
+/**
+ * Pydantic Constraint Parser
+ *
+ * Parses Field() constraints from function call arguments.
+ * Handles numeric, string, and custom constraints.
+ *
+ * @module pydantic/constraint-parser
+ */
+import type { FieldConstraints } from '../types.js';
+import type { FieldArgument } from './types.js';
+/**
+ * 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 declare class ConstraintParser {
+    /**
+     * Parse constraints from Field() arguments.
+     *
+     * @param args - Field() call arguments
+     * @returns Parsed constraints
+     */
+    parseConstraints(args: FieldArgument[]): FieldConstraints;
+    /**
+     * Parse a single constraint argument.
+     */
+    private parseConstraintArg;
+    /**
+     * Parse a numeric value.
+     */
+    private parseNumber;
+    /**
+     * Parse a string value (remove quotes).
+     */
+    private parseString;
+    /**
+     * Parse a boolean value.
+     */
+    private parseBoolean;
+    /**
+     * Parse json_schema_extra value.
+     */
+    private parseJsonSchemaExtra;
+    /**
+     * Parse a simple Python dict literal.
+     */
+    private parseSimpleDict;
+    /**
+     * Split dict pairs respecting nested structures.
+     */
+    private splitDictPairs;
+    /**
+     * Parse a generic Python value.
+     */
+    private parseValue;
+    /**
+     * Parse a simple Python list literal.
+     */
+    private parseSimpleList;
+    /**
+     * Parse validator references.
+     */
+    private parseValidators;
+}
+/**
+ * Check if constraints have any numeric constraints.
+ */
+export declare function hasNumericConstraints(constraints: FieldConstraints): boolean;
+/**
+ * Check if constraints have any string constraints.
+ */
+export declare function hasStringConstraints(constraints: FieldConstraints): boolean;
+/**
+ * Check if constraints are empty.
+ */
+export declare function hasAnyConstraints(constraints: FieldConstraints): boolean;
+/**
+ * Merge two constraint objects.
+ */
+export declare function mergeConstraints(base: FieldConstraints, override: Partial<FieldConstraints>): FieldConstraints;
+//# sourceMappingURL=constraint-parser.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"constraint-parser.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/constraint-parser.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAMhD;;;;;;;GAOG;AACH,qBAAa,gBAAgB;IAC3B;;;;;OAKG;IACH,gBAAgB,CAAC,IAAI,EAAE,aAAa,EAAE,GAAG,gBAAgB;IAwBzD;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA0D1B;;OAEG;IACH,OAAO,CAAC,WAAW;IAYnB;;OAEG;IACH,OAAO,CAAC,WAAW;IAsBnB;;OAEG;IACH,OAAO,CAAC,YAAY;IAKpB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAc5B;;OAEG;IACH,OAAO,CAAC,eAAe;IAuBvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyCtB;;OAEG;IACH,OAAO,CAAC,UAAU;IAkClB;;OAEG;IACH,OAAO,CAAC,eAAe;IAQvB;;OAEG;IACH,OAAO,CAAC,eAAe;CAQxB;AAMD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,gBAAgB,GAAG,OAAO,CAQ5E;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,WAAW,EAAE,gBAAgB,GAAG,OAAO,CAM3E;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,WAAW,EAAE,gBAAgB,GAAG,OAAO,CAQxE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,IAAI,EAAE,gBAAgB,EACtB,QAAQ,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAClC,gBAAgB,CAclB"}