get zones from datacenters

sc_crawler/schemas.py

@@ -3,6 +3,7 @@
 
 from .location import Location
 
+from collections import ChainMap
 from importlib import import_module
 from types import ModuleType
 from typing import Dict, List, Literal, Optional, ForwardRef
@@ -54,9 +55,6 @@ def __init__(self, **kwargs):
             raise NotImplementedError("Unsupported vendor")
             pass
 
-    def get_instance_types(self):
-        return self._methods.get_instance_types()
-
     def get_datacenters(self, identifiers: [str] = None):
         """Get datacenters of the vendor.
 
@@ -74,6 +72,18 @@ def get_datacenters(self, identifiers: [str] = None):
             ]
         return datacenters
 
+    def get_zones(self):
+        """Get zones of the vendor."""
+        # make sure datacenters filled in
+        self._methods.get_datacenters(self)
+        # unlist
+        self._zones = dict(
+            ChainMap(*[datacenter._zones for datacenter in self._datacenters])
+        )
+
+    def get_instance_types(self):
+        return self._methods.get_instance_types()  # TODO
+
 
 class Datacenter(BaseModel):
     identifier: str

sc_crawler/vendors/aws.py

@@ -247,7 +247,7 @@ def get_datacenters(vendor, *args, **kwargs):
     ]
 
     # make it easier to access by region name
-    #datacenters = {datacenter.identifier: datacenter for datacenter in datacenters}
+    # datacenters = {datacenter.identifier: datacenter for datacenter in datacenters}
 
     # add zones
     for datacenter in datacenters: