Added migrations

main.py

@@ -1,9 +1,13 @@
 import contextlib
 import logging
+from argparse import ArgumentParser
 from os import getpid
+from pathlib import Path
 
+from libbot import sync
 from libbot.pyrogram.classes import PyroClient
 
+from modules.migrator import migrate_database
 from modules.scheduler import scheduler
 
 # Uncomment this and the line below client declaration
@@ -21,6 +25,16 @@ logging.basicConfig(
 
 logger = logging.getLogger(__name__)
 
+parser = ArgumentParser(
+    prog="__name__",
+    description="__description__",
+)
+
+# Remove if no database is being used
+parser.add_argument("--migrate", action="store_true")
+
+args = parser.parse_args()
+
 with contextlib.suppress(ImportError):
     import uvloop
 
@@ -28,9 +42,15 @@ with contextlib.suppress(ImportError):
 
 
 def main():
-    client = PyroClient(scheduler=scheduler)
+    client = PyroClient(
+        scheduler=scheduler, commands_source=sync.json_read(Path("commands.json"))
+    )
     # Conversation(client)
 
+    # Remove if no database is being used
+    if args.migrate:
+        migrate_database()
+
     try:
         client.run()
     except KeyboardInterrupt: