iceaxe 0.7.1__cp313-cp313-macosx_11_0_arm64.whl

iceaxe/schemas/cli.py

@@ -0,0 +1,30 @@
+from inspect import isclass
+from typing import Type
+
+from iceaxe.base import DBModelMetaclass, TableBase
+from iceaxe.schemas.actions import DatabaseActions
+from iceaxe.schemas.db_memory_serializer import DatabaseMemorySerializer
+from iceaxe.session import DBConnection
+
+
+async def create_all(
+    db_connection: DBConnection, models: list[Type[TableBase]] | None = None
+):
+    # Get all of the instances that have been registered
+    # in memory scope by the user.
+    if models is None:
+        models = [
+            cls
+            for cls in DBModelMetaclass.get_registry()
+            if isclass(cls) and issubclass(cls, TableBase)
+        ]
+
+    migrator = DatabaseMemorySerializer()
+    db_objects = list(migrator.delegate(models))
+    next_ordering = migrator.order_db_objects(db_objects)
+
+    # Create the tables in the database
+    actor = DatabaseActions(dry_run=False, db_connection=db_connection)
+    await migrator.build_actions(
+        actor, [], {}, [obj for obj, _ in db_objects], next_ordering
+    )