From 986645d7e1eb7f02b2ea257df877279c3c1eabdf Mon Sep 17 00:00:00 2001
From: Jack Roberts <jroberts@turing.ac.uk>
Date: Sun, 1 Oct 2023 17:36:34 +0100
Subject: [PATCH] [#638] fix ability to skip confirmation step in set lineup

---
 airsenal/scripts/make_transfers.py | 2 +-
 airsenal/scripts/set_lineup.py     | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/airsenal/scripts/make_transfers.py b/airsenal/scripts/make_transfers.py
index 347e5384..ae8dca3d 100644
--- a/airsenal/scripts/make_transfers.py
+++ b/airsenal/scripts/make_transfers.py
@@ -350,7 +350,7 @@ def main():
     confirm = args.confirm or False
     try:
         make_transfers(args.fpl_team_id, confirm)
-        set_lineup(args.fpl_team_id)
+        set_lineup(args.fpl_team_id, skip_check=confirm)
     except Exception as e:
         raise Exception(
             "Something went wrong when making transfers. Check your team and make "
diff --git a/airsenal/scripts/set_lineup.py b/airsenal/scripts/set_lineup.py
index deefe4a8..4a4730c7 100644
--- a/airsenal/scripts/set_lineup.py
+++ b/airsenal/scripts/set_lineup.py
@@ -84,7 +84,9 @@ def make_squad_transfers(squad: Squad, priced_transfers: List[dict]) -> None:
 
 
 def set_lineup(
-    fpl_team_id: Optional[int] = None, verbose: Optional[bool] = False
+    fpl_team_id: Optional[int] = None,
+    verbose: Optional[bool] = False,
+    skip_check: bool = False,
 ) -> None:
     """
     Retrieve the latest lineup and apply the latest prediction to it.
@@ -104,7 +106,7 @@ def set_lineup(
 
     squad.optimize_lineup(NEXT_GAMEWEEK, get_latest_prediction_tag())
 
-    if check_proceed(squad):
+    if check_proceed(squad) and not skip_check:
         payload = build_lineup_payload(squad)
         fetcher.post_lineup(payload)
 
@@ -112,9 +114,10 @@ def set_lineup(
 def main():
     parser = argparse.ArgumentParser("Set the starting 11 and captain")
     parser.add_argument("--fpl_team_id", help="ID of the squad in FPL API", type=int)
+    parser.add_argument("--confirm", help="skip confirmation step", action="store_true")
     args = parser.parse_args()
     try:
-        set_lineup(args.fpl_team_id)
+        set_lineup(args.fpl_team_id, skip_check=args.confirm)
     except Exception as e:
         raise Exception(
             "Something went wrong when setting lineup. Check your lineup manually on "