mirror of
https://github.com/civsocit/olgram.git
synced 2023-07-22 01:29:12 +03:00
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""Наши собственные миграции, которые нельзя описать на языке SQL и с которыми не справится TortoiseORM/Aerich"""
|
||
|
||
import aioredis
|
||
from tortoise import transactions, Tortoise
|
||
from olgram.settings import TORTOISE_ORM, ServerSettings
|
||
from olgram.models.models import MetaInfo, Bot
|
||
import logging
|
||
|
||
|
||
async def upgrade_1():
|
||
"""Шифруем токены"""
|
||
meta_info = await MetaInfo.first()
|
||
if meta_info.version != 0:
|
||
logging.info("skip")
|
||
return
|
||
|
||
async with transactions.in_transaction():
|
||
bots = await Bot.all()
|
||
for bot in bots:
|
||
bot.token = bot.encrypted_token(bot.token)
|
||
await bot.save()
|
||
meta_info.version = 1
|
||
await meta_info.save()
|
||
logging.info("done")
|
||
|
||
|
||
async def upgrade_2():
|
||
"""Отменяем малый TTL для старых сообщений"""
|
||
meta_info = await MetaInfo.first()
|
||
if meta_info.version != 1:
|
||
logging.info("skip")
|
||
return
|
||
|
||
con = await aioredis.create_connection(ServerSettings.redis_path())
|
||
client = aioredis.Redis(con)
|
||
|
||
i, keys = await client.scan()
|
||
for key in keys:
|
||
if not key.startswith(b"thread"):
|
||
await client.pexpire(key, ServerSettings.redis_timeout_ms())
|
||
|
||
meta_info.version = 2
|
||
await meta_info.save()
|
||
logging.info("done")
|
||
|
||
|
||
async def upgrade_3():
|
||
"""start_text и second_text должны быть валидными HTML"""
|
||
import html
|
||
|
||
meta_info = await MetaInfo.first()
|
||
if meta_info.version != 2:
|
||
logging.info("skip")
|
||
return
|
||
|
||
async with transactions.in_transaction():
|
||
bots = await Bot.all()
|
||
for bot in bots:
|
||
if bot.start_text:
|
||
bot.start_text = html.escape(bot.start_text)
|
||
if bot.second_text:
|
||
bot.second_text = html.escape(bot.second_text)
|
||
await bot.save(update_fields=["start_text", "second_text"])
|
||
meta_info.version = 3
|
||
await meta_info.save()
|
||
logging.info("done")
|
||
|
||
|
||
# Не забудь добавить миграцию в этот лист!
|
||
_migrations = [upgrade_1, upgrade_2, upgrade_3]
|
||
|
||
|
||
async def migrate():
|
||
logging.info("Run custom migrations...")
|
||
await Tortoise.init(config=TORTOISE_ORM)
|
||
|
||
for migration in _migrations:
|
||
logging.info(f"Migration {migration.__name__}...")
|
||
await migration()
|