Add WithName mixin

mreg_cli/api/abstracts.py

@@ -11,6 +11,7 @@
 
 from mreg_cli.api.endpoints import Endpoint
 from mreg_cli.api.history import HistoryItem, HistoryResource
+from mreg_cli.exceptions import CliError
 from mreg_cli.log import cli_error, cli_warning
 from mreg_cli.outputmanager import OutputManager
 from mreg_cli.utilities.api import (
@@ -193,6 +194,58 @@ def get_by_field(cls, field: str, value: str) -> BMT | None:
 
         return cast(BMT, cls(**data))
 
+    @classmethod
+    def get_by_field_or_raise(
+        cls,
+        field: str,
+        value: str,
+        exc_type: type[Exception] = CliError,
+        exc_message: str | None = None,
+    ) -> BMT:
+        """Get an object by a field and raise if not found.
+
+        Used for cases where the object must exist for the operation to continue.
+
+        :param field: The field to search by.
+        :param value: The value to search for.
+        :param exc_type: The exception type to raise.
+        :param exc_message: The exception message. Overrides the default message.
+
+        :returns: The object if found.
+        """
+        obj = cls.get_by_field(field, value)
+        if not obj:
+            if not exc_message:
+                exc_message = f"{cls} with {field} {value!r} not found."
+            raise exc_type(exc_message)
+        return obj
+
+    @classmethod
+    def get_by_field_and_raise(
+        cls,
+        field: str,
+        value: str,
+        exc_type: type[Exception] = CliError,
+        exc_message: str | None = None,
+    ) -> None:
+        """Get an object by a field and raise if found.
+
+        Used for cases where the object must NOT exist for the operation to continue.
+
+        :param field: The field to search by.
+        :param value: The value to search for.
+        :param exc_type: The exception type to raise.
+        :param exc_message: The exception message. Overrides the default message.
+
+        :raises Exception: If the object is found.
+        """
+        obj = cls.get_by_field(field, value)
+        if obj:
+            if not exc_message:
+                exc_message = f"{cls} with {field} {value!r} already exists."
+            raise exc_type(exc_message)
+        return None
+
     @classmethod
     def get_list_by_field(
         cls, field: str, value: str | int, ordering: str | None = None

mreg_cli/api/models.py

@@ -5,7 +5,7 @@
 import ipaddress
 import re
 from datetime import date, datetime
-from typing import Any, Literal, cast
+from typing import Any, Literal, Self, cast
 
 from pydantic import (
     AliasChoices,
@@ -256,6 +256,42 @@ def valid_numeric_ttl(self, ttl: int) -> int:
         return ttl
 
 
+class WithName(APIMixin[Any]):
+    """Model for an object that has a name element."""
+
+    @classmethod
+    def ensure_name_not_exists(cls, name: str) -> None:
+        """Ensure a name is not already used.
+
+        :param name: The name to check for uniqueness.
+        """
+        # TODO: pass in exception type and message?
+        cls.get_by_field_and_raise("name", name)
+
+    @classmethod
+    def ensure_name_exists(cls, name: str) -> None:
+        """Ensure a name exists.
+
+        :param name: The name to check for existence.
+        """
+        instance = cls.get_by_field_or_raise("name", name)
+        if not instance:
+            raise ValueError(f"{cls} with the name '{name}' does not exist.")
+
+    @classmethod
+    def get_by_name(cls, name: str) -> Self:
+        """Get a resource by name.
+
+        :param name: The role name to search for.
+        :returns: The resource if found.
+        :raises CliWarning: If the role is not found.
+        """
+        data = get_item_by_key_value(cls.endpoint(), "name", name)
+        if not data:
+            cli_warning(f"{cls} with name {name} not found.")
+        return cls(**data)
+
+
 class NameServer(FrozenModelWithTimestamps, WithTTL):
     """Model for representing a nameserver within a DNS zone."""
 
@@ -385,7 +421,7 @@ def created_at(self) -> datetime:
         return self.created_at_tz_naive.replace(tzinfo=self.updated_at.tzinfo)
 
     @classmethod
-    def get_by_name(cls, name: str) -> Atom | Role:
+    def get_role_or_atom(cls, name: str) -> Atom | Role:
         """Get an Atom or Role by name.
 
         :param name: The name to search for.
@@ -422,7 +458,7 @@ def output(self, padding: int = 14) -> None:
         output_manager.add_line(f"{'Description:':<{padding}}{self.description}")
 
 
-class Role(HostPolicy, APIMixin["Role"]):
+class Role(HostPolicy, WithName, APIMixin["Role"]):
     """Model for a role."""
 
     id: int  # noqa: A003
@@ -435,19 +471,6 @@ def endpoint(cls) -> Endpoint:
         """Return the endpoint for the class."""
         return Endpoint.HostPolicyRoles
 
-    @classmethod
-    def get_by_name(cls, name: str) -> Role:
-        """Get a Role by name.
-
-        :param name: The role name to search for.
-        :returns: The role if found.
-        :raises CliWarning: If the role is not found.
-        """
-        data = get_item_by_key_value(Endpoint.HostPolicyRoles, "name", name)
-        if not data:
-            cli_warning(f"Role with name {name} not found.")
-        return cls(**data)
-
     def output(self, padding: int = 14) -> None:
         """Output the role to the console.
 
@@ -485,7 +508,7 @@ def get_labels(self) -> list[Label]:
         return [Label.get_by_id_or_raise(id_) for id_ in self.labels]
 
 
-class Atom(HostPolicy, APIMixin["Atom"]):
+class Atom(HostPolicy, WithName, APIMixin["Atom"]):
     """Model for an atom."""
 
     id: int  # noqa: A003
@@ -496,19 +519,6 @@ def endpoint(cls) -> Endpoint:
         """Return the endpoint for the class."""
         return Endpoint.HostPolicyAtoms
 
-    @classmethod
-    def get_by_name(cls, name: str) -> Atom:
-        """Get an Atom by name.
-
-        :param name: The atom name to search for.
-        :returns: The atom if found.
-        :raises CliWarning: If the atom is not found.
-        """
-        data = get_item_by_key_value(Endpoint.HostPolicyAtoms, "name", name)
-        if not data:
-            cli_warning(f"Atom with name {name} not found.")
-        return cls(**data)
-
     def output(self, padding: int = 14) -> None:
         """Output the role to the console.
 
@@ -654,7 +664,7 @@ def __hash__(self):
         return hash((self.id, self.network))
 
 
-class IPAddress(FrozenModelWithTimestamps, WithHost, APIMixin["IPAddress"]):
+class IPAddress(FrozenModelWithTimestamps, WithHost, WithName, APIMixin["IPAddress"]):
     """Represents an IP address with associated details."""
 
     id: int  # noqa: A003

mreg_cli/commands/policy.py

@@ -247,7 +247,7 @@ def info(args: argparse.Namespace) -> None:
     """
     names: list[str] = args.name
     for name in names:
-        role_or_atom = HostPolicy.get_by_name(name)
+        role_or_atom = HostPolicy.get_role_or_atom(name)
         role_or_atom.output()