Vanilla/artemis
1
0
Fork 0
forked from Hay1tsme/artemis
Code Pull Requests Activity

diva: add game_init and attend handlers

Browse Source
This commit is contained in:
Hay1tsme
2023-03-13 04:04:08 -04:00
parent 3076bdf575
commit 21c9b23617
3 changed files with 62 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
titles/diva/handlers/base.py
View File

@ -1,6 +1,6 @@
from urllib import parse
from datetime import datetime
from typing import Union
from typing import Union, Dict
class DivaRequestParseException(Exception):
@ -45,53 +45,53 @@ class BaseRequest:
def __init__(self, raw: Union[str, bytes]) -> None:
self.raw = raw
self.raw_dict = dict(parse.parse_qsl(raw))
self.raw_dict: Dict[bytes, bytes] = dict(parse.parse_qsl(raw))
if "cmd" not in self.raw_dict:
if b"cmd" not in self.raw_dict:
raise DivaRequestParseException(f"cmd not in request data {self.raw_dict}")
if "req_id" not in self.raw_dict:
if b"req_id" not in self.raw_dict:
raise DivaRequestParseException(
f"req_id not in request data {self.raw_dict}"
)
if "place_id" not in self.raw_dict:
if b"place_id" not in self.raw_dict:
raise DivaRequestParseException(
f"place_id not in request data {self.raw_dict}"
)
if "start_up_mode" not in self.raw_dict:
if b"start_up_mode" not in self.raw_dict:
raise DivaRequestParseException(
f"start_up_mode not in request data {self.raw_dict}"
)
if "cmm_dly_mod" not in self.raw_dict:
if b"cmm_dly_mod" not in self.raw_dict:
raise DivaRequestParseException(
f"cmm_dly_mod not in request data {self.raw_dict}"
)
if "cmm_dly_sec" not in self.raw_dict:
if b"cmm_dly_sec" not in self.raw_dict:
raise DivaRequestParseException(
f"cmm_dly_sec not in request data {self.raw_dict}"
)
if "cmm_err_mod" not in self.raw_dict:
if b"cmm_err_mod" not in self.raw_dict:
raise DivaRequestParseException(
f"cmm_err_mod not in request data {self.raw_dict}"
)
if "region_code" not in self.raw_dict:
if b"region_code" not in self.raw_dict:
raise DivaRequestParseException(
f"region_code not in request data {self.raw_dict}"
)
if "time_stamp" not in self.raw_dict:
if b"time_stamp" not in self.raw_dict:
raise DivaRequestParseException(
f"time_stamp not in request data {self.raw_dict}"
)
for k, v in self.raw_dict:
setattr(self, k, v)
for k, v in self.raw_dict.items():
setattr(self, k.decode(), v.decode())
self.place_id = int(self.place_id)
self.start_up_mode = int(self.start_up_mode)
@ -110,17 +110,34 @@ class BaseResponse:
self.stat = "ok"
def make(self) -> str:
all_vars = vars(self)
all_vars.pop("cmd")
all_vars.pop("req_id")
all_vars.pop("stat")
return f"cmd={self.cmd}&req_id={self.req_id}&stat={self.stat}"
ret = f"cmd={self.cmd}&req_id={self.req_id}&stat={self.stat}"
additional_params = parse.urlencode(
all_vars, doseq=True, quote_via=parse.quote, safe=","
)
class GameInitRequest(BaseRequest):
def __init__(self, raw: Union[str, bytes]) -> None:
super().__init__(raw)
if additional_params != "":
ret += f"&{additional_params}"
class AttendRequest(BaseRequest):
def __init__(self, raw: Union[str, bytes]) -> None:
super().__init__(raw)
self.power_on = int(self.power_on)
self.is_bb = bool(int(self.power_on))
class AttendResponse(BaseResponse):
def __init__(self, cmd_id: str, req_id: int) -> None:
super().__init__(cmd_id, req_id)
self.atnd_prm1 = [1] * 100
self.atnd_prm2 = [1] * 100
self.atnd_prm3 = [1] * 100
self.atnd_lut = datetime.now()
def make(self) -> str:
ret = super().make()
ret_dict = {
"atnd_prm1": ','.join([str(i) for i in self.atnd_prm1]),
"atnd_prm2": ','.join([str(i) for i in self.atnd_prm2]),
"atnd_prm3": ','.join([str(i) for i in self.atnd_prm3]),
"atnd_lut": parse.quote(self.atnd_lut.strftime('%Y-%m-%d %H:%M:%S:16.0'))
}
ret += "&" + parse.urlencode(ret_dict, safe=",")
return ret