[client] Implement expectations helpers

Author: SamuelHassine

pyobas/apis/__init__.py

@@ -2,6 +2,7 @@
 from .collector import *  # noqa: F401,F403
 from .document import *  # noqa: F401,F403
 from .inject import *  # noqa: F401,F403
+from .inject_expectation import *  # noqa: F401,F403
 from .injector import *  # noqa: F401,F403
 from .kill_chain_phase import *  # noqa: F401,F403
 from .me import *  # noqa: F401,F403

pyobas/apis/inject_expectation.py

@@ -0,0 +1,20 @@
+from typing import Any, Dict
+
+from pyobas import exceptions as exc
+from pyobas.base import RESTManager, RESTObject
+from pyobas.mixins import ListMixin, UpdateMixin
+
+
+class InjectExpectation(RESTObject):
+    pass
+
+
+class InjectExpectationManager(ListMixin, UpdateMixin, RESTManager):
+    _path = "/injects/expectations"
+    _obj_cls = InjectExpectation
+
+    @exc.on_http_error(exc.OpenBASUpdateError)
+    def expectations_for_source(self, source_id: str, **kwargs: Any) -> Dict[str, Any]:
+        path = f"{self.path}/" + source_id
+        result = self.openbas.http_get(path, **kwargs)
+        return result

pyobas/client.py

@@ -65,6 +65,7 @@ def __init__(
         self.attack_pattern = apis.AttackPatternManager(self)
         self.team = apis.TeamManager(self)
         self.user = apis.UserManager(self)
+        self.inject_expectation = apis.InjectExpectationManager(self)
 
     @staticmethod
     def _check_redirects(result: requests.Response) -> None:

pyobas/utils.py

@@ -4,7 +4,7 @@
 import json
 import logging
 import urllib.parse
-from typing import Any, Callable, Dict, Iterator, List, Literal, Optional, Tuple, Union
+from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple, Union
 
 import requests
 from pythonjsonlogger import jsonlogger