@an5/orm 1.0.8 → 1.1.0

package/dist/generator/src/python-generator.js

@@ -1,105 +1,428 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PythonGenerator = void 0;
-const fs_1 = __importDefault(require("fs"));
-class PythonGenerator {
-    constructor(outputPath) {
-        this.outputPath = outputPath;
-    }
-    generate(models) {
-        let pyContent = '# This file is auto-generated. Do not edit directly.\n\n';
-        // 1. MODEL_TO_TABLE
-        pyContent += 'MODEL_TO_TABLE = {\n';
-        for (const model of models) {
-            const props = this.getAllPropertyVariations(model.name);
-            const fullTableName = `[${model.schemaName}].[${model.tableName}]`;
-            for (const prop of props) {
-                pyContent += `    "${prop}": "${fullTableName}",\n`;
-            }
-        }
-        pyContent += '}\n\n';
-        // 2. MODEL_DESCRIPTIONS
-        pyContent += 'MODEL_DESCRIPTIONS = {\n';
-        for (const model of models) {
-            const props = this.getAllPropertyVariations(model.name);
-            for (const prop of props) {
-                pyContent += `    "${prop}": ${this.pyString(model.description)},\n`;
-            }
-        }
-        pyContent += '}\n\n';
-        // 3. MODEL_FIELDS
-        pyContent += 'MODEL_FIELDS = {\n';
-        for (const model of models) {
-            const props = this.getAllPropertyVariations(model.name);
-            const fieldsStr = this.formatFields(model);
-            for (const prop of props) {
-                pyContent += `    "${prop}": ${fieldsStr},\n`;
-            }
-        }
-        pyContent += '}\n\n';
-        // 4. RELATION_MAP
-        pyContent += 'RELATION_MAP = {\n';
-        for (const model of models) {
-            const props = this.getAllPropertyVariations(model.name);
-            let relationsContent = `{\n`;
-            for (const rel of model.relations) {
-                relationsContent += `        "${rel.name}": {\n`;
-                relationsContent += `            "modelName": "${this.toCamelCase(rel.type)}",\n`;
-                relationsContent += `            "relationType": "${rel.isArray ? 'many' : 'one'}",\n`;
-                relationsContent += `            "foreignKey": "${rel.foreignKey || 'id'}",\n`;
-                relationsContent += `            "localKey": "${rel.localKey || 'id'}"\n`;
-                relationsContent += `        },\n`;
-            }
-            relationsContent += `    }`;
-            for (const prop of props) {
-                pyContent += `    "${prop}": ${relationsContent},\n`;
-            }
-        }
-        pyContent += '}\n';
-        fs_1.default.writeFileSync(this.outputPath, pyContent);
-    }
-    getAllPropertyVariations(modelName) {
-        const variations = new Set();
-        // 1. camelCase (User -> user, McpServer -> mcpServer)
-        variations.add(this.toCamelCase(modelName));
-        // 2. snake_case (User -> user, McpServer -> mcp_server)
-        variations.add(this.toSnakeCase(modelName));
-        const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
-        for (const acronym of acronyms) {
-            if (modelName.startsWith(acronym)) {
-                variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
-                variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
-                variations.add(this.toSnakeCase(acronym.toLowerCase() + modelName.slice(acronym.length)));
-            }
-        }
-        return Array.from(variations);
-    }
-    formatFields(model) {
-        if (model.fields.length === 0)
-            return '[]';
-        const fields = model.fields.map(f => {
-            return `        { "name": "${f.name}", "type": "${f.type}${f.isOptional ? '?' : ''}", "sql": "${f.sqlType}", "isOptional": ${f.isOptional ? 'True' : 'False'}, "hasDefault": ${f.hasDefault ? 'True' : 'False'}, "isId": ${f.isId ? 'True' : 'False'}, "description": ${this.pyString(f.description)} }`;
-        });
-        return `[\n${fields.join(',\n')}\n    ]`;
-    }
-    pyString(value) {
-        return value ? JSON.stringify(value) : 'None';
-    }
-    toCamelCase(str) {
-        if (!str)
-            return '';
-        return str[0].toLowerCase() + str.slice(1);
-    }
-    toSnakeCase(str) {
-        if (!str)
-            return '';
-        return str
-            .replace(/([A-Z]+)([A-Z][a-z])/g, '$1_$2')
-            .replace(/([a-z\d])([A-Z])/g, '$1_$2')
-            .toLowerCase();
-    }
-}
-exports.PythonGenerator = PythonGenerator;
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PythonGenerator = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const types_1 = require("./types");
+const type_kinds_1 = require("./type-kinds");
+class PythonGenerator {
+    constructor(outputPath) {
+        this.outputPath = outputPath;
+    }
+    /**
+     * The metadata module name, taken from the configured file name itself.
+     *
+     * The imports used to hardcode `an5_metadata` while `outputPath` was
+     * configurable, so setting `python.metadataFile` to another name (say
+     * `an5Metadata.py`) still generated `from an5_metadata import ...` and the
+     * client broke on import — a failure that only showed up at runtime.
+     */
+    metadataModule() {
+        return path_1.default.basename(this.outputPath).replace(/\.py$/i, '');
+    }
+    generate(models) {
+        this.provider = models[0]?.provider;
+        const outputDir = path_1.default.dirname(this.outputPath);
+        if (!fs_1.default.existsSync(outputDir)) {
+            fs_1.default.mkdirSync(outputDir, { recursive: true });
+        }
+        // 1. Generate an5_metadata.py
+        this.generateMetadata(models);
+        // 2. Generate per-model entity files like dotnet (<Model>.py)
+        for (const model of models) {
+            this.generateModelFile(model, outputDir);
+        }
+        // 3. Generate an5_models.py as backward-compat aggregator
+        this.generateModelsIndex(models, outputDir);
+        // 4. Generate an5_orm_types.py - type-safe ORM filter/args dataclasses
+        this.generateOrmTypes(models, outputDir);
+        // 5. Generate an5_client.py
+        this.generateClient(models, outputDir);
+        // 6. Generate __init__.py
+        this.generateInit(models, outputDir);
+    }
+    getPyFilterType(field) {
+        switch ((0, type_kinds_1.fieldKind)(field, this.provider)) {
+            case 'date': return 'DateTimeFilter';
+            case 'bool': return 'BoolFilter';
+            case 'int':
+            case 'bigint': return 'IntFilter';
+            case 'float': return 'NumberFilter';
+            default: return 'StringFilter';
+        }
+    }
+    generateOrmTypes(models, outputDir) {
+        let content = '# This file is auto-generated. Do not edit directly.\n';
+        content += '"""\nAN5 ORM typed filter/args dataclasses for type-safe queries.\n\nUsage example::\n\n    db.user.find_many(where=UserWhereInput(name=StringFilter(contains="John")),\n                      order_by=UserOrderByInput(created_at="desc"),\n                      take=10)\n"""\n';
+        content += 'from __future__ import annotations\n';
+        content += 'from dataclasses import dataclass, field\n';
+        content += 'from typing import List, Optional, Any\n';
+        content += 'from datetime import datetime\n\n';
+        // Base filter types
+        content += '# ─── Base filter types ────────────────────────────────────────────────────────\n\n';
+        content += `@dataclass\nclass StringFilter:\n    """Type-safe filter for string fields."""\n    equals: Optional[str] = None\n    not_: Optional[str] = None\n    contains: Optional[str] = None\n    starts_with: Optional[str] = None\n    ends_with: Optional[str] = None\n    in_: Optional[List[str]] = None\n    not_in: Optional[List[str]] = None\n    gt: Optional[str] = None\n    gte: Optional[str] = None\n    lt: Optional[str] = None\n    lte: Optional[str] = None\n\n`;
+        content += `@dataclass\nclass IntFilter:\n    """Type-safe filter for integer fields."""\n    equals: Optional[int] = None\n    not_: Optional[int] = None\n    in_: Optional[List[int]] = None\n    not_in: Optional[List[int]] = None\n    gt: Optional[int] = None\n    gte: Optional[int] = None\n    lt: Optional[int] = None\n    lte: Optional[int] = None\n\n`;
+        content += `@dataclass\nclass NumberFilter:\n    """Type-safe filter for float/decimal fields."""\n    equals: Optional[float] = None\n    not_: Optional[float] = None\n    in_: Optional[List[float]] = None\n    not_in: Optional[List[float]] = None\n    gt: Optional[float] = None\n    gte: Optional[float] = None\n    lt: Optional[float] = None\n    lte: Optional[float] = None\n\n`;
+        content += `@dataclass\nclass BoolFilter:\n    """Type-safe filter for boolean fields."""\n    equals: Optional[bool] = None\n\n`;
+        content += `@dataclass\nclass DateTimeFilter:\n    """Type-safe filter for datetime fields."""\n    equals: Optional[datetime] = None\n    not_: Optional[datetime] = None\n    in_: Optional[List[datetime]] = None\n    not_in: Optional[List[datetime]] = None\n    gt: Optional[datetime] = None\n    gte: Optional[datetime] = None\n    lt: Optional[datetime] = None\n    lte: Optional[datetime] = None\n\n`;
+        // Per-model types
+        for (const model of models) {
+            const name = this.capitalize(model.name);
+            content += `# ─── ${name} ORM Types ────────────────────────────────────────────────────────\n\n`;
+            // WhereInput
+            content += `@dataclass\nclass ${name}WhereInput:\n    """Type-safe WHERE filter for ${name} queries."""\n`;
+            content += `    AND: Optional[List['${name}WhereInput']] = None\n`;
+            content += `    OR: Optional[List['${name}WhereInput']] = None\n`;
+            content += `    NOT: Optional['${name}WhereInput'] = None\n`;
+            for (const f of model.fields) {
+                const ft = this.getPyFilterType(f);
+                content += `    ${this.toSnakeCase(f.name)}: Optional[${ft}] = None\n`;
+            }
+            content += '\n';
+            // OrderByInput
+            content += `@dataclass\nclass ${name}OrderByInput:\n    """Type-safe ORDER BY for ${name} queries. Value: 'asc' or 'desc'."""\n`;
+            for (const f of model.fields) {
+                content += `    ${this.toSnakeCase(f.name)}: Optional[str] = None\n`;
+            }
+            content += '\n';
+            // CreateInput
+            const requiredFields = model.fields.filter(f => !f.isId && !f.isOptional && !f.hasDefault);
+            const optionalFields = model.fields.filter(f => !f.isId && (f.isOptional || f.hasDefault));
+            content += `@dataclass\nclass ${name}CreateInput:\n    """Typed data for creating a new ${name} record."""\n`;
+            for (const f of requiredFields) {
+                content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+            }
+            for (const f of optionalFields) {
+                content += `    ${this.toSnakeCase(f.name)}: Optional[${this.mapPyType(f)}] = None\n`;
+            }
+            if (requiredFields.length === 0 && optionalFields.length === 0)
+                content += '    pass\n';
+            content += '\n';
+            // UpdateInput
+            const updateFields = model.fields.filter(f => !f.isId);
+            content += `@dataclass\nclass ${name}UpdateInput:\n    """Typed data for updating an existing ${name} record."""\n`;
+            for (const f of updateFields) {
+                content += `    ${this.toSnakeCase(f.name)}: Optional[${this.mapPyType(f)}] = None\n`;
+            }
+            if (updateFields.length === 0)
+                content += '    pass\n';
+            content += '\n';
+            // FindManyArgs
+            content += `@dataclass\nclass ${name}FindManyArgs:\n    """ORM-style args for ${name}.find_many()."""\n`;
+            content += `    where: Optional[${name}WhereInput] = None\n`;
+            content += `    order_by: Optional[${name}OrderByInput] = None\n`;
+            content += `    take: Optional[int] = None\n`;
+            content += `    skip: int = 0\n`;
+            content += `    select: Optional[List[str]] = None\n\n`;
+            // FindFirstArgs
+            content += `@dataclass\nclass ${name}FindFirstArgs:\n    """ORM-style args for ${name}.find_first()."""\n`;
+            content += `    where: Optional[${name}WhereInput] = None\n`;
+            content += `    order_by: Optional[${name}OrderByInput] = None\n`;
+            content += `    select: Optional[List[str]] = None\n\n`;
+        }
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, 'an5_orm_types.py'), content);
+    }
+    generateMetadata(models) {
+        let pyContent = '# This file is auto-generated. Do not edit directly.\n\n';
+        // 1. MODEL_TO_TABLE
+        pyContent += 'MODEL_TO_TABLE = {\n';
+        for (const model of models) {
+            const props = this.getAllPropertyVariations(model.name);
+            const fullTableName = (0, types_1.bracketedTableName)(model);
+            for (const prop of props) {
+                pyContent += `    "${prop}": "${fullTableName}",\n`;
+            }
+        }
+        pyContent += '}\n\n';
+        // 2. MODEL_DESCRIPTIONS
+        pyContent += 'MODEL_DESCRIPTIONS = {\n';
+        for (const model of models) {
+            const props = this.getAllPropertyVariations(model.name);
+            for (const prop of props) {
+                pyContent += `    "${prop}": ${this.pyString(model.description)},\n`;
+            }
+        }
+        pyContent += '}\n\n';
+        // 3. MODEL_FIELDS
+        pyContent += 'MODEL_FIELDS = {\n';
+        for (const model of models) {
+            const props = this.getAllPropertyVariations(model.name);
+            const fieldsStr = this.formatFields(model);
+            for (const prop of props) {
+                pyContent += `    "${prop}": ${fieldsStr},\n`;
+            }
+        }
+        pyContent += '}\n\n';
+        // 4. RELATION_MAP
+        pyContent += 'RELATION_MAP = {\n';
+        for (const model of models) {
+            const props = this.getAllPropertyVariations(model.name);
+            let relationsContent = `{\n`;
+            for (const rel of model.relations) {
+                relationsContent += `        "${rel.name}": {\n`;
+                relationsContent += `            "modelName": "${this.toCamelCase(rel.type)}",\n`;
+                relationsContent += `            "relationType": "${rel.isArray ? 'many' : 'one'}",\n`;
+                relationsContent += `            "foreignKey": "${rel.foreignKey || 'id'}",\n`;
+                relationsContent += `            "localKey": "${rel.localKey || 'id'}"\n`;
+                relationsContent += `        },\n`;
+            }
+            relationsContent += `    }`;
+            for (const prop of props) {
+                pyContent += `    "${prop}": ${relationsContent},\n`;
+            }
+        }
+        pyContent += '}\n';
+        fs_1.default.writeFileSync(this.outputPath, pyContent);
+    }
+    /**
+     * The Python type for a field.
+     *
+     * From the declared type, so `DECIMAL` is a `float` and `BIGINT` stays an `int`:
+     * both arrived as `number` before and took the integer branch.
+     */
+    mapPyType(field) {
+        switch ((0, type_kinds_1.fieldKind)(field, this.provider)) {
+            case 'int':
+            case 'bigint': return 'int';
+            case 'float': return 'float';
+            case 'bool': return 'bool';
+            case 'date': return 'datetime';
+            case 'bytes': return 'bytes';
+            case 'json': return 'Any';
+            case 'vector': return 'List[float]';
+            default: return 'str';
+        }
+    }
+    generateModelFile(model, outputDir) {
+        // Single-entity file like dotnet `${model.name}.cs` / typescript `${model.name}.ts`.
+        // Contains the @dataclass entity + TypedDict row shapes for that model only.
+        let content = '# This file is auto-generated. Do not edit directly.\n';
+        content += 'from dataclasses import dataclass, field\n';
+        content += 'from typing import Optional, List, Any, TypedDict\n';
+        content += 'from datetime import datetime\n\n';
+        if (model.description) {
+            content += `"""${model.description}"""\n`;
+        }
+        content += `@dataclass\nclass ${model.name}:\n`;
+        if (model.fields.length === 0 && model.relations.length === 0) {
+            content += '    pass\n\n';
+        }
+        else {
+            // Required fields first
+            const required = model.fields.filter(f => !f.isOptional && !f.hasDefault);
+            const optionals = model.fields.filter(f => f.isOptional || f.hasDefault);
+            for (const f of required) {
+                const pyType = this.mapPyType(f);
+                const snakeName = this.toSnakeCase(f.name);
+                content += `    ${snakeName}: ${pyType}\n`;
+            }
+            for (const f of optionals) {
+                const pyType = this.mapPyType(f);
+                const snakeName = this.toSnakeCase(f.name);
+                content += `    ${snakeName}: Optional[${pyType}] = None\n`;
+            }
+            // Relations
+            for (const rel of model.relations) {
+                const relName = this.toSnakeCase(rel.name);
+                if (rel.isArray) {
+                    content += `    ${relName}: List[Any] = field(default_factory=list)\n`;
+                }
+                else {
+                    content += `    ${relName}: Optional[Any] = None\n`;
+                }
+            }
+            content += '\n';
+            // TypedDict row shape: mirrors what the runtime actually returns
+            // (plain dicts), so `AdapterTableClient["<Model>Row"]` delegates
+            // type-check. Required keys stay required via a total=True base;
+            // everything else is optional (total=False), keeping partial
+            // selects valid without extra dependencies.
+            const rowRequired = model.fields.filter(f => !f.isOptional && !f.hasDefault);
+            const rowOptional = model.fields.filter(f => f.isOptional || f.hasDefault);
+            content += `class _${model.name}Required(TypedDict):\n`;
+            content += `    """Required keys of a ${model.name} row."""\n`;
+            if (rowRequired.length === 0) {
+                content += '    pass\n';
+            }
+            for (const f of rowRequired) {
+                content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+            }
+            content += `\nclass ${model.name}Row(_${model.name}Required, total=False):\n`;
+            content += `    """Row shape returned for ${model.name} queries."""\n`;
+            if (rowOptional.length === 0 && model.relations.length === 0) {
+                content += '    pass\n';
+            }
+            for (const f of rowOptional) {
+                content += `    ${this.toSnakeCase(f.name)}: ${this.mapPyType(f)}\n`;
+            }
+            for (const rel of model.relations) {
+                const relName = this.toSnakeCase(rel.name);
+                content += rel.isArray ? `    ${relName}: List[Any]\n` : `    ${relName}: Any\n`;
+            }
+            content += '\n';
+        }
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, `${model.name}.py`), content);
+    }
+    generateModelsIndex(models, outputDir) {
+        // Backward-compat aggregator: `from an5_models import User` keeps working.
+        // New code can also import directly: `from .User import User, UserRow`.
+        // Supports both package-relative (`python -m`) and top-level (`sys.path`
+        // pointing at the output dir) imports via try/except fallback.
+        let content = '# This file is auto-generated. Do not edit directly.\n';
+        content += '"""Backward-compat aggregator re-exporting per-model entity files."""\n';
+        for (const model of models) {
+            content += `try:\n`;
+            content += `    from .${model.name} import ${model.name}, ${model.name}Row, _${model.name}Required\n`;
+            content += `except ImportError:\n`;
+            content += `    from ${model.name} import ${model.name}, ${model.name}Row, _${model.name}Required\n`;
+        }
+        if (models.length > 0) {
+            content += '\n__all__ = [\n';
+            for (const model of models) {
+                content += `    "${model.name}", "${model.name}Row", "_${model.name}Required",\n`;
+            }
+            content += ']\n';
+        }
+        else {
+            content += '\n__all__ = []\n';
+        }
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, 'an5_models.py'), content);
+    }
+    generateClient(models, outputDir) {
+        let content = '# This file is auto-generated. Do not edit directly.\n';
+        content += 'import os\n';
+        content += 'from typing import Dict, List, Optional, Any, Callable, TYPE_CHECKING\n\n';
+        // Runtime adapter comes from the `an5-adapters` PyPI package
+        // (module `an5_adapter`). There is no local adapter file in the
+        // generated output dir, so no relative-import fallback here — a
+        // relative fallback would be unresolvable and trip Pylance
+        // `reportMissingImports` on every generated client.
+        content += 'from an5_adapter import An5Adapter, AdapterTableClient, create_an5_adapter, set_adapter_metadata\n\n';
+        const metadataModule = this.metadataModule();
+        content += 'try:\n';
+        content += `    from .${metadataModule} import MODEL_TO_TABLE, MODEL_FIELDS\n`;
+        content += 'except ImportError:\n';
+        content += `    from ${metadataModule} import MODEL_TO_TABLE, MODEL_FIELDS\n\n`;
+        if (models.length > 0) {
+            content += 'if TYPE_CHECKING:\n';
+            for (const m of models) {
+                content += '    try:\n';
+                content += `        from .${m.name} import ${m.name}Row\n`;
+                content += '    except ImportError:\n';
+                content += `        from .an5_models import ${m.name}Row  # fallback aggregator\n`;
+            }
+            content += '\n';
+        }
+        content += 'class An5Client:\n';
+        content += '    """AN5 Python ORM Client - type-safe database access.\n\n';
+        content += '    Usage:\n';
+        content += '        db = An5Client()\n';
+        content += '        users = db.user.find_many(where={"name": {"contains": "John"}}, order_by={"created_at": "asc"}, take=10)\n';
+        content += '        user  = db.user.find_first(where={"id": "abc"})\n';
+        content += '        new   = db.user.create(data={"name": "Alice", "email": "alice@example.com"})\n';
+        content += '    """\n';
+        content += '    def __init__(self, connection_string: Optional[str] = None):\n';
+        content += '        conn_str = (connection_string or \n';
+        content += '                    os.getenv("DATABASE_URL") or \n';
+        content += '                    "Server=localhost;Database=master;Trusted_Connection=True;TrustServerCertificate=True;")\n';
+        content += '        set_adapter_metadata({"model_to_table": MODEL_TO_TABLE, "model_fields": MODEL_FIELDS})\n';
+        content += '        self.adapter: An5Adapter = create_an5_adapter(conn_str)\n\n';
+        for (const model of models) {
+            const propName = this.toSnakeCase(model.name) + 's';
+            const singleName = this.toSnakeCase(model.name);
+            content += `        client = AdapterTableClient(self.adapter, "${model.name}")\n`;
+            content += `        self.${model.name}: AdapterTableClient["${model.name}Row"] = client\n`;
+            content += `        self.${model.name}s: AdapterTableClient["${model.name}Row"] = client\n`;
+            content += `        self.${singleName}: AdapterTableClient["${model.name}Row"] = client\n`;
+            content += `        self.${propName}: AdapterTableClient["${model.name}Row"] = client\n`;
+        }
+        content += '\n';
+        content += '    def __getattr__(self, name: str) -> AdapterTableClient[Any]:\n';
+        content += '        return self.adapter.table(name)\n\n';
+        content += '    def query_raw(self, sql: str, *params) -> List[Dict]:\n';
+        content += '        return self.adapter.query_raw(sql, *params)\n\n';
+        content += '    def execute_raw(self, sql: str, *params) -> int:\n';
+        content += '        return self.adapter.execute_raw(sql, *params)\n\n';
+        content += '    def transaction(self, fn: Callable) -> Any:\n';
+        content += '        return self.adapter.transaction(fn)\n';
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, 'an5_client.py'), content);
+    }
+    generateInit(models, outputDir) {
+        let content = '# This file is auto-generated. Do not edit directly.\n';
+        content += `from .${this.metadataModule()} import MODEL_TO_TABLE, MODEL_DESCRIPTIONS, MODEL_FIELDS, RELATION_MAP\n`;
+        content += 'from .an5_models import *\n';
+        content += 'from .an5_orm_types import (\n';
+        content += '    StringFilter, IntFilter, NumberFilter, BoolFilter, DateTimeFilter,\n';
+        for (const m of models) {
+            const name = this.capitalize(m.name);
+            content += `    ${name}WhereInput, ${name}OrderByInput,\n`;
+            content += `    ${name}CreateInput, ${name}UpdateInput,\n`;
+            content += `    ${name}FindManyArgs, ${name}FindFirstArgs,\n`;
+        }
+        content += ')\n';
+        content += 'from .an5_client import An5Client\n\n';
+        content += '__all__ = [\n';
+        content += '    "MODEL_TO_TABLE",\n';
+        content += '    "MODEL_DESCRIPTIONS",\n';
+        content += '    "MODEL_FIELDS",\n';
+        content += '    "RELATION_MAP",\n';
+        content += '    "An5Client",\n';
+        content += '    "StringFilter", "IntFilter", "NumberFilter", "BoolFilter", "DateTimeFilter",\n';
+        for (const m of models) {
+            const name = this.capitalize(m.name);
+            content += `    "${m.name}",\n`;
+            content += `    "${name}WhereInput", "${name}OrderByInput",\n`;
+            content += `    "${name}CreateInput", "${name}UpdateInput",\n`;
+            content += `    "${name}FindManyArgs", "${name}FindFirstArgs",\n`;
+        }
+        content += ']\n';
+        fs_1.default.writeFileSync(path_1.default.join(outputDir, '__init__.py'), content);
+    }
+    getAllPropertyVariations(modelName) {
+        const variations = new Set();
+        variations.add(this.toCamelCase(modelName));
+        variations.add(this.toSnakeCase(modelName));
+        const acronyms = ['LLM', 'AI', 'MCP', 'IT', 'QC', 'HR', 'MR', 'WH', 'SSIS', 'API', 'URL', 'ID', 'JSON'];
+        for (const acronym of acronyms) {
+            if (modelName.startsWith(acronym)) {
+                variations.add(acronym.toLowerCase() + modelName.slice(acronym.length));
+                variations.add(acronym[0].toLowerCase() + acronym.slice(1) + modelName.slice(acronym.length));
+                variations.add(this.toSnakeCase(acronym.toLowerCase() + modelName.slice(acronym.length)));
+            }
+        }
+        return Array.from(variations);
+    }
+    formatFields(model) {
+        if (model.fields.length === 0)
+            return '[]';
+        const fields = model.fields.map(f => {
+            return `        { "name": "${f.name}", "type": "${f.type}${f.isOptional ? '?' : ''}", "sql": "${f.sqlType}", "isOptional": ${f.isOptional ? 'True' : 'False'}, "hasDefault": ${f.hasDefault ? 'True' : 'False'}, "isId": ${f.isId ? 'True' : 'False'}, "description": ${this.pyString(f.description)} }`;
+        });
+        return `[\n${fields.join(',\n')}\n    ]`;
+    }
+    pyString(value) {
+        return value ? JSON.stringify(value) : 'None';
+    }
+    toCamelCase(str) {
+        if (!str)
+            return '';
+        return str[0].toLowerCase() + str.slice(1);
+    }
+    toSnakeCase(str) {
+        if (!str)
+            return '';
+        return str
+            .replace(/([A-Z]+)([A-Z][a-z])/g, '$1_$2')
+            .replace(/([a-z\d])([A-Z])/g, '$1_$2')
+            .toLowerCase();
+    }
+    capitalize(str) {
+        if (!str)
+            return '';
+        return str[0].toUpperCase() + str.slice(1);
+    }
+}
+exports.PythonGenerator = PythonGenerator;

package/dist/generator/src/rust-generator.d.ts

@@ -0,0 +1,38 @@
+import { Model } from './types';
+export declare class RustGenerator {
+    private outputDir;
+    /** The database being generated for; decides types like `TIMESTAMP`. */
+    private provider;
+    constructor(outputDir: string);
+    generate(models: Model[]): void;
+    private generateCargoToml;
+    private generateLibRs;
+    private generateFiltersRs;
+    private generateModelsRs;
+    private buildModelsBody;
+    private generateMetadataRs;
+    private generateConfigRs;
+    private generateClientRs;
+    /**
+     * Typed, executing accessors for a model.
+     *
+     * Rust has no dynamic property access, so `db.user` cannot exist the way it
+     * does in TypeScript or Python. Following the Go generator (which emits a
+     * real `ctx.User` field), each model gets a generated method returning a
+     * typed handle whose queries run through the an5-adapters runtime.
+     */
+    private buildModelRuntimeHandle;
+    private isBoolField;
+    /**
+     * The Rust type for a field.
+     *
+     * From the declared type, so `DECIMAL` is an `f64` and `BIGINT` an `i64`: both
+     * arrived as `number` and were written as `i64`.
+     */
+    private mapRustType;
+    private getRustFilterType;
+    private getAllPropertyVariations;
+    private capitalize;
+    private toCamelCase;
+    private toSnakeCase;
+}