1 files changed, 10 insertions, 2 deletions
diff --git a/src/match_up/__init__.py b/src/match_up/__init__.py
index 1f625a7..3967cb9 100644
--- a/src/match_up/__init__.py
+++ b/src/match_up/__init__.py
@@ -1,9 +1,10 @@
 from flask import Flask
 from dotenv import load_dotenv
+from argparse import ArgumentParser
 from importlib.resources import path
 from match_up.controller import RoundGenerator
 from match_up.model import CourtList, PlayerList
-
+from match_up.utilities import Timer
 
 load_dotenv()
 with path("match_up", "static") as p:
@@ -11,10 +12,17 @@ with path("match_up", "static") as p:
 with path("match_up", "templates") as p:
     template_folder = p
 app = Flask("Match Up! Backend", static_folder=static_folder, template_folder=template_folder)
-rg = RoundGenerator(courts=CourtList(), players=PlayerList())
+rg = RoundGenerator()
 
 
 from match_up import routes
 
 def main():
+    parser = ArgumentParser()
+    parser.add_argument("-t", "--timer", help="Total mins per round", type=float, default=12)
+    parser.add_argument("-c", "--courts", help="Total number of courts", type=int, default=12)
+    args = parser.parse_args()
+    global rg
+    rg.courts = CourtList(total=args.courts)
+    rg.timer = Timer(max_mins=args.timer)
     app.run()