mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-07-15 17:08:10 +00:00
2fd4d7e2e0
Add the replace_icon_name_with_url parameter to the client's fetch_user and fetch_user_v1 methods.
108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
from typing import Final, TypeVar
|
|
|
|
from .models import Character, StarrailInfoParsed
|
|
from .models.v1 import Character, StarrailInfoParsedV1
|
|
|
|
RawData = TypeVar("RawData")
|
|
ParsedData = TypeVar("ParsedData", StarrailInfoParsed, StarrailInfoParsedV1)
|
|
|
|
ASSET_URL: Final[str] = "https://raw.githubusercontent.com/Mar-7th/StarRailRes/master"
|
|
|
|
|
|
def remove_empty_dict(data: RawData) -> RawData:
|
|
"""
|
|
Recursively removes empty dictionaries from the given raw data.
|
|
|
|
Args:
|
|
- data (`RawData`): The input raw data.
|
|
|
|
Returns:
|
|
- `RawData`: The data with empty dictionaries removed.
|
|
"""
|
|
if isinstance(data, dict):
|
|
for key in data.keys():
|
|
data[key] = None if (data[key] == {}) else remove_empty_dict(data[key])
|
|
elif isinstance(data, list):
|
|
for i in range(len(data)):
|
|
data[i] = remove_empty_dict(data[i])
|
|
return data
|
|
|
|
|
|
def replace_icon_name_with_url(data: RawData) -> RawData:
|
|
"""
|
|
Replaces icon file names with asset URLs in the given raw data.
|
|
|
|
Example: Replace "/icon/avatar/1201.png" with
|
|
"https://raw.githubusercontent.com/Mar-7th/StarRailRes/master/icon/avatar/1201.png"
|
|
|
|
Args:
|
|
- data (`RawData`): The input raw data.
|
|
|
|
Returns:
|
|
- `RawData`: The data with icon file names replaced by asset URLs.
|
|
"""
|
|
if isinstance(data, dict):
|
|
for key in data.keys():
|
|
data[key] = replace_icon_name_with_url(data[key])
|
|
elif isinstance(data, list):
|
|
for i in range(len(data)):
|
|
data[i] = replace_icon_name_with_url(data[i])
|
|
elif isinstance(data, str):
|
|
if ".png" in data:
|
|
data = ASSET_URL + "/" + data
|
|
return data
|
|
|
|
|
|
def replace_trailblazer_name(data: StarrailInfoParsedV1) -> StarrailInfoParsedV1:
|
|
"""
|
|
Replaces the trailblazer name with the player's name.
|
|
|
|
Args:
|
|
- data (`StarrailInfoParsed`): The input StarrailInfoParsed data.
|
|
|
|
Returns:
|
|
- `StarrailInfoParsed`: The updated StarrailInfoParsed data.
|
|
"""
|
|
for i in range(len(data.characters)):
|
|
if data.characters[i].name == r"{NICKNAME}":
|
|
data.characters[i].name = data.player.name
|
|
return data
|
|
|
|
|
|
def remove_duplicate_character(data: ParsedData) -> ParsedData:
|
|
"""
|
|
Removes duplicate characters from the given StarrailInfoParsed data.
|
|
|
|
Args:
|
|
- data (`ParsedData`): The input StarrailInfoParsed data.
|
|
|
|
Returns:
|
|
- `ParsedData`: The updated StarrailInfoParsed data without duplicate characters.
|
|
"""
|
|
new_characters = []
|
|
characters_ids: set[str] = set()
|
|
for character in data.characters:
|
|
if character.id not in characters_ids:
|
|
new_characters.append(character)
|
|
characters_ids.add(character.id)
|
|
data.characters = new_characters
|
|
return data
|
|
|
|
|
|
def merge_character_data(new_data: ParsedData, old_data: ParsedData) -> ParsedData:
|
|
"""
|
|
Append the old data characters to the list of new data characters.
|
|
The player's info from the old data will be omitted/discarded.
|
|
|
|
Args:
|
|
- new_data (`ParsedData`): The new data to be merged.
|
|
- old_data (`ParsedData`): The old data to merge into.
|
|
|
|
Returns:
|
|
- `ParsedData`: The merged new data.
|
|
"""
|
|
for character in old_data.characters:
|
|
new_data.characters.append(character)
|
|
new_data = remove_duplicate_character(new_data)
|
|
return new_data
|