taranis-models 1.2.0.dev4__tar.gz → 1.2.0.dev5__tar.gz

{taranis_models-1.2.0.dev4/taranis_models.egg-info → taranis_models-1.2.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taranis-models
-Version: 1.2.0.dev4
+Version: 1.2.0.dev5
 Summary: Taranis AI Models
 Maintainer-email: AIT <benjamin.akhras@ait.ac.at>
 License-Expression: EUPL-1.2

{taranis_models-1.2.0.dev4 → taranis_models-1.2.0.dev5}/models/admin.py

@@ -100,6 +100,7 @@ class Role(TaranisBaseModel):
     _core_endpoint = "/config/roles"
     _model_name = "role"
     _search_fields = ["name", "description"]
+    _pretty_name = "Role"
 
     id: int | None = None
     name: str = ""

{taranis_models-1.2.0.dev4 → taranis_models-1.2.0.dev5}/models/types.py

@@ -1,4 +1,5 @@
-from enum import StrEnum, auto, IntEnum
+from enum import StrEnum, auto, IntEnum, nonmember
+import contextlib
 
 
 class TLPLevel(StrEnum):
@@ -8,6 +9,45 @@ class TLPLevel(StrEnum):
     AMBER = "amber"
     RED = "red"
 
+    _ACCESSIBLE_NAMES = nonmember(
+        {
+            "RED": ["RED", "AMBER_STRICT", "AMBER", "GREEN", "CLEAR"],
+            "AMBER_STRICT": ["AMBER_STRICT", "AMBER", "GREEN", "CLEAR"],
+            "AMBER": ["AMBER", "GREEN", "CLEAR"],
+            "GREEN": ["GREEN", "CLEAR"],
+            "CLEAR": ["CLEAR"],
+        }
+    )
+
+    def get_accessible_levels(self) -> list[str]:
+        """
+        Return the list of TLPLevel members this level can access.
+        """
+        names = type(self)._ACCESSIBLE_NAMES.get(self.name, [])
+        return [type(self)[nm].value for nm in names]
+
+    @classmethod
+    def get_most_restrictive_tlp(cls, tlp_levels: list["TLPLevel"]) -> "TLPLevel":
+        """
+        Get the most restrictive TLP level from a list of TLP levels.
+        If the list is empty, return the default TLP level (CLEAR).
+        """
+        if not tlp_levels:
+            return cls.CLEAR
+
+        provided = {tlp.name for tlp in tlp_levels}
+
+        return next(
+            (cls[level_name] for level_name in cls._ACCESSIBLE_NAMES.keys() if level_name in provided),
+            cls.CLEAR,
+        )
+
+    @classmethod
+    def get_tlp_level(cls, tlp_level: str) -> "TLPLevel | None":
+        with contextlib.suppress(ValueError):
+            return TLPLevel(tlp_level)
+        return None
+
 
 class ItemType(StrEnum):
     OSINT_SOURCE = auto()

{taranis_models-1.2.0.dev4 → taranis_models-1.2.0.dev5/taranis_models.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taranis-models
-Version: 1.2.0.dev4
+Version: 1.2.0.dev5
 Summary: Taranis AI Models
 Maintainer-email: AIT <benjamin.akhras@ait.ac.at>
 License-Expression: EUPL-1.2