script: +update_cache.

Author: saitewasreset

script/load_mission/load_mission.py

@@ -4,6 +4,39 @@
 import re
 import json
 
+def print_endpoint(endpoint: str, name: str):
+    print("update {} endpoint: {}".format(name, endpoint))
+
+def update_cache(endpoint: str, name: str):
+    print("Updating {} cache...".format(name))
+    r = requests.get(endpoint)
+    try:
+        res = r.json()
+        if res["code"] != 200:
+            print("Server returned an error:  ", res)
+            input("Press enter to exit...")
+            sys.exit(1)
+        else:
+            print("Success! time: {}ms".format(res["data"]["time_ms"]))
+    except json.JSONDecodeError:
+        print("Invalid response from server: ", r.text)
+        input("Press enter to exit...")
+        sys.exit(1)
+
+def update_all_cache(admin_endpoint: str):
+    update_mission_kpi_endpoint = "{}/update_mission_kpi".format(admin_endpoint)
+    update_endpoint = "{}/update_essential".format(admin_endpoint)
+    update_damage_endpoint = "{}/update_damage".format(admin_endpoint)
+    update_general_endpoint = "{}/update_general".format(admin_endpoint)
+
+    update_list = [(update_mission_kpi_endpoint, "mission kpi"), (update_endpoint, "essential"), (update_damage_endpoint, "damage"), (update_general_endpoint, "general")]
+
+    for endpoint, name in update_list:
+        print_endpoint(endpoint, name)
+    for endpoint, name in update_list:
+        update_cache(endpoint, name)
+
+
 source_addr = "https://github.com/saitewasreset/DRG_MissionMonitor"
 
 print("Mission Monitor: load_mission")
@@ -37,14 +70,10 @@
 
 already_uploaded_endpoint = "{}/mission_list".format(admin_endpoint)
 upload_endpoint = "{}/load_mission".format(admin_endpoint)
-update_endpoint = "{}/update_essential".format(admin_endpoint)
-update_damage_endpoint = "{}/update_damage".format(admin_endpoint)
 
 print("log path: {}".format(log_path))
 print("already uploaded endpoint: {}".format(already_uploaded_endpoint))
 print("upload endpoint: {}".format(upload_endpoint))
-print("update endpoint: {}".format(update_endpoint))
-print("update damage endpoint: {}".format(update_damage_endpoint))
 
 to_load_list = []
 
@@ -102,35 +131,7 @@
     input("Press enter to exit...")
     sys.exit(1)
 
-print("Updating essential cache...")
-r = requests.get(update_endpoint)
-try:
-    res = r.json()
-    if res["code"] != 200:
-        print("Server returned an error:  ", res)
-        input("Press enter to exit...")
-        sys.exit(1)
-    else:
-        print("Success!")
-except json.JSONDecodeError:
-    print("Invalid response from server: ", r.text)
-    input("Press enter to exit...")
-    sys.exit(1)
-
-print("Updating damage cache...")
-r = requests.get(update_damage_endpoint)
-try:
-    res = r.json()
-    if res["code"] != 200:
-        print("Server returned an error:  ", res)
-        input("Press enter to exit...")
-        sys.exit(1)
-    else:
-        print("Success!")
-except json.JSONDecodeError:
-    print("Invalid response from server: ", r.text)
-    input("Press enter to exit...")
-    sys.exit(1)
+update_all_cache(admin_endpoint)
 
 print("Rock and stone!")
 input("Press enter to exit...")