[client] Add cve endpoints

Author: MarineLeM

pyobas/apis/__init__.py

@@ -1,5 +1,6 @@
 from .attack_pattern import *  # noqa: F401,F403
 from .collector import *  # noqa: F401,F403
+from .cve import *  # noqa: F401,F403
 from .document import *  # noqa: F401,F403
 from .endpoint import *  # noqa: F401,F403
 from .inject import *  # noqa: F401,F403

pyobas/apis/collector.py

@@ -1,3 +1,6 @@
+from typing import Any, Dict
+
+from pyobas import exceptions as exc
 from pyobas.base import RESTManager, RESTObject
 from pyobas.mixins import CreateMixin, GetMixin, ListMixin, UpdateMixin
 from pyobas.utils import RequiredOptional
@@ -18,3 +21,9 @@ class CollectorManager(GetMixin, ListMixin, CreateMixin, UpdateMixin, RESTManage
             "collector_period",
         )
     )
+
+    @exc.on_http_error(exc.OpenBASUpdateError)
+    def get(self, collector_id: str, **kwargs: Any) -> Dict[str, Any]:
+        path = f"{self.path}/" + collector_id
+        result = self.openbas.http_get(path, **kwargs)
+        return result

pyobas/apis/cve.py

@@ -0,0 +1,18 @@
+from typing import Any, Dict
+
+from pyobas import exceptions as exc
+from pyobas.base import RESTManager, RESTObject
+
+
+class Cve(RESTObject):
+    _id_attr = "cve_id"
+
+
+class CveManager(RESTManager):
+    _path = "/cves"
+
+    @exc.on_http_error(exc.OpenBASUpdateError)
+    def upsert(self, data: Dict[str, Any], **kwargs: Any) -> Dict[str, Any]:
+        path = f"{self.path}/bulk"
+        result = self.openbas.http_post(path, post_data=data, **kwargs)
+        return result

pyobas/client.py

@@ -59,6 +59,7 @@ def __init__(
         self.organization = apis.OrganizationManager(self)
         self.injector = apis.InjectorManager(self)
         self.collector = apis.CollectorManager(self)
+        self.cve = apis.CveManager(self)
         self.inject = apis.InjectManager(self)
         self.document = apis.DocumentManager(self)
         self.kill_chain_phase = apis.KillChainPhaseManager(self)