allow uid to be None when init

building_dao/building_generic.py

@@ -7,7 +7,6 @@ import requests
 from requests.exceptions import HTTPError
 
 from retrying import retry
-from simplejson import JSONDecodeError
 from .settings import Settings
 
 MAX_RETRIES = 5  # max retries number
@@ -21,21 +20,30 @@ class BuildingGeneric:
 
     """Stores the methods used to receive or update the building info from ES doc"""
 
-    def __init__(self, uid: str, **kwargs) -> None:
+    def __init__(self, uid: str = None, **kwargs) -> None:
 
         staging = kwargs.get("staging", False)
         registry = kwargs.get("registry")
 
         self.logger = logging.getLogger(self.__class__.__name__)
         self._uid = uid
+        self._building = None
 
         self.settings = Settings(staging=staging, registry=registry)
-        self.fetch()
+
+        if self._uid:
+            self.fetch()
 
     def __getitem__(self, item):
         return self._building.get(item)
 
     # ----------------- Building level ---------------------
+    def post_building(self, payload) -> None:
+        """Post a new building"""
+        res = self._req_with_retries("post", join(self.settings.base_url), json=payload)
+        self._uid = res["id"]
+        self.fetch()
+
     def fetch(self) -> None:
         """Fetch the building document"""
         self._building = self._req_with_retries("get", join(self.settings.base_url, self._uid))