sera-2 1.14.6__tar.gz → 1.15.0__tar.gz

{sera_2-1.14.6 → sera_2-1.15.0}/PKG-INFO

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

{sera_2-1.14.6 → sera_2-1.15.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sera-2"
-version = "1.14.6"
+version = "1.15.0"
 description = ""
 authors = ["Binh Vu <bvu687@gmail.com>"]
 readme = "README.md"

{sera_2-1.14.6 → sera_2-1.15.0}/sera/make/make_typescript_model.py

@@ -1104,7 +1104,19 @@ def make_typescript_data_model(schema: Schema, target_pkg: Package):
                 for dep in tstype.deps:
                     program.import_(dep, True)
                 tsprop = [
-                    (expr.ExprIdent("datatype"), expr.ExprConstant(tstype.type)),
+                    (
+                        expr.ExprIdent("datatype"),
+                        (
+                            expr.ExprConstant(tstype.type)
+                            if tstype.type not in schema.enums
+                            else expr.ExprConstant("enum")
+                        ),
+                    ),
+                    *(
+                        [(expr.ExprIdent("enumType"), expr.ExprIdent(tstype.type))]
+                        if tstype.type in schema.enums
+                        else []
+                    ),
                     (
                         expr.ExprIdent("isRequired"),
                         expr.ExprConstant(

{sera_2-1.14.6 → sera_2-1.15.0}/sera/misc/__init__.py

@@ -3,6 +3,7 @@ from sera.misc._utils import (
     assert_isinstance,
     assert_not_null,
     filter_duplication,
+    load_data,
     to_camel_case,
     to_pascal_case,
     to_snake_case,
@@ -18,4 +19,5 @@ __all__ = [
     "to_pascal_case",
     "Formatter",
     "File",
+    "load_data",
 ]

{sera_2-1.14.6 → sera_2-1.15.0}/sera/misc/_utils.py

@@ -1,8 +1,14 @@
 from __future__ import annotations
 
 import re
+from pathlib import Path
 from typing import Any, Callable, Iterable, Optional, TypeVar
 
+import serde.csv
+from sqlalchemy import Engine, text
+from sqlalchemy.orm import Session
+from tqdm import tqdm
+
 T = TypeVar("T")
 reserved_keywords = {
     "and",
@@ -102,3 +108,43 @@ def filter_duplication(
             keys.add(k)
             new_lst.append(k)
     return new_lst
+
+
+def load_data(
+    engine: Engine,
+    create_db_and_tables: Callable[[], None],
+    table_files: list[tuple[type, Path]],
+    table_desers: dict[type, Callable[[dict], Any]],
+    verbose: bool = False,
+):
+    """
+    Load data into the database from specified CSV files.
+
+    Args:
+        engine: SQLAlchemy engine to connect to the database.
+        create_db_and_tables: Function to create database and tables.
+        table_files: List of tuples containing the class type and the corresponding CSV file path.
+        table_desers: Dictionary mapping class types to their deserializer functions.
+        verbose: If True, show progress bars during loading.
+    """
+    with Session(engine) as session:
+        create_db_and_tables()
+
+        for tbl, file in tqdm(table_files, disable=not verbose, desc="Loading data"):
+            if file.name.endswith(".csv"):
+                records = serde.csv.deser(file, deser_as_record=True)
+            else:
+                raise ValueError(f"Unsupported file format: {file.name}")
+            deser = table_desers[tbl]
+            records = [deser(row) for row in records]
+            for r in tqdm(records, desc=f"load {tbl.__name__}", disable=not verbose):
+                session.merge(r)
+            session.flush()
+
+            # Reset the sequence for each table
+            session.execute(
+                text(
+                    f"SELECT setval('{tbl.__tablename__}_id_seq', (SELECT MAX(id) FROM \"{tbl.__tablename__}\"));"
+                )
+            )
+        session.commit()