clash/mihomo/tools.py
KT c433b5d845 feat: Add data merging functionality to the tools
Added the following functions:
- remove_duplicate_character
- merge_character_data
2023-05-29 19:24:40 +08:00

80 lines
2.4 KiB
Python

from typing import TypeVar
from .models import Character, StarrailInfoParsed
T = TypeVar("T")
def remove_empty_dict(data: T) -> T:
"""
Recursively removes empty dictionaries from the given raw data.
Args:
- data (`T`): The input data.
Returns:
- `T`: 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_trailblazer_name(data: StarrailInfoParsed) -> StarrailInfoParsed:
"""
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: StarrailInfoParsed) -> StarrailInfoParsed:
"""
Removes duplicate characters from the given StarrailInfoParsed data.
Args:
- data (`StarrailInfoParsed`): The input StarrailInfoParsed data.
Returns:
- `StarrailInfoParsed`: The updated StarrailInfoParsed data without duplicate characters.
"""
new_characters: list[Character] = []
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: StarrailInfoParsed, old_data: StarrailInfoParsed
) -> StarrailInfoParsed:
"""
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 (`StarrailInfoParsed`): The new data to be merged.
- old_data (`StarrailInfoParsed`): The old data to merge into.
Returns:
- `StarrailInfoParsed`: 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