advanced-python-homework-2023-marmi/equipment/device.py

from dataclasses import dataclass
from typing import Optional, Collection, Any
from abc import ABC, abstractmethod

from enum import IntEnum

class DeviceLifecycleState(IntEnum):
    INIT = 0
    OPEN = 1
    CLOSE = 2


class DevaceError(Exception):
    pass


class NonReadableTrait(Exception):
    pass


class NonWritableTrait(Exception):
    pass


@dataclass
class TraitDescriptor:
    name: str
    info: Optional[str] = None
    readable: bool = True
    writable: bool = False


@dataclass
class ActionDescriptor:
    name: str
    arguments: dict[str, type]
    info: Optional[str] = None


class Device(ABC):
    
    _state = DeviceLifecycleState.INIT

    @property
    def state(self) -> DeviceLifecycleState:
        return self._state

    def close(self):
        self._state = DeviceLifecycleState.CLOSE
        
    @property
    @abstractmethod
    def trait_descriptors(self) -> Collection[TraitDescriptor]:
        pass
    
    @property
    @abstractmethod
    def action_descriptors(self) -> Collection[ActionDescriptor]:
        pass

    @abstractmethod
    def __getitem__(self, trait_name: str) -> Optional[Any]:
        """Return logical state of trait `trait_name`."""
        pass


class SynchronyDevice(Device):

    def open(self):
        self._state = DeviceLifecycleState.OPEN

    @abstractmethod
    def execute(self, action_name: str, *args, **kwargs):
        """Execute action `action_name`, using `args` and `kwargs` as action argument."""
        pass

    @abstractmethod
    def read(self, trait_name: str) -> Any:
        """Read physical state of trait `trait_name` from device."""
        raise NonReadableTrait

    @abstractmethod
    def write(self, trait_name: str, value: Any) -> bool:
        """Pass `value` to trait `trait_name` of device."""
        raise NonWritableTrait

    @abstractmethod
    def invalidate(self, trait_name: str):
        """Invalidate logical state of trait `trait_name`"""
        pass