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

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()

{algosec_appviz-0.0.4.dist-info → algosec_appviz-0.0.6.dist-info}/METADATA

@@ -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.6.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=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,,

algosec_appviz-0.0.4.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=F5sUXiVkvT_RKraXpGC_oQNvTgI4jl4P1SF_UwUoJBY,3631
-algosec_appviz-0.0.4.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-algosec_appviz-0.0.4.dist-info/METADATA,sha256=mDnVU2_jSu6mhW9a5fylhvF_IPKKr6KfT_2WBuxKhbg,1798
-algosec_appviz-0.0.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-algosec_appviz-0.0.4.dist-info/RECORD,,