add button to GUI

appointment.py

@@ -2,6 +2,7 @@ import PySimpleGUI as sg
 
 # First the window layout in 2 columns
 from main import start_book
+from pojo.ModeEnum import ModeEnum
 
 KEY_CHOOSE_STORE = "CHOOSE_STORE"
 KEY_START_NUMBER = "KEY_START_NUMBER"
@@ -13,6 +14,9 @@ KEY_PROXY_CC = "KEY_PROXY_CC"
 KEY_FAUBOURG = "KEY_FAUBOURG"
 KEY_GEORGE = "KEY_GEORGE"
 KEY_SEVRES = "KEY_SEVRES"
+KEY_AUTOMATIC = "KEY_AUTOMATIC"
+KEY_MANUAL = "KEY_MANUAL"
+GROUP_MODE = "GROUP_MODE"
 GROUP_STORE = "STORE"
 GROUP_PROXY = "GROUP_PROXY"
 
@@ -36,12 +40,19 @@ proxy_settings_column = [
     [sg.Radio('res(速度)', group_id=GROUP_PROXY, key=KEY_PROXY_RES, default=True)],
     [sg.Radio('cc(稳定)', group_id=GROUP_PROXY, key=KEY_PROXY_CC, default=False)],
 ]
+
+mode_settings_column = [
+    [sg.Text("约会模式")],
+    [sg.Radio('手动', group_id=GROUP_MODE, key=KEY_MANUAL, default=True)],
+    [sg.Radio('自动', group_id=GROUP_MODE, key=KEY_AUTOMATIC, default=False)],
+]
 # ----- Full layout -----
 layout = [
     [
         sg.Column(file_list_column),
         sg.Column(store_settings_column),
-        sg.Column(proxy_settings_column)
+        sg.Column(proxy_settings_column),
+        sg.Column(mode_settings_column)
     ]
 ]
 
@@ -57,6 +68,7 @@ while True:
         end_line = int(values[KEY_END_NUMBER])
         max_workers = int(values[KEY_MAX_WORKERS])
         store_type = 0
+        mode = ModeEnum.MANUAL
         if values[KEY_FAUBOURG]:
             store_type = 1
         elif values[KEY_GEORGE]:
@@ -69,7 +81,11 @@ while True:
             proxy_type = 1
         elif values[KEY_PROXY_RES]:
             proxy_type = 0
-        start_book(start_line, end_line, store_choose_state=store_type, max_workers=max_workers, proxy_type=proxy_type)
+
+        if values[KEY_AUTOMATIC]:
+            mode = ModeEnum.AUTOMATIC
+        start_book(start_line, end_line, store_choose_state=store_type, max_workers=max_workers, proxy_type=proxy_type,
+                   mode=mode)
         # except Exception as error:
         #     print("Not Integer: ")
         #     print(error)