algosec-appviz 0.0.4__tar.gz → 0.0.6__tar.gz

{algosec_appviz-0.0.4 → algosec_appviz-0.0.6}/PKG-INFO

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

{algosec_appviz-0.0.4 → algosec_appviz-0.0.6}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 name = "algosec_appviz"
 repository = "https://github.com/bogdan-iot/algosec-appviz"
 documentation = "https://github.com/bogdan-iot/algosec-appviz/blob/master/README.md"
-version = "0.0.4"
+version = "0.0.6"
 license = "LICENSE"
 readme = "README.md"
 description = "AlgoSec AppViz Library"

{algosec_appviz-0.0.4 → algosec_appviz-0.0.6}/src/algosec_appviz/main.py

@@ -14,12 +14,13 @@ regions = {
 
 
 class AppViz:
-    def __init__(self, region='eu', tenant_id=None, client_id=None, client_secret=None):
+    def __init__(self, region='eu', tenant_id=None, client_id=None, client_secret=None, proxies=None):
         if region not in regions.keys():
             raise ValueError(f"Invalid region, must be one of: {', '.join(regions.keys())}")
 
-        login_url = f"https://{regions[region]}/api/algosaas/auth/v1/access-keys/login"
+        self.proxies = proxies
 
+        login_url = f"https://{regions[region]}/api/algosaas/auth/v1/access-keys/login"
         data = {
             "tenantId": tenant_id or environment.get_tenant_id(),
             "clientId": client_id or environment.get_client_id(),
@@ -34,10 +35,26 @@ class AppViz:
         response = requests.post(login_url, json=data, headers=headers)
         if response.status_code != 200:
             raise ConnectionError(f"Authentication to AppViz failed: {response.text}")
+
         self.url = 'https://' + regions[region]
         self._token_type = response.json()['token_type']
         self._token = response.json()['access_token']
 
+    def create_application(self, name=None, **kwargs):
+        if not name:
+            raise ValueError("Name is required")
+
+        body = {
+            'name': name,
+            **kwargs
+        }
+
+        result = self._make_api_call('POST',
+                                     '/BusinessFlow/rest/v1/applications/new',
+                                     body=body)
+
+        return result
+
     def create_network_object(self, name=None, obj_type=None, content=None, **kwargs):
         valid_object_types = ['Range', 'Host', 'Group', 'Abstract']
 
@@ -59,7 +76,7 @@ class AppViz:
                                      '/BusinessFlow/rest/v1/network_objects/new',
                                      body=body)
 
-        print(result)
+        return result
 
     def get_applications(self):
         response = self._make_api_call('GET',
@@ -87,11 +104,14 @@ class AppViz:
             'Authorization': f'{self._token_type} {self._token}'
         }
 
+        url = self.url + url_path
+
         if method.lower() == 'get':
-            response = requests.get(self.url + url_path, headers=headers, json=body, params=params)
+            response = requests.get(url, headers=headers, json=body, params=params, proxies=self.proxies)
         elif method.lower() == 'post':
-            response = requests.post(self.url + url_path, headers=headers, json=body, params=params)
+            response = requests.post(url, headers=headers, json=body, params=params, proxies=self.proxies)
         else:
-            raise AssertionError("Invalid method, must be: 'GET' or 'POST'")
+            raise ValueError("Invalid method, must be: 'GET' or 'POST'")
 
+        response.raise_for_status()
         return response.json()