From f31879f985cac5c6de8eccb52b02b20d1ace9636 Mon Sep 17 00:00:00 2001
From: Markus Nyman <markus@nyman.dev>
Date: Thu, 12 Jan 2023 15:05:35 +0200
Subject: [PATCH] Extract menu selection to own function

---
 TimeToTrakt.py | 25 +++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/TimeToTrakt.py b/TimeToTrakt.py
index a475961..4e0a2f8 100644
--- a/TimeToTrakt.py
+++ b/TimeToTrakt.py
@@ -773,8 +773,7 @@ def process_movies():
                     f"({rows_count}/{rows_total}) - Already imported, skipping '{movie_name}'."
                 )
 
-
-def start():
+def menu_selection() -> int:
     # Display a menu selection
     print(">> What do you want to do?")
     print("    1) Import Watch History for TV Shows from TV Time")
@@ -784,34 +783,40 @@ def start():
 
     while True:
         try:
-            menu_selection = input("Enter your menu selection: ")
-            menu_selection = 3 if not menu_selection else int(menu_selection)
+            selection = input("Enter your menu selection: ")
+            selection = 3 if not selection else int(selection)
             break
         except ValueError:
             logging.warning("Invalid input. Please enter a numerical number.")
-        # Check if the input is valid
-    if not 1 <= menu_selection <= 4:
+    # Check if the input is valid
+    if not 1 <= selection <= 4:
         logging.warning("Sorry - that's an unknown menu selection")
         exit()
     # Exit if the 4th option was chosen
-    if menu_selection == 4:
+    if selection == 4:
         logging.info("Exiting as per user's selection.")
         exit()
+
+    return selection
+
+
+def start():
+    selection = menu_selection()
     # Create the initial authentication with Trakt, before starting the process
     if init_trakt_auth():
         # Start the process which is required
-        if menu_selection == 1:
+        if selection == 1:
             # Invoke the method which will import episodes which have been watched
             # from TV Time into Trakt
             logging.info("Processing watched shows.")
             process_watched_shows()
             # TODO: Add support for followed shows
-        elif menu_selection == 2:
+        elif selection == 2:
             # Invoke the method which will import movies which have been watched
             # from TV Time into Trakt
             logging.info("Processing movies.")
             process_movies()
-        elif menu_selection == 3:
+        elif selection == 3:
             # Invoke both the episodes and movies import methods
             logging.info("Processing both watched shows and movies.")
             process_watched_shows()