[client] Add payloads helpers

Author: SamuelHassine

pyobas/apis/__init__.py

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

pyobas/apis/attack_pattern.py

@@ -4,13 +4,13 @@
 from pyobas.base import RESTManager, RESTObject
 
 
-class AttackPatternPhase(RESTObject):
+class AttackPattern(RESTObject):
     _id_attr = "attack_pattern_id"
 
 
 class AttackPatternManager(RESTManager):
     _path = "/attack_patterns"
-    _obj_cls = AttackPatternPhase
+    _obj_cls = AttackPattern
 
     @exc.on_http_error(exc.OpenBASUpdateError)
     def upsert(

pyobas/apis/payload.py

@@ -0,0 +1,19 @@
+from typing import Any, Dict, List
+
+from pyobas import exceptions as exc
+from pyobas.base import RESTManager, RESTObject
+
+
+class Payload(RESTObject):
+    _id_attr = "payload_id"
+
+
+class PayloadManager(RESTManager):
+    _path = "/payloads"
+    _obj_cls = Payload
+
+    @exc.on_http_error(exc.OpenBASUpdateError)
+    def upsert(self, payload: Dict[str, Any], **kwargs: Any) -> Dict[str, Any]:
+        path = f"{self.path}/upsert"
+        result = self.openbas.http_post(path, post_data=payload, **kwargs)
+        return result

pyobas/client.py

@@ -67,6 +67,7 @@ def __init__(
         self.endpoint = apis.EndpointManager(self)
         self.user = apis.UserManager(self)
         self.inject_expectation = apis.InjectExpectationManager(self)
+        self.payload = apis.PayloadManager(self)
 
     @staticmethod
     def _check_redirects(result: requests.Response) -> None: