#252 - Add custom field endpoint

thehive4py/client.py

@@ -14,6 +14,7 @@
     UserEndpoint,
 )
 from thehive4py.endpoints.cortex import CortexEndpoint
+from thehive4py.endpoints.custom_field import CustomFieldEndpoint
 from thehive4py.session import TheHiveSession
 
 
@@ -51,6 +52,9 @@ def __init__(
         self.organisation = OrganisationEndpoint(self.session)
         self.profile = ProfileEndpoint(self.session)
 
+        # entity endpoints
+        self.custom_field = CustomFieldEndpoint(self.session)
+
         # connector endpoints
         self.cortex = CortexEndpoint(self.session)
 

thehive4py/endpoints/__init__.py

@@ -2,6 +2,7 @@
 from .case import CaseEndpoint
 from .comment import CommentEndpoint
 from .cortex import CortexEndpoint
+from .custom_field import CustomFieldEndpoint
 from .observable import ObservableEndpoint
 from .organisation import OrganisationEndpoint
 from .procedure import ProcedureEndpoint

thehive4py/endpoints/custom_field.py

@@ -0,0 +1,28 @@
+from typing import List
+
+from thehive4py.endpoints._base import EndpointBase
+from thehive4py.types.custom_field import (
+    InputCustomField,
+    InputUpdateCustomField,
+    OutputCustomField,
+)
+
+
+class CustomFieldEndpoint(EndpointBase):
+    def create(self, custom_field: InputCustomField) -> OutputCustomField:
+        return self._session.make_request(
+            "POST", path="/api/v1/customField", json=custom_field
+        )
+
+    def list(self) -> List[OutputCustomField]:
+        return self._session.make_request("GET", path="/api/v1/customField")
+
+    def delete(self, custom_field_id: str) -> None:
+        return self._session.make_request(
+            "DELETE", path=f"/api/v1/customField/{custom_field_id}"
+        )
+
+    def update(self, custom_field_id: str, fields: InputUpdateCustomField) -> None:
+        return self._session.make_request(
+            "PATCH", path=f"/api/v1/customField/{custom_field_id}", json=fields
+        )