60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
"""
|
|
Автоматические миграции БД для max_bot.
|
|
|
|
При старте приложения:
|
|
1. Создаёт таблицу _applied_migrations (если не существует)
|
|
2. Проверяет какие SQL-файлы уже применены
|
|
3. Применяет новые миграции по порядку
|
|
"""
|
|
import os
|
|
import logging
|
|
from sqlalchemy import text
|
|
from app.database import engine
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
MIGRATIONS_DIR = os.path.join(os.path.dirname(__file__), "migrations")
|
|
|
|
|
|
async def run_migrations():
|
|
"""Запуск всех неприменённых миграций из директории migrations/."""
|
|
async with engine.begin() as conn:
|
|
# Создаём таблицу для отслеживания миграций
|
|
await conn.execute(text("""
|
|
CREATE TABLE IF NOT EXISTS _applied_migrations (
|
|
id SERIAL PRIMARY KEY,
|
|
filename VARCHAR(255) UNIQUE NOT NULL,
|
|
applied_at TIMESTAMP DEFAULT NOW()
|
|
)
|
|
"""))
|
|
|
|
# Получаем список уже применённых миграций
|
|
result = await conn.execute(text("SELECT filename FROM _applied_migrations"))
|
|
applied = {row[0] for row in result.all()}
|
|
|
|
# Сканируем директорию миграций
|
|
if not os.path.isdir(MIGRATIONS_DIR):
|
|
logger.info("Директория migrations/ не найдена, пропускаем")
|
|
return
|
|
|
|
sql_files = sorted(f for f in os.listdir(MIGRATIONS_DIR) if f.endswith(".sql"))
|
|
|
|
for filename in sql_files:
|
|
if filename in applied:
|
|
continue
|
|
|
|
filepath = os.path.join(MIGRATIONS_DIR, filename)
|
|
with open(filepath, "r", encoding="utf-8") as f:
|
|
sql = f.read()
|
|
|
|
logger.info(f"Применение миграции: {filename}")
|
|
try:
|
|
await conn.execute(text(sql))
|
|
await conn.execute(
|
|
text("INSERT INTO _applied_migrations (filename) VALUES (:fn)"),
|
|
{"fn": filename}
|
|
)
|
|
logger.info(f"Миграция применена: {filename}")
|
|
except Exception as e:
|
|
logger.warning(f"Миграция {filename} уже применена или ошибка: {e}")
|