chore: rename llmpa.models to llmpa schemas (#27)

nuffin · Oct 19, 2024 · f528a8d · f528a8d
1 parent 3e27bfc
commit f528a8d
Show file tree

Hide file tree

Showing 7 changed files with 6 additions and 6 deletions.
diff --git a/alembic/env.py b/alembic/env.py
@@ -13,7 +13,7 @@
 
 from alembic import context
 
-from db import db, init_models
+from db import db, init_schemas
 
 # this is the Alembic Config object, which provides
 # access to the values within the .ini file in use.
@@ -26,7 +26,7 @@
 
 
 ## to init db.metadata
-init_models()
+init_schemas()
 
 # add your model's MetaData object here
 # for 'autogenerate' support

diff --git a/llmpa/db.py b/llmpa/db.py
@@ -13,16 +13,16 @@
 async_session = sessionmaker(async_engine, class_=AsyncSession, expire_on_commit=False)
 
 
-def init_models():
-    from models import File, User
+def init_schemas():
+    from schemas import File, User
 
     __models = [File, User]
 
 
 def init_db(app: Flask):
     global async_engine, async_session
 
-    init_models()
+    init_schemas()
 
     db.init_app(app)
 

diff --git a/llmpa/models/__init__.py → llmpa/schemas/__init__.py b/llmpa/models/__init__.py → llmpa/schemas/__init__.py
diff --git a/llmpa/models/base.py → llmpa/schemas/base.py b/llmpa/models/base.py → llmpa/schemas/base.py
diff --git a/llmpa/models/file.py → llmpa/schemas/file.py b/llmpa/models/file.py → llmpa/schemas/file.py
diff --git a/llmpa/models/user.py → llmpa/schemas/user.py b/llmpa/models/user.py → llmpa/schemas/user.py
diff --git a/llmpa/tests/test_user.py b/llmpa/tests/test_user.py
@@ -13,7 +13,7 @@
 
 from app import create_app
 from db import db, async_session
-from models import User
+from schemas import User
 
 
 class TestUserModel(unittest.IsolatedAsyncioTestCase):