91 lines
2.0 KiB
Python
91 lines
2.0 KiB
Python
from dataclasses import dataclass
|
|
from typing import Optional, Collection, Any
|
|
from abc import ABC, abstractmethod
|
|
from enum import Enum, auto
|
|
|
|
class DeviceLifecycleState(Enum):
|
|
INIT = "init"
|
|
OPEN = "open"
|
|
CLOSE = "close"
|
|
|
|
|
|
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
|
|
|
|
def close(self):
|
|
return super().close() |