added save

network_manager/pyramid_manager.py

@@ -11,11 +11,14 @@ class AppPyramidManager:
     TIMEOUT = 30
 
     def __init__(self, base_url, resource_url, api_key, network_manager):
-        self.__url__ = resource_url
+        self.__res_url__ = resource_url
         self.__api_key__ = api_key
         self.__base_url__ = base_url
         self.__network_manager__ = network_manager
-        logger.debug("Initialization completed: \nBASE_URL= {} \nRESOURCE_URL=   {}\nAPI_KEY=  {}".format(base_url, resource_url, api_key))
+        self.__app_token__ = None
+        logger.debug(
+            "Initialization completed: \nBASE_URL= {} \nRESOURCE_URL=   {}\nAPI_KEY=  {}".format(base_url, resource_url,
+                                                                                                 api_key))
 
     def login(self) -> Optional[AppToken]:
         try:
@@ -27,7 +30,32 @@ class AppPyramidManager:
             s = self.__network_manager__.get_session()
             response = s.post(url, data=payload, headers=headers, timeout=AppPyramidManager.TIMEOUT)
             json_repr = json.loads(response.text)
-            return app_token_from_dict(json_repr)
+            self.__app_token__ = app_token_from_dict(json_repr)
+            return self.__app_token__
         except Exception as e:
             logger.error("\n[NETWORK]: Error while fetching police report list: {} \n".format(e))
         return None
+
+    def save(self, doc) -> bool:
+        """
+        Saves a document inside pyramid
+        :param doc:
+        :return: True or False
+        """
+        try:
+            url = "{}/{}".format(self.__base_url__, self.__res_url__)
+
+            headers = {
+                'AUTHORIZATION': self.__app_token__.token,
+                'Content-Type': "application/json"
+            }
+            s = self.__network_manager__.get_session()
+            payload = json.dumps(doc.to_dict())
+            response = s.post(url, data=payload, headers=headers, timeout=AppPyramidManager.TIMEOUT)
+            if response.status_code != 200:
+                return False
+        except Exception as e:
+            logger.error("Error while uploading injunctions to server")
+            logger.exception(e)
+            return False
+        return True