diff --git a/olgram/migrations/custom.py b/olgram/migrations/custom.py index da1b3d4..87a8a20 100644 --- a/olgram/migrations/custom.py +++ b/olgram/migrations/custom.py @@ -1,7 +1,8 @@ """Наши собственные миграции, которые нельзя описать на языке SQL и с которыми не справится TortoiseORM/Aerich""" +import aioredis from tortoise import transactions, Tortoise -from olgram.settings import TORTOISE_ORM +from olgram.settings import TORTOISE_ORM, ServerSettings from olgram.models.models import MetaInfo, Bot import logging @@ -22,8 +23,21 @@ async def upgrade_1(): await meta_info.save() logging.info("done") + +async def upgrade_2(): + """Отменяем малый TTL для старых сообщений""" + + 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()) + + # Не забудь добавить миграцию в этот лист! -_migrations = [upgrade_1] +_migrations = [upgrade_1, upgrade_2] async def migrate():