algosec-appviz 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

algosec_appviz/main.py

@@ -32,7 +32,7 @@ class AppViz:
             "Accept": "application/json"
         }
 
-        response = requests.post(login_url, json=data, headers=headers)
+        response = requests.post(login_url, json=data, headers=headers, proxies=self.proxies)
         if response.status_code != 200:
             raise ConnectionError(f"Authentication to AppViz failed: {response.text}")
 
@@ -78,13 +78,55 @@ 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: True if successful, false if not
+        """
+        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 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 +141,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 +153,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.6.dist-info → algosec_appviz-0.0.8.dist-info}/METADATA

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

algosec_appviz-0.0.8.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=NAkuuMtSFgZcnhMnBV-22ebT3_ObocFGsdj-KAas3AI,5783
+algosec_appviz-0.0.8.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+algosec_appviz-0.0.8.dist-info/METADATA,sha256=isNuuOwxax_jp7kBJmVDN0w52uExys-4OxP_eAJzf40,1798
+algosec_appviz-0.0.8.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+algosec_appviz-0.0.8.dist-info/RECORD,,

algosec_appviz-0.0.6.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=2WCC6PaCaeUF6HTRjE_EoFCj7k1R0AngXsbCOmGww4c,4153
-algosec_appviz-0.0.6.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-algosec_appviz-0.0.6.dist-info/METADATA,sha256=yq6_qrQYtflqYsvWv5nG-iLswU5VGf6Ym3xnZNRMKDw,1798
-algosec_appviz-0.0.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-algosec_appviz-0.0.6.dist-info/RECORD,,