From 27616867f5420d38316f6fcaebe51eab6b2c2d74 Mon Sep 17 00:00:00 2001
From: Cyberes <cyberes@evulid.cc>
Date: Mon, 8 Apr 2024 00:46:12 -0600
Subject: [PATCH] move stuff under main()

---
 main.py | 41 ++++++++++++++++-------------------------
 1 file changed, 16 insertions(+), 25 deletions(-)

diff --git a/main.py b/main.py
index a4e7440..59767d6 100644
--- a/main.py
+++ b/main.py
@@ -22,33 +22,20 @@ SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
 logging.basicConfig()
 logger = logging.getLogger('MatrixGPT')
 
-parser = argparse.ArgumentParser(description='MatrixGPT Bot')
-parser.add_argument('--config', default=Path(SCRIPT_DIR, 'config.yaml'), help='Path to config.yaml if it is not located next to this executable.')
-args = parser.parse_args()
 
-# Load config
-args.config = Path(args.config)
-if not args.config.exists():
-    logger.critical('Config file does not exist:', args.config)
-    sys.exit(1)
+async def main(args):
+    args.config = Path(args.config)
+    if not args.config.exists():
+        logger.critical('Config file does not exist:', args.config)
+        sys.exit(1)
 
-global_config.load(args.config)
-try:
-    global_config.validate()
-except SchemeValidationError as e:
-    logger.critical(f'Config validation error: {e}')
-    sys.exit(1)
+    global_config.load(args.config)
+    try:
+        global_config.validate()
+    except SchemeValidationError as e:
+        logger.critical(f'Config validation error: {e}')
+        sys.exit(1)
 
-
-def retry(msg=None):
-    if msg:
-        logger.warning(f'{msg}, retrying in 15s...')
-    else:
-        logger.warning(f'Retrying in 15s...')
-    time.sleep(15)
-
-
-async def main():
     if global_config['logging']['log_level'] == 'info':
         log_level = logging.INFO
     elif global_config['logging']['log_level'] == 'debug':
@@ -148,9 +135,13 @@ async def main():
 
 
 if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='MatrixGPT Bot')
+    parser.add_argument('--config', default=Path(SCRIPT_DIR, 'config.yaml'), help='Path to config.yaml if it is not located next to this executable.')
+    args = parser.parse_args()
+
     while True:
         try:
-            asyncio.run(main())
+            asyncio.run(main(args))
         except KeyboardInterrupt:
             os.kill(os.getpid(), signal.SIGTERM)
         except Exception: