95df85419b
Fixes potential boolean checks on the relevant attributes
67 lines
1.2 KiB
Python
67 lines
1.2 KiB
Python
from dataclasses import dataclass
|
|
|
|
|
|
class Actions:
|
|
undefined = 0
|
|
finished = 1
|
|
error = 2
|
|
message = 3
|
|
state_update = 4
|
|
|
|
|
|
@dataclass
|
|
class BaseModel:
|
|
action: int
|
|
app_name: str
|
|
|
|
@classmethod
|
|
def from_json(cls, data: dict):
|
|
return cls(
|
|
action=data["action"],
|
|
app_name=data["app_name"]
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class FinishedModel(BaseModel):
|
|
exit_code: int
|
|
playtime: int # seconds
|
|
|
|
@classmethod
|
|
def from_json(cls, data):
|
|
return cls(
|
|
**BaseModel.from_json(data).__dict__,
|
|
exit_code=data["exit_code"],
|
|
playtime=data["playtime"],
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class StateChangedModel(BaseModel):
|
|
class States:
|
|
started = 1
|
|
# for future
|
|
syncing_cloud = 2
|
|
cloud_sync_finished = 3
|
|
|
|
new_state: int
|
|
|
|
@classmethod
|
|
def from_json(cls, data):
|
|
return cls(
|
|
action=data["action"],
|
|
app_name=data["app_name"],
|
|
new_state=data["new_state"]
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class ErrorModel(BaseModel):
|
|
error_string: str
|
|
|
|
@classmethod
|
|
def from_json(cls, data):
|
|
return cls(
|
|
**BaseModel.from_json(data).__dict__,
|
|
error_string=data["error_string"]
|
|
)
|