forked from Hay1tsme/artemis
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from typing import Dict, List
|
|
from titles.wacca.handlers.base import BaseRequest, BaseResponse
|
|
from titles.wacca.handlers.helpers import StageInfo, StageupClearType, GenericItemRecv
|
|
|
|
|
|
# --user/trial/get--
|
|
class UserTrialGetRequest(BaseRequest):
|
|
profileId: int = 0
|
|
|
|
def __init__(self, data: Dict) -> None:
|
|
super().__init__(data)
|
|
self.profileId = self.params[0]
|
|
|
|
|
|
class UserTrialGetResponse(BaseResponse):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
self.stageList: List[StageInfo] = []
|
|
|
|
def make(self) -> Dict:
|
|
dans = []
|
|
for x in self.stageList:
|
|
dans.append(x.make())
|
|
|
|
self.params = [dans]
|
|
return super().make()
|
|
|
|
|
|
# --user/trial/update--
|
|
class UserTrialUpdateRequest(BaseRequest):
|
|
def __init__(self, data: Dict) -> None:
|
|
super().__init__(data)
|
|
self.profileId: int = self.params[0]
|
|
self.stageId: int = self.params[1]
|
|
self.stageLevel: int = self.params[2]
|
|
self.clearType = StageupClearType(self.params[3])
|
|
self.songScores: List[int] = self.params[4]
|
|
self.numSongsCleared: int = self.params[5]
|
|
self.itemsObtained: List[GenericItemRecv] = []
|
|
self.unk7: List = []
|
|
|
|
for x in self.params[6]:
|
|
self.itemsObtained.append(GenericItemRecv(x[0], x[1], x[2]))
|
|
|
|
if len(self.params) == 8:
|
|
self.unk7 = self.params[7]
|
|
|
|
|
|
class UserTrialUpdateResponse(BaseResponse):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
def make(self) -> Dict:
|
|
return super().make()
|