[client] Implement user / team helpers

Author: SamuelHassine

pyobas/_apis/__init__.py

@@ -6,5 +6,7 @@
 from .kill_chain_phase import *  # noqa: F401,F403
 from .me import *  # noqa: F401,F403
 from .organization import *  # noqa: F401,F403
+from .team import *  # noqa: F401,F403
+from .user import *  # noqa: F401,F403
 
 __all__ = [name for name in dir() if not name.startswith("_")]

pyobas/_version.py

@@ -3,4 +3,4 @@
 __email__ = "[email protected]"
 __license__ = "Apache 2.0"
 __title__ = "python-openbas"
-__version__ = "1.0.0"
+__version__ = "4.0.0"

pyobas/client.py

@@ -60,6 +60,8 @@ def __init__(
         self.document = _apis.DocumentManager(self)
         self.kill_chain_phase = _apis.KillChainPhaseManager(self)
         self.attack_pattern = _apis.AttackPatternManager(self)
+        self.team = _apis.TeamManager(self)
+        self.user = _apis.UserManager(self)
 
     @staticmethod
     def _check_redirects(result: requests.Response) -> None:
@@ -172,6 +174,7 @@ def http_request(
         opts["headers"]["Content-type"] = send_data.content_type
 
         # cur_retries = 0
+        print(send_data)
         while True:
             # noinspection PyTypeChecker
             result = self._backend.http_request(

pyobas/helpers.py

@@ -258,9 +258,9 @@ def ping(self) -> None:
         while not self.exit_event.is_set():
             try:
                 if self.ping_type == "injector":
-                    self.api.injector.create(self.config)
+                    self.api.injector.create(self.config, False)
                 else:
-                    self.api.collector.create(self.config)
+                    self.api.collector.create(self.config, False)
             except Exception as e:  # pylint: disable=broad-except
                 print(str(e))
             self.exit_event.wait(40)

pyobas/mixins.py

@@ -202,7 +202,12 @@ def create(
         self._create_attrs.validate_attrs(data=data)
         # Handle specific URL for creation
         path = kwargs.pop("path", self.path)
-        files = {} if icon is None else {"icon": icon}
+        if icon:
+            files = {"icon": icon}
+        elif icon is False:
+            files = {}
+        else:
+            files = None
         server_data = self.openbas.http_post(
             path, post_data=data, files=files, **kwargs
         )