Some database changes, started to add threat rings

scripts/python/addRanges.py

@@ -0,0 +1,63 @@
+import sys
+import json
+import inspect
+import difflib
+from slpp import slpp as lua
+
+SEARCH_FOLDER = "D:\\Eagle Dynamics\\DCS World OpenBeta"
+
+sys.path.append("..\..\..\dcs-master\dcs-master")
+
+from dcs.weapons_data import Weapons
+from dcs.planes import *
+from dcs.helicopters import *
+        
+# The database file on which to operate is the first standard argument of the call
+if len(sys.argv) > 1:
+    if (sys.argv[1] == "aircraft"):
+        filename = '..\\..\\client\\public\\databases\\units\\aircraftdatabase.json' 
+    elif (sys.argv[1] == "helicopter"):
+        filename = '..\\..\\client\\public\\databases\\units\\helicopterdatabase.json' 
+    elif (sys.argv[1] == "groundunit"):
+        filename = '..\\..\\client\\public\\databases\\units\\groundunitdatabase.json' 
+    elif (sys.argv[1] == "navyunit"):
+        filename = '..\\..\\client\\public\\databases\\units\\navyunitdatabase.json' 
+
+    # Loads the database 
+    with open(filename) as f:
+        database = json.load(f)
+        for unit_name in database:
+            database[unit_name]["enabled"] = True
+    
+    # Loop on all the units in the database
+    for unit_name in database:
+        try:
+            # Get the pydcs Python class for the unit
+            if (sys.argv[1] == "aircraft"):
+                unitmap = plane_map 
+            elif (sys.argv[1] == "helicopter"):
+                unitmap = helicopter_map 
+            elif (sys.argv[1] == "groundunit"):
+                unitmap = vehicle_map 
+            elif (sys.argv[1] == "navyunit"):
+                unitmap = ship_map 
+            lowercase_keys = [key.lower() for key in unitmap.keys()]
+            res = difflib.get_close_matches(unit_name.lower(), lowercase_keys)
+            if len(res) > 0:
+                found_name = list(unitmap.keys())[lowercase_keys.index(res[0])]
+                cls = unitmap[found_name]
+            else:
+                print(f"Warning, could not find {unit_name} in classes list. Skipping...")
+                continue
+
+        except Exception as e:
+            print(f"Could not find data for aircraft of type {unit_name}: {e}, skipping...")
+
+    # Dump everything in the database
+    with open(filename, "w") as f:
+        json.dump(database, f, indent=2)
+
+    # Done!
+    print("Done!")
+
+    

scripts/python/temp.py

@@ -1,15 +0,0 @@
-import json
-import difflib
-
-countries = ['USA', 'GRG', 'GER', 'DZA', 'FRA', 'CAN', 'AUS', 'UKR', 'ITA', 'GRC', 'SPN', 'RUS', 'NETH', 'DEN', 'TUR', 'UK', 'BEL', 'ISR', 'NOR', 'JPN', 'ARE', 'QAT', 'IND', 'SAU', 'EGY', 'KOR', 'HND', 'CHL', 'BLUE', 'AUSAF', 'RED', 'VNM', 'SVK', 'SDN', 'GDR', 'JOR', 'PER', 'CHN', 'IDN', 'PHL', 'BOL', 'MAR', 'YEM', 'KWT', 'SUI', 'GHA', 'CYP', 'BHR', 'YUG', 'CZE', 'KAZ', 'AUT', 'HUN', 'MYS', 'ROU', 'THA', 'LBN', 'FIN', 'PRT', 'OMN', 'MEX', 'IRQ', 'BRA', 'SWE', 'NZG', 'CUB', 'INS', 'RSO', 'RSA', 'HRV', 'ABH', 'ARG', 'LBY', 'PRK', 'VEN', 'TUN', 'IRN', 'ETH', 'BLR', 'SUN', 'BGR', 'PAK', 'NGA', 'POL', 'SVN', 'SYR', 'SRB', 'UN', 'RSI', 'SPA', 'ECU', '', 'USAF', 'hide', 'EGP', 'LIB']
-
-with open('C:\\Users\\dpass\\Documents\\DCSOlympus\\client\\public\\images\\nations\\codes.json', "r") as f:
-    codes = json.load(f)
-
-    for country in countries:
-        keys = difflib.get_close_matches(country, codes.keys(), cutoff=.35)
-        if len(keys) > 0:
-            codes[keys[0]]["liveryCodes"].append(country)
-
-with open('C:\\Users\\dpass\\Documents\\DCSOlympus\\client\\public\\images\\nations\\codes.json', "w") as f:
-    json.dump(codes, f)