etlantic-sqlmodel 0.14.0__tar.gz → 0.19.0__tar.gz

{etlantic_sqlmodel-0.14.0 → etlantic_sqlmodel-0.19.0}/.gitignore

@@ -20,6 +20,8 @@ htmlcov/
 # macOS
 .DS_Store
 **/.DS_Store
+._*
+**/._*
 
 examples/_file_storage_out/
 examples/_generated_*.py

{etlantic_sqlmodel-0.14.0 → etlantic_sqlmodel-0.19.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: etlantic-sqlmodel
-Version: 0.14.0
+Version: 0.19.0
 Summary: ContractModel ↔ SQLModel bridge for ETLantic.
 Project-URL: Homepage, https://github.com/eddiethedean/etlantic
 Project-URL: Documentation, https://github.com/eddiethedean/etlantic/tree/main/docs
@@ -9,7 +9,7 @@ Project-URL: Issues, https://github.com/eddiethedean/etlantic/issues
 Project-URL: Changelog, https://github.com/eddiethedean/etlantic/blob/main/CHANGELOG.md
 Author-email: Odo Matthews <odosmatthews@gmail.com>
 License-Expression: MIT
-Classifier: Development Status :: 3 - Alpha
+Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
@@ -18,17 +18,17 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Typing :: Typed
 Requires-Python: >=3.11
-Requires-Dist: etlantic<0.15,>=0.14.0
+Requires-Dist: etlantic<0.20,>=0.19.0
 Requires-Dist: sqlmodel<1,>=0.0.22
 Description-Content-Type: text/markdown
 
 # etlantic-sqlmodel
 
 Optional bridge between ETLantic `Data` contracts and
-[SQLModel](https://sqlmodel.tiangolo.com/) table models for ETLantic 0.14.
+[SQLModel](https://sqlmodel.tiangolo.com/) table models for ETLantic 0.19.
 
 ```bash
-pip install 'etlantic==0.14.0' 'etlantic-sqlmodel==0.14.0'
+pip install 'etlantic==0.19.0' 'etlantic-sqlmodel==0.19.0'
 # or: pip install 'etlantic[sqlmodel]'
 ```
 

{etlantic_sqlmodel-0.14.0 → etlantic_sqlmodel-0.19.0}/README.md

@@ -1,10 +1,10 @@
 # etlantic-sqlmodel
 
 Optional bridge between ETLantic `Data` contracts and
-[SQLModel](https://sqlmodel.tiangolo.com/) table models for ETLantic 0.14.
+[SQLModel](https://sqlmodel.tiangolo.com/) table models for ETLantic 0.19.
 
 ```bash
-pip install 'etlantic==0.14.0' 'etlantic-sqlmodel==0.14.0'
+pip install 'etlantic==0.19.0' 'etlantic-sqlmodel==0.19.0'
 # or: pip install 'etlantic[sqlmodel]'
 ```
 

{etlantic_sqlmodel-0.14.0 → etlantic_sqlmodel-0.19.0}/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "etlantic-sqlmodel"
-version = "0.14.0"
+version = "0.19.0"
 description = "ContractModel ↔ SQLModel bridge for ETLantic."
 readme = "README.md"
 license = "MIT"
 requires-python = ">=3.11"
 authors = [{ name = "Odo Matthews", email = "odosmatthews@gmail.com" }]
 classifiers = [
-    "Development Status :: 3 - Alpha",
+    "Development Status :: 5 - Production/Stable",
     "Intended Audience :: Developers",
     "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
@@ -17,7 +17,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dependencies = [
-    "etlantic>=0.14.0,<0.15",
+    "etlantic>=0.19.0,<0.20",
     "sqlmodel>=0.0.22,<1",
 ]
 

{etlantic_sqlmodel-0.14.0 → etlantic_sqlmodel-0.19.0}/src/etlantic_sqlmodel/__init__.py

@@ -20,13 +20,14 @@ from etlantic.schema_drift import (
 )
 from sqlmodel import Field, SQLModel
 
-__version__ = "0.14.0"
+__version__ = "0.19.0"
 
 __all__ = [
     "SqlModelIntegrationPlugin",
     "__version__",
     "compare_metadata",
     "contract_to_sqlmodel",
+    "contract_to_sqlmodel_source",
     "create_plugin",
     "run_conformance_checks",
     "sqlmodel_to_contract",
@@ -82,6 +83,99 @@ def contract_to_sqlmodel(
     return SQLModelMetaclass(model_name, (SQLModel,), attrs, table=True)
 
 
+def contract_to_sqlmodel_source(
+    contract_cls: type[Data],
+    *,
+    table_name: str | None = None,
+    primary_key: tuple[str, ...] | None = None,
+) -> str:
+    """Generate importable SQLModel source for a ``Data`` / ContractModel class."""
+    if not is_data_contract_type(contract_cls):
+        raise TypeError("Expected a Data / ContractModel subclass")
+
+    resolved_table = table_name or _default_table_name(contract_cls)
+    pk_fields = set(primary_key or ())
+    model_name = f"{contract_cls.__name__}Table"
+    imports: set[str] = set()
+    field_lines: list[str] = []
+
+    for name, field_info in contract_cls.model_fields.items():
+        python_type = _python_type_from_annotation(field_info.annotation)
+        type_name, extra_imports = _annotation_source(python_type, optional=False)
+        imports.update(extra_imports)
+        optional = not field_info.is_required()
+        if optional:
+            type_name = f"{type_name} | None"
+        if name in pk_fields:
+            field_lines.append(f"    {name}: {type_name} = Field(primary_key=True)")
+        elif field_info.default is not ...:
+            default_repr = _default_source(field_info.default, imports)
+            field_lines.append(
+                f"    {name}: {type_name} = Field(default={default_repr})"
+            )
+        elif optional:
+            field_lines.append(f"    {name}: {type_name} = Field(default=None)")
+        else:
+            field_lines.append(f"    {name}: {type_name}")
+
+    import_lines = ["from sqlmodel import Field, SQLModel"]
+    if "datetime" in imports:
+        import_lines.insert(0, "from datetime import datetime")
+    if "date" in imports:
+        import_lines.insert(0, "from datetime import date")
+    if "Decimal" in imports:
+        import_lines.insert(0, "from decimal import Decimal")
+    # De-duplicate datetime imports when both date and datetime needed
+    if "datetime" in imports and "date" in imports:
+        import_lines = [
+            line
+            for line in import_lines
+            if line
+            not in {"from datetime import date", "from datetime import datetime"}
+        ]
+        import_lines.insert(0, "from datetime import date, datetime")
+
+    body = "\n".join(field_lines) if field_lines else "    pass"
+    return (
+        "\n".join(import_lines)
+        + "\n\n\n"
+        + f"class {model_name}(SQLModel, table=True):\n"
+        + f'    __tablename__ = "{resolved_table}"\n'
+        + f"{body}\n"
+    )
+
+
+def _annotation_source(
+    python_type: type[Any], *, optional: bool
+) -> tuple[str, set[str]]:
+    _ = optional
+    mapping = {
+        str: ("str", set()),
+        int: ("int", set()),
+        float: ("float", set()),
+        bool: ("bool", set()),
+        datetime: ("datetime", {"datetime"}),
+        date: ("date", {"date"}),
+        Decimal: ("Decimal", {"Decimal"}),
+    }
+    if python_type in mapping:
+        return mapping[python_type]
+    return ("str", set())
+
+
+def _default_source(value: Any, imports: set[str]) -> str:
+    if isinstance(value, Decimal):
+        imports.add("Decimal")
+        return f"Decimal({str(value)!r})"
+    if isinstance(value, datetime):
+        imports.add("datetime")
+        return repr(value)
+    if isinstance(value, date):
+        imports.add("date")
+        return repr(value)
+    return repr(value)
+
+
 def sqlmodel_to_contract(model_cls: type[Any]) -> dict[str, Any]:
     """Extract draft contract metadata from a SQLModel table class."""
     if not _is_sqlmodel_table(model_cls):