From e1a7b6309eb9c79f5aacb47d0cf70e0197fa8c67 Mon Sep 17 00:00:00 2001
From: profitroll <vozhd.kk@gmail.com>
Date: Sun, 26 May 2024 22:56:30 +0200
Subject: [PATCH] Closes #49

---
 main.py                    |  8 +++++---
 migrations/202405261500.py | 16 ++++++++++++++++
 2 files changed, 21 insertions(+), 3 deletions(-)
 create mode 100644 migrations/202405261500.py

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")