57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import os
|
|
import sys
|
|
|
|
# Добавляем путь к корневой директории проекта
|
|
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
|
|
|
|
from database.db import BotDB
|
|
|
|
# Получаем текущую директорию
|
|
current_dir = os.path.dirname(__file__)
|
|
|
|
# Переходим на уровень выше
|
|
parent_dir = os.path.dirname(current_dir)
|
|
|
|
BotDB = BotDB(parent_dir, 'tg-bot-database.db')
|
|
|
|
|
|
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 < 3:
|
|
# Скрипт миграции для создания таблицы our_users
|
|
create_table_sql = """
|
|
CREATE TABLE IF NOT EXISTS "our_users" (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
user_id INTEGER NOT NULL UNIQUE,
|
|
first_name TEXT,
|
|
full_name TEXT,
|
|
username TEXT,
|
|
is_bot BOOLEAN DEFAULT 0,
|
|
language_code TEXT,
|
|
date_added TEXT,
|
|
date_changed TEXT,
|
|
has_stickers BOOLEAN DEFAULT 0
|
|
);
|
|
"""
|
|
|
|
# Применение миграции
|
|
BotDB.create_table(create_table_sql)
|
|
filename = get_filename()
|
|
|
|
BotDB.update_version(3, filename)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|