algosec-appviz 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

algosec_appviz/main.py

@@ -78,13 +78,78 @@ class AppViz:
 
         return result
 
+    def delete_network_object(self, obj_id=None):
+        """
+        Deletes a network object in AppViz
+        :param obj_id: The object ID
+        :return: Change details if successful, empty string otherwise
+        """
+        if not obj_id:
+            raise ValueError("Object ID is mandatory")
+
+        result = self._make_api_call('DELETE',
+                                     f'/BusinessFlow/rest/v1/network_objects/{obj_id}')
+
+        if result['httpStatusCode'] != 200:
+            print(f"Error deleting object: {result['message']}")
+            return ""
+
+        return result
+
+    def update_network_object(self, obj_id=None, **kwargs):
+        """
+        Updates a network object in AppViz
+        :return: The new object details, if successful, empty string otherwise
+        """
+        if not obj_id:
+            raise ValueError("Object ID is mandatory")
+
+        result = self._make_api_call('POST',
+                                     f'/BusinessFlow/rest/v1/network_objects/{obj_id}',
+                                     body={**kwargs})
+
+        if isinstance(result, dict) and 'networkObject' in result.keys():
+            return result['networkObject']
+
+        try:
+            print(result[1])
+        except KeyError:
+            if 'success' in result.keys() and not result['success']:
+                print(result['message'])
+
+        return ""
+
     def get_applications(self):
         response = self._make_api_call('GET',
                                        '/BusinessFlow/rest/v1/applications')
 
         return [MyDict(x) for x in response]
 
+    def get_all_network_objects(self):
+        """
+        Gets all the network objects from AppViz. This could take some time, depending on the number of objects.
+        :return: The list of objects
+        """
+        appviz_objects = []
+        page = 1
+
+        while True:
+            print(f"Getting AppViz Network Objects, page {page}...")
+            objects = self.list_network_objects(page_number=page)
+            page = page + 1
+            appviz_objects.extend(objects)
+            if len(objects) < 1000:
+                break
+
+        return appviz_objects
+
     def list_network_objects(self, page_number=1, page_size=1000):
+        """
+        Get a list of objects based on the page_size (the number of objects to be retrieved) and the page number
+        :param page_number: Page number, defaults to 1
+        :param page_size: Page size, defaults to 1000
+        :return: The list of objects
+        """
         response = self._make_api_call('GET',
                                        '/BusinessFlow/rest/v1/network_objects/',
                                        params={'page_number': page_number, 'page_size': page_size})
@@ -99,6 +164,7 @@ class AppViz:
         return [MyDict(x) for x in response]
 
     def _make_api_call(self, method, url_path, body=None, params=None):
+        valid_methods = ['get', 'post', 'delete']
         headers = {
             'Accept': 'application/json',
             'Authorization': f'{self._token_type} {self._token}'
@@ -110,8 +176,9 @@ class AppViz:
             response = requests.get(url, headers=headers, json=body, params=params, proxies=self.proxies)
         elif method.lower() == 'post':
             response = requests.post(url, headers=headers, json=body, params=params, proxies=self.proxies)
+        elif method.lower() == 'delete':
+            response = requests.delete(url, headers=headers, json=body, params=params, proxies=self.proxies)
         else:
-            raise ValueError("Invalid method, must be: 'GET' or 'POST'")
+            raise ValueError(f"Invalid method, must be: {', '.join(valid_methods)}")
 
-        response.raise_for_status()
         return response.json()

{algosec_appviz-0.0.7.dist-info → algosec_appviz-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: algosec_appviz
-Version: 0.0.7
+Version: 0.0.9
 Summary: AlgoSec AppViz Library
 Home-page: https://github.com/bogdan-iot/algosec-appviz
 License: LICENSE

algosec_appviz-0.0.9.dist-info/RECORD

@@ -0,0 +1,7 @@
+algosec_appviz/__init__.py,sha256=epwLWpezNwfYP0Zuqk32rWEvNeloAKzbrRFi9tzlsXo,46
+algosec_appviz/environment.py,sha256=ycrVSpR6oLpj9p54SvDdy0nqwS9IwHAh-1x-zOaTQ5I,1775
+algosec_appviz/main.py,sha256=4t6EQxiSnqzvClgWr16FzpjgEV8rcbtaxH_cco5hny8,6591
+algosec_appviz-0.0.9.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+algosec_appviz-0.0.9.dist-info/METADATA,sha256=h1bBuqUvU8_Tnwdehn_dVy4xN521-YMAdgptir-C0bw,1798
+algosec_appviz-0.0.9.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+algosec_appviz-0.0.9.dist-info/RECORD,,

algosec_appviz-0.0.7.dist-info/RECORD

@@ -1,7 +0,0 @@
-algosec_appviz/__init__.py,sha256=epwLWpezNwfYP0Zuqk32rWEvNeloAKzbrRFi9tzlsXo,46
-algosec_appviz/environment.py,sha256=ycrVSpR6oLpj9p54SvDdy0nqwS9IwHAh-1x-zOaTQ5I,1775
-algosec_appviz/main.py,sha256=2yYudC6XbIMeBIfC8Sg4TYqEbIx_vH-qXiMNc_cGfHY,4175
-algosec_appviz-0.0.7.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-algosec_appviz-0.0.7.dist-info/METADATA,sha256=OVzoNc-MhpHNT8-PAz79OVcOXw4J1uMDA_WeTtVDgYM,1798
-algosec_appviz-0.0.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-algosec_appviz-0.0.7.dist-info/RECORD,,