mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-30 03:16:08 +00:00
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
import datetime
|
||
|
|
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from module.config.utils import DEFAULT_TIME
|
||
|
|
from module.logger import logger
|
||
|
|
|
||
|
|
|
||
|
|
def now():
|
||
|
|
return datetime.datetime.now().replace(microsecond=0)
|
||
|
|
|
||
|
|
|
||
|
|
class StateValue(BaseModel):
|
||
|
|
value: int = 0
|
||
|
|
time: datetime.datetime = DEFAULT_TIME
|
||
|
|
|
||
|
|
|
||
|
|
TrailblazePowerMax = 180
|
||
|
|
ImmersifierMax = 8
|
||
|
|
|
||
|
|
|
||
|
|
class StateStorage(BaseModel):
|
||
|
|
TrailblazePower = StateValue()
|
||
|
|
Immersifier = StateValue()
|
||
|
|
|
||
|
|
def __setattr__(self, key, value):
|
||
|
|
if key in super().__getattribute__('__fields__'):
|
||
|
|
storage = super().__getattribute__(key)
|
||
|
|
storage.value = value
|
||
|
|
storage.time = now()
|
||
|
|
else:
|
||
|
|
super().__setattr__(key, value)
|
||
|
|
|
||
|
|
def __getattribute__(self, item):
|
||
|
|
if item in super().__getattribute__('__fields__'):
|
||
|
|
storage = super().__getattribute__(item)
|
||
|
|
if storage.time == DEFAULT_TIME:
|
||
|
|
logger.warning(f'Trying to get state {item} but it is never set')
|
||
|
|
return storage.value
|
||
|
|
else:
|
||
|
|
return super().__getattribute__(item)
|
||
|
|
|
||
|
|
|
||
|
|
class StateMixin:
|
||
|
|
state = StateStorage()
|