463 lines
16 KiB
Python
463 lines
16 KiB
Python
"""Prettifiers for genshinstats api returns.
|
||
|
||
Fixes the huge problem of outdated field names in the api,
|
||
that were leftover from during development
|
||
"""
|
||
import re
|
||
from datetime import datetime
|
||
|
||
character_icons = {
|
||
"PlayerGirl": "Traveler",
|
||
"PlayerBoy": "Traveler",
|
||
"Ambor": "Amber",
|
||
"Qin": "Jean",
|
||
"Hutao": "Hu Tao",
|
||
"Feiyan": "Yanfei",
|
||
"Kazuha": "Kadehara Kazuha",
|
||
"Sara": "Kujou Sara",
|
||
"Shougun": "Raiden Shogun",
|
||
"Tohma": "Thoma",
|
||
}
|
||
|
||
|
||
def _recognize_character_icon(url: str) -> str:
|
||
"""Recognizes a character's icon url and returns its name."""
|
||
exp = r"game_record/genshin/character_.*_(\w+)(?:@\dx)?.png"
|
||
match = re.search(exp, url)
|
||
if match is None:
|
||
raise ValueError(f"{url!r} is not a character icon or image url")
|
||
character = match.group(1)
|
||
return character_icons.get(character) or character
|
||
|
||
|
||
def prettify_stats(data):
|
||
s = data["stats"]
|
||
h = data["homes"][0] if data["homes"] else None
|
||
return {
|
||
"stats": {
|
||
"achievements": s["achievement_number"],
|
||
"active_days": s["active_day_number"],
|
||
"characters": s["avatar_number"],
|
||
"spiral_abyss": s["spiral_abyss"],
|
||
"anemoculi": s["anemoculus_number"],
|
||
"geoculi": s["geoculus_number"],
|
||
"electroculi": s["electroculus_number"],
|
||
"common_chests": s["common_chest_number"],
|
||
"exquisite_chests": s["exquisite_chest_number"],
|
||
"precious_chests": s["precious_chest_number"],
|
||
"luxurious_chests": s["luxurious_chest_number"],
|
||
"unlocked_waypoints": s["way_point_number"],
|
||
"unlocked_domains": s["domain_number"],
|
||
},
|
||
"characters": [
|
||
{
|
||
"name": i["name"],
|
||
"rarity": i["rarity"]
|
||
if i["rarity"] < 100
|
||
else i["rarity"] - 100, # aloy has 105 stars
|
||
"element": i["element"],
|
||
"level": i["level"],
|
||
"friendship": i["fetter"],
|
||
"icon": i["image"],
|
||
"id": i["id"],
|
||
}
|
||
for i in data["avatars"]
|
||
],
|
||
"teapot": {
|
||
# only unique data between realms are names and icons
|
||
"realms": [{"name": s["name"], "icon": s["icon"]} for s in data["homes"]],
|
||
"level": h["level"],
|
||
"comfort": h["comfort_num"],
|
||
"comfort_name": h["comfort_level_name"],
|
||
"comfort_icon": h["comfort_level_icon"],
|
||
"items": h["item_num"],
|
||
"visitors": h["visit_num"], # currently not in use
|
||
}
|
||
if h
|
||
else None,
|
||
"explorations": [
|
||
{
|
||
"name": i["name"],
|
||
"explored": round(i["exploration_percentage"] / 10, 1),
|
||
"type": i["type"],
|
||
"level": i["level"],
|
||
"icon": i["icon"],
|
||
"offerings": i["offerings"],
|
||
}
|
||
for i in data["world_explorations"]
|
||
],
|
||
}
|
||
|
||
|
||
def prettify_characters(data):
|
||
return [
|
||
{
|
||
"name": i["name"],
|
||
"rarity": i["rarity"] if i["rarity"] < 100 else i["rarity"] - 100, # aloy has 105 stars
|
||
"element": i["element"],
|
||
"level": i["level"],
|
||
"friendship": i["fetter"],
|
||
"constellation": sum(c["is_actived"] for c in i["constellations"]),
|
||
"icon": i["icon"],
|
||
"image": i["image"],
|
||
"id": i["id"],
|
||
"collab": i["rarity"] >= 100,
|
||
**(
|
||
{"traveler_name": "Aether" if "Boy" in i["icon"] else "Lumine"}
|
||
if "Player" in i["icon"]
|
||
else {}
|
||
),
|
||
"weapon": {
|
||
"name": i["weapon"]["name"],
|
||
"rarity": i["weapon"]["rarity"],
|
||
"type": i["weapon"]["type_name"],
|
||
"level": i["weapon"]["level"],
|
||
"ascension": i["weapon"]["promote_level"],
|
||
"refinement": i["weapon"]["affix_level"],
|
||
"description": i["weapon"]["desc"],
|
||
"icon": i["weapon"]["icon"],
|
||
"id": i["weapon"]["id"],
|
||
},
|
||
"artifacts": [
|
||
{
|
||
"name": a["name"],
|
||
"pos_name": {
|
||
1: "flower",
|
||
2: "feather",
|
||
3: "hourglass",
|
||
4: "goblet",
|
||
5: "crown",
|
||
}[a["pos"]],
|
||
"full_pos_name": a["pos_name"],
|
||
"pos": a["pos"],
|
||
"rarity": a["rarity"],
|
||
"level": a["level"],
|
||
"set": {
|
||
"name": a["set"]["name"],
|
||
"effect_type": ["none", "single", "classic"][len(a["set"]["affixes"])],
|
||
"effects": [
|
||
{
|
||
"pieces": e["activation_number"],
|
||
"effect": e["effect"],
|
||
}
|
||
for e in a["set"]["affixes"]
|
||
],
|
||
"set_id": int(re.search(r"UI_RelicIcon_(\d+)_\d+", a["icon"]).group(1)), # type: ignore
|
||
"id": a["set"]["id"],
|
||
},
|
||
"icon": a["icon"],
|
||
"id": a["id"],
|
||
}
|
||
for a in i["reliquaries"]
|
||
],
|
||
"constellations": [
|
||
{
|
||
"name": c["name"],
|
||
"effect": c["effect"],
|
||
"is_activated": c["is_actived"],
|
||
"index": c["pos"],
|
||
"icon": c["icon"],
|
||
"id": c["id"],
|
||
}
|
||
for c in i["constellations"]
|
||
],
|
||
"outfits": [
|
||
{"name": c["name"], "icon": c["icon"], "id": c["id"]} for c in i["costumes"]
|
||
],
|
||
}
|
||
for i in data
|
||
]
|
||
|
||
|
||
def prettify_abyss(data):
|
||
fchars = lambda d: [
|
||
{
|
||
"value": a["value"],
|
||
"name": _recognize_character_icon(a["avatar_icon"]),
|
||
"rarity": a["rarity"] if a["rarity"] < 100 else a["rarity"] - 100, # aloy has 105 stars
|
||
"icon": a["avatar_icon"],
|
||
"id": a["avatar_id"],
|
||
}
|
||
for a in d
|
||
]
|
||
todate = lambda x: datetime.fromtimestamp(int(x)).strftime("%Y-%m-%d")
|
||
totime = lambda x: datetime.fromtimestamp(int(x)).isoformat(" ")
|
||
return {
|
||
"season": data["schedule_id"],
|
||
"season_start_time": todate(data["start_time"]),
|
||
"season_end_time": todate(data["end_time"]),
|
||
"stats": {
|
||
"total_battles": data["total_battle_times"],
|
||
"total_wins": data["total_win_times"],
|
||
"max_floor": data["max_floor"],
|
||
"total_stars": data["total_star"],
|
||
},
|
||
"character_ranks": {
|
||
"most_played": fchars(data["reveal_rank"]),
|
||
"most_kills": fchars(data["defeat_rank"]),
|
||
"strongest_strike": fchars(data["damage_rank"]),
|
||
"most_damage_taken": fchars(data["take_damage_rank"]),
|
||
"most_bursts_used": fchars(data["normal_skill_rank"]),
|
||
"most_skills_used": fchars(data["energy_skill_rank"]),
|
||
},
|
||
"floors": [
|
||
{
|
||
"floor": f["index"],
|
||
"stars": f["star"],
|
||
"max_stars": f["max_star"],
|
||
"icon": f["icon"],
|
||
"chambers": [
|
||
{
|
||
"chamber": l["index"],
|
||
"stars": l["star"],
|
||
"max_stars": l["max_star"],
|
||
"has_halves": len(l["battles"]) == 2,
|
||
"battles": [
|
||
{
|
||
"half": b["index"],
|
||
"timestamp": totime(b["timestamp"]),
|
||
"characters": [
|
||
{
|
||
"name": _recognize_character_icon(c["icon"]),
|
||
"rarity": c["rarity"]
|
||
if c["rarity"] < 100
|
||
else c["rarity"] - 100, # aloy has 105 stars
|
||
"level": c["level"],
|
||
"icon": c["icon"],
|
||
"id": c["id"],
|
||
}
|
||
for c in b["avatars"]
|
||
],
|
||
}
|
||
for b in l["battles"]
|
||
],
|
||
}
|
||
for l in f["levels"]
|
||
],
|
||
}
|
||
for f in data["floors"]
|
||
],
|
||
}
|
||
|
||
|
||
def prettify_activities(data):
|
||
activities = {
|
||
k: v if v.get("exists_data") else {"records": []}
|
||
for activity in data["activities"]
|
||
for k, v in activity.items()
|
||
}
|
||
return {
|
||
"hyakunin": [
|
||
{
|
||
"id": r["challenge_id"],
|
||
"name": r["challenge_name"],
|
||
"difficulty": r["difficulty"],
|
||
"medal_icon": r["heraldry_icon"],
|
||
"score": r["max_score"],
|
||
"multiplier": r["score_multiple"],
|
||
"lineups": [
|
||
{
|
||
"characters": [
|
||
{
|
||
"name": _recognize_character_icon(c["icon"]),
|
||
"rarity": c["rarity"]
|
||
if c["rarity"] < 100
|
||
else c["rarity"] - 100, # aloy has 105 stars
|
||
"level": c["level"],
|
||
"icon": c["icon"],
|
||
"id": c["id"],
|
||
"trial": c["is_trail_avatar"],
|
||
}
|
||
for c in l["avatars"]
|
||
],
|
||
"skills": [
|
||
{"name": s["name"], "desc": s["desc"], "icon": s["icon"], "id": s["id"]}
|
||
for s in l["skills"]
|
||
],
|
||
}
|
||
for l in r["lineups"]
|
||
],
|
||
}
|
||
for r in activities["sumo"]["records"]
|
||
],
|
||
"labyrinth": None,
|
||
}
|
||
|
||
|
||
def prettify_notes(data):
|
||
return {
|
||
"resin": data["current_resin"],
|
||
"until_resin_limit": data["resin_recovery_time"],
|
||
"max_resin": data["max_resin"],
|
||
"total_commissions": data["total_task_num"],
|
||
"completed_commissions": data["finished_task_num"],
|
||
"claimed_commission_reward": data["is_extra_task_reward_received"],
|
||
"max_boss_discounts": data["resin_discount_num_limit"],
|
||
"remaining_boss_discounts": data["remain_resin_discount_num"],
|
||
"expeditions": [
|
||
{
|
||
"icon": exp["avatar_side_icon"],
|
||
"remaining_time": exp["remained_time"],
|
||
"status": exp["status"],
|
||
}
|
||
for exp in data["expeditions"]
|
||
],
|
||
"max_expeditions": data["max_expedition_num"],
|
||
"realm_currency": data["current_home_coin"],
|
||
"max_realm_currency": data["max_home_coin"],
|
||
"until_realm_currency_limit": data["home_coin_recovery_time"],
|
||
}
|
||
|
||
|
||
def prettify_game_accounts(data):
|
||
return [
|
||
{
|
||
"uid": int(a["game_uid"]),
|
||
"server": a["region_name"],
|
||
"level": a["level"],
|
||
"nickname": a["nickname"],
|
||
# idk what these are for:
|
||
"biz": a["game_biz"],
|
||
"is_chosen": a["is_chosen"],
|
||
"is_official": a["is_official"],
|
||
}
|
||
for a in data
|
||
]
|
||
|
||
|
||
def prettify_wish_history(data, banner_name=None):
|
||
return [
|
||
{
|
||
"type": i["item_type"],
|
||
"name": i["name"],
|
||
"rarity": int(i["rank_type"]),
|
||
"time": i["time"],
|
||
"id": int(i["id"]),
|
||
"banner": banner_name,
|
||
"banner_type": int(i["gacha_type"]),
|
||
"uid": int(i["uid"]),
|
||
}
|
||
for i in data
|
||
]
|
||
|
||
|
||
def prettify_gacha_items(data):
|
||
return [
|
||
{
|
||
"name": i["name"],
|
||
"type": i["item_type"],
|
||
"rarity": int(i["rank_type"]),
|
||
"id": 10000000 + int(i["item_id"]) - 1000
|
||
if len(i["item_id"]) == 4
|
||
else int(i["item_id"]),
|
||
}
|
||
for i in data
|
||
]
|
||
|
||
|
||
def prettify_banner_details(data):
|
||
per = lambda p: None if p == "0%" else float(p[:-1].replace(",", "."))
|
||
fprobs = (
|
||
lambda l: [
|
||
{
|
||
"type": i["item_type"],
|
||
"name": i["item_name"],
|
||
"rarity": int(i["rank"]),
|
||
"is_up": bool(i["is_up"]),
|
||
"order_value": i["order_value"],
|
||
}
|
||
for i in l
|
||
]
|
||
if l
|
||
else []
|
||
)
|
||
fitems = (
|
||
lambda l: [
|
||
{
|
||
"type": i["item_type"],
|
||
"name": i["item_name"],
|
||
"element": {
|
||
"风": "Anemo",
|
||
"火": "Pyro",
|
||
"水": "Hydro",
|
||
"雷": "Electro",
|
||
"冰": "Cryo",
|
||
"岩": "Geo",
|
||
"？": "Dendro",
|
||
"": None,
|
||
}[i["item_attr"]],
|
||
"icon": i["item_img"],
|
||
}
|
||
for i in l
|
||
]
|
||
if l
|
||
else []
|
||
)
|
||
return {
|
||
"banner_type_name": {
|
||
100: "Novice Wishes",
|
||
200: "Permanent Wish",
|
||
301: "Character Event Wish",
|
||
302: "Weapon Event Wish",
|
||
}[int(data["gacha_type"])],
|
||
"banner_type": int(data["gacha_type"]),
|
||
"banner": re.sub(r"<.*?>", "", data["title"]).strip(),
|
||
"title": data["title"],
|
||
"content": data["content"],
|
||
"date_range": data["date_range"],
|
||
"r5_up_prob": per(data["r5_up_prob"]), # probability for rate-up 5*
|
||
"r4_up_prob": per(data["r4_up_prob"]), # probability for rate-up 4*
|
||
"r5_prob": per(data["r5_prob"]), # probability for 5*
|
||
"r4_prob": per(data["r4_prob"]), # probability for 4*
|
||
"r3_prob": per(data["r3_prob"]), # probability for 3*
|
||
"r5_guarantee_prob": per(data["r5_baodi_prob"]), # probability for 5* incl. guarantee
|
||
"r4_guarantee_prob": per(data["r4_baodi_prob"]), # probability for 4* incl. guarantee
|
||
"r3_guarantee_prob": per(data["r3_baodi_prob"]), # probability for 3* incl. guarantee
|
||
"r5_up_items": fitems(
|
||
data["r5_up_items"]
|
||
), # list of 5* rate-up items that you can get from banner
|
||
"r4_up_items": fitems(
|
||
data["r4_up_items"]
|
||
), # list of 4* rate-up items that you can get from banner
|
||
"r5_items": fprobs(data["r5_prob_list"]), # list 5* of items that you can get from banner
|
||
"r4_items": fprobs(data["r4_prob_list"]), # list 4* of items that you can get from banner
|
||
"r3_items": fprobs(data["r3_prob_list"]), # list 3* of items that you can get from banner
|
||
"items": fprobs(
|
||
sorted(
|
||
data["r5_prob_list"] + data["r4_prob_list"] + data["r3_prob_list"],
|
||
key=lambda x: x["order_value"],
|
||
)
|
||
),
|
||
}
|
||
|
||
|
||
def prettify_trans(data, reasons={}):
|
||
if data and "name" in data[0]:
|
||
# transaction item
|
||
return [
|
||
{
|
||
"time": i["time"],
|
||
"name": i["name"],
|
||
"rarity": int(i["rank"]),
|
||
"amount": int(i["add_num"]),
|
||
"reason": reasons.get(int(i["reason"]), ""),
|
||
"reason_id": int(i["reason"]),
|
||
"uid": int(i["uid"]),
|
||
"id": int(i["id"]),
|
||
}
|
||
for i in data
|
||
]
|
||
else:
|
||
# transaction
|
||
return [
|
||
{
|
||
"time": i["time"],
|
||
"amount": int(i["add_num"]),
|
||
"reason": reasons.get(int(i["reason"]), ""),
|
||
"reason_id": int(i["reason"]),
|
||
"uid": int(i["uid"]),
|
||
"id": int(i["id"]),
|
||
}
|
||
for i in data
|
||
]
|