From 905175c210c3eab4214710d9b4d4055ef30fc24c Mon Sep 17 00:00:00 2001
From: Dan Albert <dan@gingerhq.net>
Date: Fri, 11 Dec 2020 01:14:51 -0800
Subject: [PATCH] Add --invert to the game generator.

---
 qt_ui/main.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/qt_ui/main.py b/qt_ui/main.py
index 2041aa4c..09620098 100644
--- a/qt_ui/main.py
+++ b/qt_ui/main.py
@@ -128,11 +128,17 @@ def parse_args() -> argparse.Namespace:
         help="Automate bluefor procurement."
     )
 
+    new_game.add_argument(
+        "--inverted", action="store_true",
+        help="Invert the campaign."
+    )
+
     return parser.parse_args()
 
 
 def create_game(campaign_path: Path, blue: str, red: str,
-                supercarrier: bool, auto_procurement: bool) -> Game:
+                supercarrier: bool, auto_procurement: bool,
+                inverted: bool) -> Game:
     campaign = Campaign.from_json(campaign_path)
     generator = GameGenerator(
         blue, red, campaign.load_theater(),
@@ -147,7 +153,7 @@ def create_game(campaign_path: Path, blue: str, red: str,
             player_budget=DEFAULT_BUDGET,
             enemy_budget=DEFAULT_BUDGET,
             midgame=False,
-            inverted=False,
+            inverted=inverted,
             no_carrier=False,
             no_lha=False,
             no_player_navy=False,
@@ -168,7 +174,8 @@ def main():
     args = parse_args()
     if args.subcommand == "new-game":
         game = create_game(args.campaign, args.blue, args.red,
-                           args.supercarrier, args.auto_procurement)
+                           args.supercarrier, args.auto_procurement,
+                           args.inverted)
 
     run_ui(game)