Source code for masterpiece.masterpiece

"""
The elementary base class of everything.

Author: Juha Meskanen
Date: 2024-10-26
"""

from __future__ import annotations

import inspect
import logging
from typing import Any, Callable, Optional, Type, Union, cast
from .url import URL


class classproperty:
    """
    A decorator that allows you to define class-level properties.
    Replaces the deprecated combination of @classmethod and @property.
    """

    def __init__(self, func: Callable[[Type[Any]], Any]) -> None:
        """
        Initialize the classproperty with the function to decorate.

        Args:
            func: A callable that accepts a class type and returns any value.
        """
        self.func = func

    def __get__(self, instance: Any, owner: Type[Any]) -> Any:
        """
        Get the class property value by calling the decorated function with the class.

        Args:
            instance: The instance of the class (not used here).
            owner: The class that owns this property.

        Returns:
            The result of calling the decorated function with the class.
        """
        return self.func(owner)


class MasterPiece:
    """An object with a name. Base class of everything. Serves as the
    foundational base class for any real-world object that can be a part of
    a hierarchy.

    """

    # non-serializable private class attributes
    _log: Optional[logging.Logger] = None
    _factory: dict[str, Type[MasterPiece]] = {}

    def __init_subclass__(cls, **kwargs: dict[str, Any]) -> None:
        """Called when a new sub-class is created.

        Automatically registers the sub class by calling its register()
        method. For more information on this method consult Python
        documentation.
        """
        super().__init_subclass__(**kwargs)
        cls.register()

    @classmethod
    def classattrs_to_dict(cls) -> dict[str, Any]:
        """Convert the class's own attributes to a dictionary, excluding inherited and private ones."""
        return {
            attr: value
            for attr, value in cls.__dict__.items()
            if not callable(value)  # Exclude regular methods
            and not isinstance(
                value, (classmethod, staticmethod, classproperty)
            )  # Exclude class/staticmethods
            and not attr.startswith("__")  # Exclude special attributes like __module__
            and not attr.startswith("_")  # Exclude private/protected attributes
        }

    @classmethod
    def classattrs_from_dict(cls, attributes: dict[str, Any]) -> None:
        """Set only the class's own attributes from a dictionary."""
        # Get the set of attributes defined in this class itself
        own_class_attrs = set(cls.__dict__.keys())

        for key, value in attributes.items():
            # Only set attributes that belong directly to this class (not inherited)
            if key in own_class_attrs:
                setattr(cls, key, value)

    @classmethod
    def has_class_method_directly(cls, method_name: str) -> bool:
        """
        Check if the method is defined directly in the class (not inherited).
        """
        method = getattr(cls, method_name, None)
        # Ensure that the method is callable and check its defining class
        if callable(method):
            # Check if the method's class is the same as the one we are querying
            return cast(bool, method.__qualname__.split(".")[0] == cls.__name__)
        return False

    @classmethod
    def factory(cls) -> dict[str, Type[MasterPiece]]:
        """Fetch the dictionary holding class names and associated classes.

        Returns:
            factory: with class names and associated classes
        """
        return cls._factory

    @classmethod
    def register(cls) -> None:
        """Register the class.

        Called immediately upon class initialization, right before the class attributes
        are loaded from the class specific configuration files.

        Subclasses can extend this with custom register functionality:

        .. code-block:: python

            class MyMasterPiece(MasterPiece):

                @classmethod
                def register(cls):
                    super().register()  # Don't forget
                    cls._custom_field = True
        """
        cls.init_class(cls)

    @classmethod
    def init_class(cls, clazz: Type[MasterPiece]) -> None:
        """Initialize class.  Registers the class into the
        class factory .

        Args:
            clazz (class): class to be initialized
        """
        if clazz.__name__ not in cls._factory:
            if inspect.isabstract(clazz):
                cls.log_info(f"Abstract {clazz.__name__} skipped")
            else:
                cls._factory[clazz.__name__] = clazz
                cls.log_info(f"Class {clazz.__name__} initialized")

    @classmethod
    def set_log(cls, l: logging.Logger) -> None:
        """Set logger.

        Args:
            l (logger): logger object
        """

        cls._log = l

    @classmethod
    def get_class_id(cls) -> str:
        """Return the class id of the class. Each class has an unique
        name that can be used for instantiating the class via
        :meth:`Object.instantiate` method.

        Args:
            cls (class): class

        Returns:
            id (str): unique class identifier through which the class can be
            instantiated by factory method pattern.
        """
        return cls.__name__

    def __init__(
        self,
        name: str = "noname",
        payload: Optional[MasterPiece] = None,
        parent: Optional[MasterPiece] = None,
    ):
        """
        Initialize a MasterPiece object.

        Args:
            name (str): The name of the object.
            payload (MasterPiece, optional): Optional payload data associated with the object. Defaults to None.
            parent (MasterPiece, optional): The parent object in the hierarchy. Defaults to None.
        """
        self.name = name
        self.payload = payload
        self.parent = parent
        self._elapsed: float = 0.0
        self._num_updates: int = 0

    @classmethod
    def log_debug(cls, msg: str, details: str = "") -> None:
        """Logs the given debug message to the application log.

        Args:
            msg (str): The  message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        full_message = f"{cls.__name__} : {msg}"
        if details:
            full_message += f" - {details}"
        if cls._log is not None:
            cls._log.debug(full_message)
        else:
            print(full_message)

    @classmethod
    def log_warning(cls, msg: str, details: str = "") -> None:
        """Logs the given debug message to the application log.

        Args:
            msg (str): The  message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        full_message = f"{cls.__name__} : {msg}"
        if details:
            full_message += f" - {details}"
        if cls._log is not None:
            cls._log.warning(full_message)
        else:
            print(full_message)

    @classmethod
    def log_info(cls, msg: str, details: str = "") -> None:
        """Logs the given  message to the application log.

        Args:
            msg (str): The  message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        full_message = f"{cls.__name__} : {msg}"
        if details:
            full_message += f" - {details}"
        if cls._log is not None:
            cls._log.info(full_message)
        else:
            print(full_message)

    @classmethod
    def log_error(cls, msg: str, details: str = "") -> None:
        """Logs the given  message to the application log.

        Args:
            msg (str): The  message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        full_message = f"{cls.__name__} : {msg}"
        if details:
            full_message += f" - {details}"
        if cls._log is not None:
            cls._log.error(full_message)
        else:
            print(full_message)

    def debug(self, msg: str, details: str = "") -> None:
        """Logs the given debug message to the application log.

        Args:
            msg (str): The information message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        self.log_debug(f"{self.name} : {msg}", details)

    def info(self, msg: str, details: str = "") -> None:
        """Logs the given information message to the application log.

        Args:
            msg (str): The information message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        self.log_info(f"{self.name} : {msg}", details)

    def warning(self, msg: str, details: str = "") -> None:
        """Logs the given warning message to the application log.

        Args:
            msg (str): The message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        self.log_warning(f"{self.name} : {msg}", details)

    def error(self, msg: str, details: str = "") -> None:
        """Logs the given error message to the application log.

        Args:
            msg (str): The message to be logged.
            details (str): Additional detailed information for the message to be logged
        """
        self.log_error(f"{self.name} : {msg}", details)

    def to_dict(self) -> dict[str, Any]:
        """Convert instance attributes to a dictionary."""

        return {
            "_class": self.get_class_id(),  # the real class
            "_version:": 0,
            "_object": {
                "name": self.name,
                "payload": (
                    self.payload.to_dict() if self.payload is not None else None
                ),
            },
        }

    def from_dict(self, data: dict[str, Any]) -> None:
        """Update instance attributes from a dictionary."""

        if self.get_class_id() != data["_class"]:
            raise ValueError(
                f"Class mismatch, expected:{self.get_class_id()}, actual:{data['_class']}"
            )
        for key, value in data["_object"].items():
            if key == "payload":
                if value is not None:
                    self.payload = MasterPiece.instantiate(value["_class"])
                    self.payload.from_dict(value)
                else:
                    self.payload = None
            else:
                setattr(self, key, value)

    def copy(self) -> MasterPiece:
        """Create and return a copy of the current object.

        This method serializes the current object to a dictionary using the `to_dict` method,
        creates a new instance of the object's class, and populates it with the serialized data
        using the `from_dict` method.

        This method uses class identifier based instantiation (see factory method pattern) to
        create a new instance of the object, and 'to_dict' and 'from_dict'  methods to initialize
        object's state.

        Returns:
            A new instance of the object's class with the same state as the original object.

        Example:
        ::

            clone_of_john = john.copy()
        """

        data = self.to_dict()
        copy_of_self = MasterPiece.instantiate(self.get_class_id())
        copy_of_self.from_dict(data)
        return copy_of_self

    def do(
        self,
        action: Callable[["MasterPiece", dict[str, Any]], bool],
        context: dict[str, Any],
    ) -> bool:
        """
        Execute the given action to the object, by calling the provided `action`.

        Args:
            action(Callable[["MasterPiece", dict[str, Any]], bool]): A callable that takes
            (node, context) and returns a boolean.
            context (dict[str, Any]): Any context data that the action may use.

        Returns:
            The return value from the executed action.
        """
        return action(self, context)

    def run(self) -> None:
        """Run the masterpiece.  Dispatches the call to `payload` object and
        returns  the control to the caller.
        """
        if self.payload is not None and isinstance(self.payload, MasterPiece):
            self.payload.run()

    def run_forever(self) -> None:
        """Run the payload forever. This method will return only when violently
        terminated. If the object does not have playload object, or it is not
        instance of 'MasterPiece' class then returns immediately and this method has
        no effect.
        """
        if self.payload is not None and isinstance(self.payload, MasterPiece):
            try:
                self.payload.run_forever()
                print("Newtorking loop exit without exception")
            except KeyboardInterrupt:
                print("Application interrupted by user.")
            except (ValueError, IOError) as e:
                print(f"Specific error occurred: {e}")
            except Exception as e:
                print(f"An error occurred: {e}")

    def shutdown(self) -> None:
        """Shutdown the payload object. If the payload object is None, or is not instance of MasterPiece,
        then the call has no effect.
        """
        if self.payload is not None and isinstance(self.payload, MasterPiece):
            self.payload.shutdown()

    @classmethod
    def instantiate(cls, class_id: str, *args: Any) -> MasterPiece:
        """Create an instance of the class corresponding to the given class identifier.

        Args:
            class_id (str): Identifier of the class to instantiate.
            *args: Optional arguments to pass to the class constructor.

        Returns:
            MasterPiece: An instance of the class corresponding to the given class identifier.
        """
        if class_id in cls._factory:
            return cls._factory[class_id](*args)
        raise ValueError(f"Attempting to instantiate unregistered class {class_id}")

    @classmethod
    def find_class(cls, class_id: str) -> Union[Type[MasterPiece], None]:
        """Create an instance of the class corresponding to the given class identifier.

        Args:
            class_id (str): Identifier of the class to instantiate.
            *args: Optional arguments to pass to the class constructor.

        Returns:
            MasterPiece: An instance of the class corresponding to the given class identifier.
        """
        if class_id in cls._factory:
            return cls._factory[class_id]
        return None

    @property
    def parent(self) -> Optional[MasterPiece]:
        return self._parent

    @parent.setter
    def parent(self, value: Optional[MasterPiece]) -> None:
        self._parent = value

    def make_url(self) -> URL:
        """Generate the URL for the composite, including all children."""
        if self.parent:
            parent_url = self.parent.make_url()
            return URL(f"{parent_url}/{self.name}")
        return URL(f"/{self.name}")

    def root(self) -> MasterPiece:
        """Fetch the root object

        Returns:
            MasterPiece: root object
        """
        if self.parent:
            return self.parent.root()
        return self

    def resolve_url(self, url: URL) -> Optional["MasterPiece"]:
        """Find a MasterPiece in the hierarchy matching the URL."""
        segments = url.segments

        if len(segments) == 0:
            return self  # Base case: if URL is empty, return self (root)

        if segments[0] == ".":
            # Current object, continue search with the rest of the segments
            return self.resolve_url(URL("/".join(segments[1:])))

        if segments[0] == "..":
            # Parent object, move to the parent (if exists)
            if self.parent:
                # Remove the ".." segment before passing the rest to the parent
                return self.parent.resolve_url(URL("/".join(segments[1:])))
            return None  # No parent to traverse up

        # Regular segment comparison (matching this object's name)
        if segments[0] == self.name:
            if len(segments) == 1:
                return self  # Found a match for the current segment
            # Continue search in parent, remove the first segment from the URL
            if self.parent:
                return self.parent.resolve_url(URL("/".join(segments[1:])))

        return None  # If nothing matches, return None

    def update_metrics(self, elapsed: float) -> None:
        """Updates the number of `update()` methods calld, and time spent in the method.
        This method is called internally from the `update()` method.

        Args:
            elapsed (float): elapsed seconds.
        """
        self._elapsed += elapsed
        self._num_updates += 1

    def acquire_time_spent(self) -> float:
        """Reads off the average time the thread has spent in its `update()` method since
        the last call, and resets the statistics for the next measurement cycle. This method can be used
        for monitoring healtiness of the thread.

        Returns:
            elapsed (float): elapsed seconds.
        """
        if self._num_updates > 0:
            time_spent: float = self._elapsed * 1000.0 / self._num_updates
            self._num_updates = 0
            self._elapsed = 0.0
            return time_spent
        else:
            return 0.0


# Register MasterPiece manually since __init_subclass__() won't be called on it.
MasterPiece.register()