1
0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00

Fix database

This commit is contained in:
MoonTestUse1
2025-01-07 06:25:57 +06:00
parent 04c29924fb
commit 117f5af0cb
3 changed files with 12 additions and 12 deletions

View File

@@ -1,8 +1,8 @@
"""Import all models for Alembic autogenerate support""" """Import all models for Alembic autogenerate support"""
from app.db.base_class import Base from app.db.base_class import Base # noqa
from app.models.employee import Employee from app.models.employee import Employee # noqa
from app.models.request import Request from app.models.request import Request # noqa
from app.models.token import Token from app.models.token import Token # noqa
# Импортируем все модели, чтобы Alembic мог их обнаружить # Импортируем все модели, чтобы Alembic мог их обнаружить
__all__ = ["Base", "Employee", "Request", "Token"] __all__ = ["Base", "Employee", "Request", "Token"]

View File

@@ -1,10 +1,6 @@
"""Models initialization""" """Models package"""
from ..database import Base, engine
from .employee import Employee from .employee import Employee
from .request import Request from .request import Request
from .token import Token from .token import Token
# Создаем все таблицы __all__ = ["Employee", "Request", "Token"]
Base.metadata.create_all(bind=engine)
__all__ = ['Base', 'Employee', 'Request', 'Token']

View File

@@ -7,10 +7,12 @@ from sqlalchemy.orm import sessionmaker
from sqlalchemy.pool import StaticPool from sqlalchemy.pool import StaticPool
from app.core.test_config import test_settings from app.core.test_config import test_settings
from app.db.base import Base from app.db.base_class import Base
from app.models.employee import Employee
from app.models.request import Request
from app.models.token import Token
from app.main import app from app.main import app
from app.dependencies import get_db from app.dependencies import get_db
from app.models.employee import Employee
from app.utils.security import get_password_hash from app.utils.security import get_password_hash
from app.utils.jwt import create_and_save_token from app.utils.jwt import create_and_save_token
@@ -37,6 +39,8 @@ def engine():
pool_size=5, pool_size=5,
max_overflow=10 max_overflow=10
) )
# Создаем все таблицы перед тестами
Base.metadata.drop_all(bind=engine)
Base.metadata.create_all(bind=engine) Base.metadata.create_all(bind=engine)
return engine return engine