mirror of
https://github.com/PaiGramTeam/EnkaNetwork.py.git
synced 2024-09-28 03:26:09 +00:00
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
from pydantic import BaseModel
|
|
|
|
from .character import CharacterInfo
|
|
|
|
from typing import Dict, List, Any
|
|
|
|
class BuildInfo(BaseModel):
|
|
id: int
|
|
name: str = ''
|
|
avatar_id: str = ''
|
|
avatar_data: CharacterInfo
|
|
order: int = 0
|
|
live: bool = False
|
|
settings: Dict[str, Any] = {}
|
|
public: bool = True
|
|
|
|
class Builds(BaseModel):
|
|
raw: Dict[str, List[BuildInfo]] = {}
|
|
|
|
def __init__(self, **data: Any) -> None:
|
|
_data = {}
|
|
for key in data:
|
|
_build = []
|
|
for build in data[key]:
|
|
_build.append(build)
|
|
|
|
_data[key] = _build
|
|
|
|
super().__init__(**{"raw": _data})
|
|
|
|
def get_build_info(self, avatar_id: str, build_id: str):
|
|
buildList = self.raw.get(str(avatar_id), None)
|
|
if buildList is None:
|
|
return None
|
|
|
|
for buildInfo in buildList:
|
|
if str(buildInfo.id) == build_id:
|
|
return buildInfo
|
|
|
|
return None
|
|
|
|
def get_character(self, avatar_id: str):
|
|
return self.raw.get(str(avatar_id), None) |