67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
|
"""Rebuild word_translations table from vocabulary
|
|||
|
|
|
|||
|
|
Очищает word_translations и заново заполняет из vocabulary,
|
|||
|
|
разбивая переводы по запятой/точке с запятой на отдельные записи.
|
|||
|
|
|
|||
|
|
Revision ID: 20251210_rebuild_translations
|
|||
|
|
Revises: 20251210_is_premium
|
|||
|
|
Create Date: 2025-12-10
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
from typing import Sequence, Union
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
from alembic import op
|
|||
|
|
import sqlalchemy as sa
|
|||
|
|
|
|||
|
|
|
|||
|
|
# revision identifiers, used by Alembic.
|
|||
|
|
revision: str = '20251210_rebuild_translations'
|
|||
|
|
down_revision: Union[str, None] = '20251210_is_premium'
|
|||
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|||
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def upgrade() -> None:
|
|||
|
|
# Получаем connection для выполнения SQL
|
|||
|
|
connection = op.get_bind()
|
|||
|
|
|
|||
|
|
# 1. Очищаем таблицу word_translations
|
|||
|
|
connection.execute(sa.text("DELETE FROM word_translations"))
|
|||
|
|
|
|||
|
|
# 2. Получаем все слова из vocabulary
|
|||
|
|
result = connection.execute(sa.text(
|
|||
|
|
"SELECT id, word_translation FROM vocabulary WHERE word_translation IS NOT NULL AND word_translation != ''"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
# 3. Для каждого слова разбиваем перевод и вставляем
|
|||
|
|
for row in result:
|
|||
|
|
vocab_id = row[0]
|
|||
|
|
translation_str = row[1]
|
|||
|
|
|
|||
|
|
if not translation_str:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# Разбиваем по запятой или точке с запятой
|
|||
|
|
parts = re.split(r'[,;]\s*', translation_str)
|
|||
|
|
translations = [p.strip() for p in parts if p.strip()]
|
|||
|
|
|
|||
|
|
for i, translation in enumerate(translations):
|
|||
|
|
is_primary = (i == 0) # Первый перевод - основной
|
|||
|
|
connection.execute(
|
|||
|
|
sa.text(
|
|||
|
|
"INSERT INTO word_translations (vocabulary_id, translation, is_primary, created_at) "
|
|||
|
|
"VALUES (:vocab_id, :translation, :is_primary, CURRENT_TIMESTAMP)"
|
|||
|
|
),
|
|||
|
|
{"vocab_id": vocab_id, "translation": translation, "is_primary": is_primary}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print(f"Word translations rebuilt successfully")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def downgrade() -> None:
|
|||
|
|
# При откате просто очищаем таблицу
|
|||
|
|
# (восстановить старые данные невозможно)
|
|||
|
|
connection = op.get_bind()
|
|||
|
|
connection.execute(sa.text("DELETE FROM word_translations"))
|