15 lines
579 B
Python
15 lines
579 B
Python
from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
|
|
from dbapi.tables import metadata_obj
|
|
|
|
|
|
class FitnessDatabseMigrator:
|
|
"""Class for performing management operations with database"""
|
|
|
|
def __init__(self, async_engine: AsyncEngine) -> None:
|
|
self.engine = async_engine
|
|
|
|
async def reset_database(self):
|
|
"""Method for dropping all tables and create them from tables metadata"""
|
|
async with self.engine.begin() as conn:
|
|
await conn.run_sync(metadata_obj.drop_all)
|
|
await conn.run_sync(metadata_obj.create_all)
|