61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import os
|
|
|
|
from database.db import BotDB
|
|
|
|
# Получаем текущую директорию
|
|
current_dir = os.path.dirname(__file__)
|
|
|
|
# Переходим на уровень выше
|
|
parent_dir = os.path.dirname(current_dir)
|
|
|
|
BotDB = BotDB(parent_dir, 'database/tg-bot-database')
|
|
|
|
|
|
def get_filename():
|
|
"""Возвращает имя файла без расширения."""
|
|
filename = os.path.basename(__file__)
|
|
filename = os.path.splitext(filename)[0]
|
|
return filename
|
|
|
|
|
|
def main():
|
|
# Проверка версии миграций
|
|
current_version = BotDB.get_current_version() # Добавьте функцию для получения версии
|
|
|
|
# Выполнение миграций и проверка последней версии
|
|
if current_version < 2:
|
|
# Скрипты миграции
|
|
create_table_sql_1 = """
|
|
CREATE TABLE IF NOT EXISTS "post_from_telegram_suggest"
|
|
(
|
|
message_id INTEGER not null,
|
|
text TEXT,
|
|
helper_text_message_id INTEGER,
|
|
author_id INTEGER,
|
|
created_at TEXT
|
|
);
|
|
"""
|
|
create_table_sql_2 = """
|
|
CREATE TABLE IF NOT EXISTS message_link_to_content (
|
|
post_id INTEGER NOT NULL,
|
|
message_id INTEGER NOT NULL
|
|
);
|
|
"""
|
|
create_table_sql_3 = """
|
|
CREATE TABLE IF NOT EXISTS content_post_from_telegram (
|
|
message_id INTEGER NOT NULL,
|
|
content_name TEXT NOT NULL
|
|
);
|
|
"""
|
|
# Применение миграции
|
|
BotDB.create_table(create_table_sql_1)
|
|
BotDB.create_table(create_table_sql_2)
|
|
BotDB.create_table(create_table_sql_3)
|
|
filename = get_filename()
|
|
|
|
BotDB.update_version(2, filename)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|