Changed how migration behaves

main.py

@@ -1,6 +1,7 @@
 import asyncio
 import contextlib
 import logging.config
+from argparse import ArgumentParser
 from logging import Logger
 from os import getpid, makedirs
 from pathlib import Path
@@ -10,6 +11,7 @@ from libbot.utils import config_get
 
 from classes.pycord_bot import PycordBot
 from modules.logging_utils import get_logging_config, get_logger
+from modules.migrator import migrate_database
 from modules.scheduler import scheduler
 
 makedirs(Path("logs/"), exist_ok=True)
@@ -18,6 +20,19 @@ logging.config.dictConfig(get_logging_config())
 
 logger: Logger = get_logger(__name__)
 
+# Declare the parser that retrieves the command line arguments
+parser: ArgumentParser = ArgumentParser(
+    prog="QuizBot"
+)
+
+# Add a switch argument --migrate to be parsed...
+parser.add_argument("--migrate", action="store_true")
+# parser.add_argument("--downgrade", action="store_true")
+# parser.add_argument("--confirm", action="store_true")
+
+# ...and parse the arguments we added
+args = parser.parse_args()
+
 # Try to import the module that improves performance
 # and ignore errors when module is not installed
 with contextlib.suppress(ImportError):
@@ -27,6 +42,21 @@ with contextlib.suppress(ImportError):
 
 
 async def main():
+    # Perform migration if command line argument was provided
+    if args.migrate:
+        migrate_database()
+
+    # if args.downgrade:
+    #     if not args.confirm:
+    #         logger.warning(
+    #             "Argument --downgrade has been provided but this operation may include irreversible changes and destroy existing data in both database and configuration. It's not recommended to do this without backing everything up. Retry this command providing an additional argument --confirm to accept the risks and explicitly execute the downgrade."
+    #         )
+    #         exit(1)
+    #
+    #     logger.info("Downgrading...")
+    #     downgrade_database()
+    #     exit()
+
     bot = PycordBot(scheduler=scheduler)
 
     bot.load_extension("cogs")