sera-2 1.12.6__py3-none-any.whl → 1.13.1__py3-none-any.whl

sera/exports/schema.py

@@ -0,0 +1,157 @@
+from pathlib import Path
+from typing import Annotated
+
+import typer
+
+from sera.models import Cardinality, Class, DataProperty, Schema, parse_schema
+from sera.models._datatype import DataType
+
+
+def get_prisma_field_type(datatype: DataType) -> str:
+    pytype = datatype.get_python_type().type
+    if pytype == "str":
+        return "String"
+    if pytype == "int":
+        return "Int"
+    if pytype == "float":
+        return "Float"
+    if pytype == "bool":
+        return "Boolean"
+    if pytype == "bytes":
+        return "Bytes"
+    if pytype == "dict":
+        return "Json"
+    if pytype == "datetime":
+        return "DateTime"
+    if pytype == "list[str]":
+        return "String[]"
+    if pytype == "list[int]":
+        return "Int[]"
+    if pytype == "list[float]":
+        return "Float[]"
+    if pytype == "list[bool]":
+        return "Boolean[]"
+    if pytype == "list[bytes]":
+        return "Bytes[]"
+    if pytype == "list[dict]":
+        return "Json[]"
+    if pytype == "list[datetime]":
+        return "DateTime[]"
+
+    raise ValueError(f"Unsupported data type for Prisma: {pytype}")
+
+
+def to_prisma_model(schema: Schema, cls: Class, lines: list[str]):
+    """Convert a Sera Class to a Prisma model string representation."""
+    lines.append(f"model {cls.name} {{")
+
+    if cls.db is None:
+        # This class has no database mapping, we must generate a default key for it
+        lines.append(
+            f"  {'id'.ljust(30)} {'Int'.ljust(10)} @id @default(autoincrement())"
+        )
+    #     lines.append(f"  @@unique([%s])" % ", ".join(cls.properties.keys()))
+
+    for prop in cls.properties.values():
+        propattrs = ""
+        if isinstance(prop, DataProperty):
+            proptype = get_prisma_field_type(prop.datatype)
+            if prop.is_optional:
+                proptype = f"{proptype}?"
+            if prop.db is not None and prop.db.is_primary_key:
+                propattrs += "@id "
+
+            lines.append(f"  {prop.name.ljust(30)} {proptype.ljust(10)} {propattrs}")
+            continue
+
+        if prop.cardinality == Cardinality.MANY_TO_MANY:
+            # For many-to-many relationships, we need to handle the join table
+            lines.append(
+                f"  {prop.name.ljust(30)} {(prop.target.name + '[]').ljust(10)}"
+            )
+        else:
+            lines.append(
+                f"  {(prop.name + '_').ljust(30)} {prop.target.name.ljust(10)} @relation(fields: [{prop.name}], references: [id])"
+            )
+            lines.append(f"  {prop.name.ljust(30)} {'Int'.ljust(10)} @unique")
+
+    lines.append("")
+    for upstream_cls, reverse_upstream_prop in schema.get_upstream_classes(cls):
+        if (
+            reverse_upstream_prop.cardinality == Cardinality.MANY_TO_ONE
+            or reverse_upstream_prop.cardinality == Cardinality.MANY_TO_MANY
+        ):
+
+            proptype = f"{upstream_cls.name}[]"
+        else:
+            proptype = upstream_cls.name + "?"
+        lines.append(f"  {upstream_cls.name.lower().ljust(30)} {proptype.ljust(10)}")
+
+    lines.append("}\n")
+
+
+def export_prisma_schema(schema: Schema, outfile: Path):
+    """Export Prisma schema file"""
+    lines = []
+
+    # Datasource
+    lines.append("datasource db {")
+    lines.append(
+        '  provider = "postgresql"'
+    )  # Defaulting to postgresql as per user context
+    lines.append('  url      = env("DATABASE_URL")')
+    lines.append("}\n")
+
+    # Generator
+    lines.append("generator client {")
+    lines.append('  provider = "prisma-client-py"')
+    lines.append("  recursive_type_depth = 5")
+    lines.append("}\n")
+
+    # Enums
+    if schema.enums:
+        for enum_name, enum_def in schema.enums.items():
+            lines.append(f"enum {enum_name} {{")
+            # Assuming enum_def.values is a list of strings based on previous errors
+            for val_str in enum_def.values:
+                lines.append(f"  {val_str}")
+            lines.append("}\\n")
+
+    # Models
+    for cls in schema.topological_sort():
+        to_prisma_model(schema, cls, lines)
+
+    with outfile.open("w", encoding="utf-8") as f:
+        f.write("\n".join(lines))
+
+
+app = typer.Typer(pretty_exceptions_short=True, pretty_exceptions_enable=False)
+
+
+@app.command()
+def cli(
+    schema_files: Annotated[
+        list[Path],
+        typer.Option(
+            "-s", help="YAML schema files. Multiple files are merged automatically"
+        ),
+    ],
+    outfile: Annotated[
+        Path,
+        typer.Option(
+            "-o", "--output", help="Output file for the Prisma schema", writable=True
+        ),
+    ],
+):
+    schema = parse_schema(
+        "sera",
+        schema_files,
+    )
+    export_prisma_schema(
+        schema,
+        outfile,
+    )
+
+
+if __name__ == "__main__":
+    app()

sera/exports/test.py

@@ -0,0 +1,70 @@
+from sqlalchemy import (
+    Column,
+    ForeignKey,
+    Integer,
+    MetaData,
+    String,
+    Table,
+    create_engine,
+)
+from sqlalchemy.schema import CreateTable
+
+# Define your SQLAlchemy engine (dialect matters for SQL output)
+# Using a specific dialect helps generate appropriate SQL
+# engine = create_engine(
+#     "postgresql+psycopg2://user:password@host:port/database", echo=False
+# )
+# Or for SQLite:
+# engine = create_engine("sqlite:///:memory:")
+# Or for MySQL:
+# engine = create_engine("mysql+mysqlconnector://user:password@host:port/database")
+
+
+metadata_obj = MetaData()
+
+user_table = Table(
+    "users",
+    metadata_obj,
+    Column("id", Integer, primary_key=True),
+    Column("name", String(50)),
+    Column("email", String(100), unique=True),
+)
+
+address_table = Table(
+    "addresses",
+    metadata_obj,
+    Column("id", Integer, primary_key=True),
+    Column("user_id", Integer, ForeignKey("users.id"), nullable=False),
+    Column("street_name", String(100)),
+    Column("city", String(50)),
+)
+
+# --- ORM Example ---
+# from sqlalchemy.orm import declarative_base, Mapped, mapped_column
+# Base = declarative_base()
+# metadata_obj = Base.metadata
+# class User(Base): # ... (define as above)
+# class Address(Base): # ... (define as above)
+# -------------------
+
+print("--- Generating DDL for PostgreSQL ---")
+
+
+def generate_schema_ddl(metadata, engine_dialect):
+    for table in metadata.sorted_tables:
+        # The CreateTable construct can be compiled to a string
+        # specific to the dialect of the engine.
+        create_table_ddl = CreateTable(table).compile(dialect=engine_dialect)
+        print(str(create_table_ddl).strip() + ";\n")
+
+
+from sqlalchemy.dialects import postgresql, sqlite
+
+generate_schema_ddl(metadata_obj, postgresql.dialect())
+
+# # Example with a different dialect (e.g., SQLite)
+# # Note: You don't need a live connection for this, just the dialect.
+
+
+# print("\n--- Generating DDL for SQLite ---")
+# generate_schema_ddl(metadata_obj, sqlite.dialect())

sera/libs/api_test_helper.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+from litestar import Litestar
+from litestar.testing import TestClient
+
+
+def test_has(
+    client: TestClient[Litestar],
+    base_url: str,
+    exist_records: list[int | str],
+    non_exist_records: list[int | str],
+) -> None:
+    for record in exist_records:
+        resp = client.head(f"{base_url}/{record}")
+        assert (
+            resp.status_code == 204
+        ), f"Record {record} should exist but got {resp.status_code}"
+
+    for record in non_exist_records:
+        resp = client.head(f"{base_url}/{record}")
+        assert (
+            resp.status_code == 404
+        ), f"Record {record} should not exist but got {resp.status_code}"
+
+
+def test_get_by_id(
+    client: TestClient[Litestar],
+    base_url: str,
+    exist_records: dict[int | str, dict],
+    non_exist_records: list[int | str],
+) -> None:
+    for record, data in exist_records.items():
+        resp = client.get(f"{base_url}/{record}")
+        assert (
+            resp.status_code == 200
+        ), f"Record {record} should exist but got {resp.status_code}"
+        assert resp.json() == data
+
+    for record in non_exist_records:
+        resp = client.get(f"{base_url}/{record}")
+        assert (
+            resp.status_code == 404
+        ), f"Record {record} should not exist but got {resp.status_code}"

sera/models/_schema.py

@@ -34,3 +34,14 @@ class Schema:
 
         # Convert sorted names back to Class objects
         return [self.classes[name] for name in sorted_names]
+
+    def get_upstream_classes(self, cls: Class) -> list[tuple[Class, ObjectProperty]]:
+        """
+        Get all classes that depend on the given class.
+        """
+        upstream_classes = []
+        for other_cls in self.classes.values():
+            for prop in other_cls.properties.values():
+                if isinstance(prop, ObjectProperty) and prop.target.name == cls.name:
+                    upstream_classes.append((other_cls, prop))
+        return upstream_classes

{sera_2-1.12.6.dist-info → sera_2-1.13.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sera-2
-Version: 1.12.6
+Version: 1.13.1
 Summary: 
 Author: Binh Vu
 Author-email: bvu687@gmail.com

{sera_2-1.12.6.dist-info → sera_2-1.13.1.dist-info}/RECORD

@@ -1,7 +1,11 @@
 sera/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sera/constants.py,sha256=mzAaMyIx8TJK0-RYYJ5I24C4s0Uvj26OLMJmBo0pxHI,123
+sera/exports/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sera/exports/schema.py,sha256=3F5Kx3r0zOmiVpDf9vnzGhtdcuF-vfgMhBbp4Ko4fgw,4740
+sera/exports/test.py,sha256=jK1EJmLGiy7eREpnY_68IIVRH43uH8S_u5Z7STPbXOM,2002
 sera/libs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sera/libs/api_helper.py,sha256=47y1kcwk3Xd2ZEMnUj_0OwCuUmgwOs5kYrE95BDVUn4,5411
+sera/libs/api_test_helper.py,sha256=3tRr8sLN4dBSrHgKAHMmyoENI0xh7K_JLel8AvujU7k,1323
 sera/libs/base_orm.py,sha256=5hOH_diUeaABm3cpE2-9u50VRqG1QW2osPQnvVHIhIA,3365
 sera/libs/base_service.py,sha256=AX1WoTHte6Z_birkkfagkNE6BrCLTlTjQE4jEsKEaAY,5152
 sera/libs/dag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -30,8 +34,8 @@ sera/models/_module.py,sha256=8QRSCubZmdDP9rL58rGAS6X5VCrkc1ZHvuMu1I1KrWk,5043
 sera/models/_multi_lingual_string.py,sha256=JETN6k00VH4wrA4w5vAHMEJV8fp3SY9bJebskFTjQLA,1186
 sera/models/_parse.py,sha256=q_YZ7PrHWIN85_WW-fPP7-2gLXlGWM2-EIdbYXuG7Xg,10052
 sera/models/_property.py,sha256=4y9F58D6DoX25-6aWPBRiE72nCPQy0KWlGNDTZXSV-8,6038
-sera/models/_schema.py,sha256=r-Gqg9Lb_wR3UrbNvfXXgt_qs5bts0t2Ve7aquuF_OI,1155
+sera/models/_schema.py,sha256=VxJEiqgVvbXgcSUK4UW6JnRcggk4nsooVSE6MyXmfNY,1636
 sera/typing.py,sha256=Q4QMfbtfrCjC9tFfsZPhsAnbNX4lm4NHQ9lmjNXYdV0,772
-sera_2-1.12.6.dist-info/METADATA,sha256=MlOqmC3MJASvXLuS8k0KO5UE8325j9Ff53RArvIn5Wo,867
-sera_2-1.12.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-sera_2-1.12.6.dist-info/RECORD,,
+sera_2-1.13.1.dist-info/METADATA,sha256=xM2la_NEr7esZtrLlTbwARiHlqp43Y0VVqQR4zuFEmo,867
+sera_2-1.13.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+sera_2-1.13.1.dist-info/RECORD,,