Spaces:
Running
Running
File size: 1,434 Bytes
48ec4db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from app.backend.controllers.base_controller import engine
from app.backend.models.messages import Message
from app.backend.models.base_model import Base
from app.backend.models.chats import Chat
from app.backend.models.users import User
from app.settings import logger
from sqlalchemy import inspect
async def table_exists(name: str) -> bool:
async with engine.begin() as conn:
inspector = inspect(conn)
tables = await conn.run_sync(inspector.get_table_names)
return name in tables
async def create_tables() -> None:
await logger.info("Start creating tables")
async with engine.begin() as conn:
await conn.run_sync(Base.metadata.create_all)
await logger.info("End creating tables")
async def drop_tables() -> None:
await logger.info("Start dropping tables")
async with engine.begin() as conn:
await conn.run_sync(Message.__table__.drop)
await conn.run_sync(Chat.__table__.drop)
await conn.run_sync(User.__table__.drop)
await logger.info("End dropping tables")
async def automigrate() -> None:
await logger.info("Start migration")
try:
await drop_tables()
except Exception as e:
await logger.error("Error during drop_tables: " + str(e))
try:
await create_tables()
except Exception as e:
await logger.error("Error during create_tables: " + str(e))
await logger.info("End migration\n\n")
|