diff --git a/main.py b/main.py index 6aa78c8..2cb0e19 100644 --- a/main.py +++ b/main.py @@ -35,14 +35,16 @@ with contextlib.suppress(ImportError): def main(): + if args.migrate: + migrate_database() + logger.info("Migration finished. Exiting...") + exit() + client = PyroClient( scheduler=scheduler, commands_source=sync.json_read(Path("commands.json")) ) Conversation(client) - if args.migrate: - migrate_database() - try: client.run() except KeyboardInterrupt: diff --git a/migrations/202405261500.py b/migrations/202405261500.py new file mode 100644 index 0000000..82521e2 --- /dev/null +++ b/migrations/202405261500.py @@ -0,0 +1,16 @@ +from libbot import sync +from mongodb_migrations.base import BaseMigration + + +class Migration(BaseMigration): + def upgrade(self): + sync.config_set("update_checker", True) + sync.config_set( + "url_updater", + "https://git.end-play.xyz/api/v1/repos/GarbageReminder/TelegramBot/releases/latest", + "strings", + ) + + def downgrade(self): + sync.config_delete("update_checker", missing_ok=True) + sync.config_delete("url_updater", "strings")