From c09b1e4a30364a9a349d22f227ec3b1ac9f82c28 Mon Sep 17 00:00:00 2001
From: Philipp <philipp@boehm.sh>
Date: Sat, 24 Sep 2022 22:34:00 +0200
Subject: [PATCH] add init_database arg

---
 compose.yml          | 1 +
 docker-entrypoint.sh | 2 +-
 src/main.py          | 8 +++++---
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/compose.yml b/compose.yml
index e02bb5e..9126141 100644
--- a/compose.yml
+++ b/compose.yml
@@ -16,3 +16,4 @@ services:
     volumes:
       - .:/code
     entrypoint: ./code/docker-entrypoint.sh
+    command: init_database
diff --git a/docker-entrypoint.sh b/docker-entrypoint.sh
index 1de11e6..477131a 100755
--- a/docker-entrypoint.sh
+++ b/docker-entrypoint.sh
@@ -4,5 +4,5 @@ echo "Installing requirements..."
 pip install -r /code/requirements.txt
 
 echo "Starting discord bot..."
-python -u /code/src/main.py
+python -u /code/src/main.py $1
 
diff --git a/src/main.py b/src/main.py
index 9b38ee0..5144874 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,11 +1,12 @@
 import os
+import sys
 import interactions
 import database
 
 from dotenv import load_dotenv
 from steamid import SteamID
 
-def main():
+def main(args):
     load_dotenv()
     token = os.getenv('DISCORD_TOKEN')
     host = os.getenv('MYSQL_HOST')
@@ -16,7 +17,8 @@ def main():
     bot = interactions.Client(token=token)
 
     mysql = database.create_connection(host, user, password, db)
-    database.init_test_database(mysql)
+    if args[0] == "init_database":
+        database.init_test_database(mysql)
 
     @bot.command(
             name="reserve_slot",
@@ -59,4 +61,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main(sys.argv[1:])