Update common.py to choose between boundary service (intenal switch)

Author: matamadio

tools/code/common.py

@@ -12,19 +12,35 @@
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 os.makedirs(CACHE_DIR,  exist_ok=True)
 
-# Define the REST API URL
-rest_api_url = "https://services.arcgis.com/iQ1dY19aHwbSDYIF/ArcGIS/rest/services/World_Bank_Global_Administrative_Divisions_VIEW/FeatureServer"
+# Define the REST API URLs
+rest_api_url_view = "https://services.arcgis.com/iQ1dY19aHwbSDYIF/ArcGIS/rest/services/World_Bank_Global_Administrative_Divisions_VIEW/FeatureServer"
+rest_api_url_new = "https://services.arcgis.com/iQ1dY19aHwbSDYIF/ArcGIS/rest/services/World_Bank_Global_Administrative_Divisions/FeatureServer"
+
+# Active service selection - Set to True to use new service, False for VIEW service
+USE_NEW_SERVICE = True
+
+# Set active URL based on service selection
+rest_api_url = rest_api_url_new if USE_NEW_SERVICE else rest_api_url_view
+
 worldpop_url = "https://data.worldpop.org/GIS/Population/"
 stac_search_url = "https://geoservice.dlr.de/eoc/ogc/stac/v1/search"
 
 
 # Mapping of administrative levels to field names for WB GAD dataset
-adm_field_mapping = {
-    0: {'code': 'HASC_0', 'name': 'NAM_0'},
-    1: {'code': 'HASC_1', 'name': 'NAM_1'},
-    2: {'code': 'HASC_2', 'name': 'NAM_2'},
-    # Add mappings for additional levels as needed
-}
+# Field mapping adjusted based on active service
+if USE_NEW_SERVICE:
+    adm_field_mapping = {
+        0: {'code': 'ISO_A3', 'name': 'NAM_0'},      # New service uses ISO_A3 instead of HASC_0
+        1: {'code': 'ADM1CD_c', 'name': 'NAM_1'},    # New service uses ADM1CD_c instead of HASC_1
+        2: {'code': 'ADM2CD_c', 'name': 'NAM_2'},    # New service uses ADM2CD_c instead of HASC_2
+    }
+else:
+    adm_field_mapping = {
+        0: {'code': 'HASC_0', 'name': 'NAM_0'},
+        1: {'code': 'HASC_1', 'name': 'NAM_1'},
+        2: {'code': 'HASC_2', 'name': 'NAM_2'},
+        # Add mappings for additional levels as needed
+    }
 
 wb_to_region = {
     'AFR': 'AFRICA',        # Sub-Saharan Africa