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

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

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

{algosec_appviz-0.0.4 → algosec_appviz-0.0.5}/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.5"
 license = "LICENSE"
 readme = "README.md"
 description = "AlgoSec AppViz Library"

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

@@ -38,6 +38,21 @@ class AppViz:
         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 +74,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',
@@ -92,6 +107,6 @@ class AppViz:
         elif method.lower() == 'post':
             response = requests.post(self.url + url_path, headers=headers, json=body, params=params)
         else:
-            raise AssertionError("Invalid method, must be: 'GET' or 'POST'")
+            raise ValueError("Invalid method, must be: 'GET' or 'POST'")
 
         return response.json()