handle not existing attr

SpareCores · Dec 27, 2023 · 786e7be · 786e7be
1 parent 75d48ea
commit 786e7be
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/sc_crawler/schemas.py b/sc_crawler/schemas.py
@@ -41,12 +41,18 @@ class Vendor(BaseModel):
     @computed_field
     @property
     def datacenters(self) -> int:
-        return len(self._datacenters)
+        if hasattr(self, "_datacenters"):
+            return len(self._datacenters)
+        else:
+            return 0
 
     @computed_field
     @property
     def zones(self) -> int:
-        return sum([datacenter.zones for datacenter in self._datacenters])
+        if hasattr(self, "_datacenters"):
+            return sum([datacenter.zones for datacenter in self._datacenters])
+        else:
+            return 0
 
     # private attributes
     _methods: ImportString[ModuleType] = PrivateAttr()