feat: add config and models for data centers

Author: Pierre Segonne

config/data_centers/aws.json

@@ -0,0 +1,16 @@
+{
+    "aws-us-east-1": {
+        "displayName": "us-east-1 (Virginia)",
+        "lonlat": [37.3382, -76.7483],
+        "provider": "Amazon Web Services",
+        "region": "us-east-1",
+        "zoneKey": "US-MIDA-PJM"
+    },
+    "aws-us-east-2": {
+        "displayName": "us-east-2 (Ohio)",
+        "lonlat": [39.9526, -82.9959],
+        "provider": "Amazon Web Services",
+        "region": "us-east-2",
+        "zoneKey": "US-MIDW-MISO"
+    }
+}

config/data_centers/gcp.json

@@ -0,0 +1,16 @@
+{
+    "gcp-europe-west1": {
+        "displayName": "europe-west1 (Belgium)",
+        "lonlat": [50.8503, 4.3517],
+        "provider": "Google Cloud Platform",
+        "region": "europe-west1",
+        "zoneKey": "BE"
+    },
+    "gcp-us-central1": {
+        "displayName": "us-central1 (Iowa)",
+        "lonlat": [41.5908, -93.6208],
+        "provider": "Google Cloud Platform",
+        "region": "us-central1",
+        "zoneKey": "US-MIDW-MISO"
+    }
+}

electricitymap/contrib/config/__init__.py

@@ -6,6 +6,7 @@
 
 from electricitymap.contrib.config.co2eq_parameters import generate_co2eq_parameters
 from electricitymap.contrib.config.reading import (
+    read_data_centers_config,
     read_defaults,
     read_exchanges_config,
     read_zones_config,
@@ -24,6 +25,7 @@
 ZONES_CONFIG = read_zones_config(CONFIG_DIR)
 RETIRED_ZONES_CONFIG = read_zones_config(CONFIG_DIR, retired=True)
 EXCHANGES_CONFIG = read_exchanges_config(CONFIG_DIR)
+DATA_CENTERS_CONFIG = read_data_centers_config(CONFIG_DIR)
 
 EU_ZONES = [
     "AT",

electricitymap/contrib/config/model.py

@@ -17,6 +17,7 @@
     EXCHANGES_CONFIG,
     ZONE_NEIGHBOURS,
     ZONES_CONFIG,
+    DATA_CENTERS_CONFIG,
 )
 from electricitymap.contrib.config.types import Point
 from electricitymap.contrib.lib.types import ZoneKey
@@ -340,3 +341,23 @@ def _load_config_model() -> ConfigModel:
 CO2EQ_CONFIG_MODEL = CO2eqConfigModel(
     direct=CO2EQ_PARAMETERS_DIRECT, lifecycle=CO2EQ_PARAMETERS_LIFECYCLE
 )
+
+
+class DataCenter(StrictBaseModel):
+    displayName: str
+    lonlat: tuple[float, float] | None
+    provider: str
+    region: str
+    status: str
+    zoneKey: ZoneKey
+
+
+class DataCenters(StrictBaseModel):
+    dataCenters: dict[str, DataCenter]
+
+
+DATA_CENTERS_CONFIG_MODEL = DataCenters(dataCenters=DATA_CENTERS_CONFIG)
+
+
+if __name__ == "__main__":
+    print(DATA_CENTERS_CONFIG_MODEL)

electricitymap/contrib/config/reading.py

@@ -1,4 +1,5 @@
 from typing import Any
+import json
 
 from ruamel.yaml import YAML
 
@@ -38,3 +39,10 @@ def read_exchanges_config(config_dir) -> dict[ZoneKey, Any]:
         with open(exchange_path, encoding="utf-8") as file:
             exchanges_config[exchange_key] = yaml.load(file)
     return exchanges_config
+
+def read_data_centers_config(config_dir) -> dict[str, Any]:
+    data_centers_config = {}
+    for data_center_path in config_dir.joinpath("data_centers").glob("*.json"):
+        with open(data_center_path, encoding="utf-8") as file:
+            data_centers_config[data_center_path.stem] = json.load(file)
+    return data_centers_config