mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-22 22:37:21 +00:00
211 lines
8.8 KiB
Python
211 lines
8.8 KiB
Python
import os
|
||
from enum import Enum
|
||
from typing import Optional
|
||
|
||
import httpx
|
||
import ujson
|
||
from bs4 import BeautifulSoup
|
||
|
||
from .helpers import get_headers, get_id_form_url
|
||
|
||
|
||
class WeaponType(Enum):
|
||
Sword = "sword" # 单手剑
|
||
Claymore = "claymore" # 双手剑
|
||
PoleArm = "polearm" # 长柄武器
|
||
Bow = "bow" # 弓
|
||
Catalyst = "catalyst" # 法器
|
||
|
||
|
||
class Weapons:
|
||
IGNORE_WEAPONS_ID = [
|
||
"1001", "1101", "1406",
|
||
"2001", "2101", "2204", "2406", "2407",
|
||
"3001", "3101", "3204", "3404",
|
||
"4001", "4101", "4201", "4403", "4405", "4406",
|
||
"5001", "5101", "5201", "5404", "5404", "5405",
|
||
] # 忽略的武器包括一星、二星武器，beta表格内无名武器，未上架到正服的武器
|
||
|
||
# 根地址
|
||
ROOT_URL = "https://genshin.honeyhunterworld.com"
|
||
|
||
TEXT_MAPPING = {
|
||
"Type": "类型",
|
||
"Rarity": "Rarity",
|
||
"Base Attack": "基础攻击力"
|
||
}
|
||
|
||
WEAPON_TYPE_MAPPING = {
|
||
"Sword": "https://genshin.honeyhunterworld.com/img/skills/s_33101.png", # 单手剑
|
||
"Claymore": "https://genshin.honeyhunterworld.com/img/skills/s_163101.png", # 双手剑
|
||
"Polearm": "https://genshin.honeyhunterworld.com/img/skills/s_233101.png", # 长枪
|
||
"Bow": "https://genshin.honeyhunterworld.com/img/skills/s_213101.png", # 弓箭
|
||
"Catalyst": "https://genshin.honeyhunterworld.com/img/skills/s_43101.png", # 法器
|
||
}
|
||
|
||
def __init__(self):
|
||
self.client = httpx.AsyncClient(headers=get_headers())
|
||
project_path = os.path.dirname(__file__)
|
||
characters_file = os.path.join(project_path, "metadata", "ascension.json")
|
||
monster_file = os.path.join(project_path, "metadata", "monster.json")
|
||
elite_file = os.path.join(project_path, "metadata", "elite.json")
|
||
with open(characters_file, "r", encoding="utf-8") as f:
|
||
self._ascension_json: dict = ujson.load(f)
|
||
with open(monster_file, "r", encoding="utf-8") as f:
|
||
self._monster_json: dict = ujson.load(f)
|
||
with open(elite_file, "r", encoding="utf-8") as f:
|
||
self._elite_json: dict = ujson.load(f)
|
||
|
||
async def _get_soup(self, url: str) -> Optional[BeautifulSoup]:
|
||
request = await self.client.get(url)
|
||
return BeautifulSoup(request.text, "lxml")
|
||
|
||
async def get_weapon_url_list(self, weapon_type: WeaponType):
|
||
weapon_url_list = []
|
||
url = self.ROOT_URL + f"/db/weapon/{weapon_type.value}/?lang=CHS"
|
||
soup = await self._get_soup(url)
|
||
weapon_table = soup.find("span", {"class": "item_secondary_title"},
|
||
string="Released (Codex) Weapons").find_next_sibling()
|
||
weapon_table_rows = weapon_table.find_all("tr")
|
||
for weapon_table_row in weapon_table_rows:
|
||
content = weapon_table_row.find_all("td")[2]
|
||
if content.find("a") is not None:
|
||
weapon_url = self.ROOT_URL + content.find("a")["href"]
|
||
weapon_id = str(get_id_form_url(weapon_url))
|
||
if weapon_id not in self.IGNORE_WEAPONS_ID:
|
||
weapon_url_list.append(weapon_url)
|
||
return weapon_url_list
|
||
|
||
async def get_all_weapon_url(self):
|
||
all_weapon_url = []
|
||
temp_data = await self.get_weapon_url_list(WeaponType.Bow)
|
||
all_weapon_url.extend(temp_data)
|
||
temp_data = await self.get_weapon_url_list(WeaponType.Sword)
|
||
all_weapon_url.extend(temp_data)
|
||
temp_data = await self.get_weapon_url_list(WeaponType.PoleArm)
|
||
all_weapon_url.extend(temp_data)
|
||
temp_data = await self.get_weapon_url_list(WeaponType.Catalyst)
|
||
all_weapon_url.extend(temp_data)
|
||
temp_data = await self.get_weapon_url_list(WeaponType.Claymore)
|
||
all_weapon_url.extend(temp_data)
|
||
return all_weapon_url
|
||
|
||
@staticmethod
|
||
def get_weapon_info_template():
|
||
weapon_info_dict = {
|
||
"name": "",
|
||
"description": "",
|
||
"source_img": "",
|
||
"atk":
|
||
{
|
||
"min": 0,
|
||
"max": 999999,
|
||
"name": "基础攻击力"
|
||
},
|
||
"secondary":
|
||
{
|
||
"min": 0.1,
|
||
"max": 999999.9,
|
||
"name": ""
|
||
},
|
||
"star":
|
||
{
|
||
"value": -1,
|
||
"icon": ""
|
||
},
|
||
"type":
|
||
{
|
||
"name": "",
|
||
"icon": ""
|
||
},
|
||
"passive_ability":
|
||
{
|
||
"name": "",
|
||
"description": ""
|
||
}
|
||
}
|
||
materials_dict = {
|
||
"name": "",
|
||
"star": {
|
||
"value": 0,
|
||
"icon": ""
|
||
},
|
||
"city": "",
|
||
"icon": ""
|
||
}
|
||
weapon_info_dict["materials"] = {
|
||
"ascension": materials_dict,
|
||
"elite": materials_dict,
|
||
"monster": materials_dict,
|
||
}
|
||
return weapon_info_dict
|
||
|
||
async def get_weapon_info(self, url: str):
|
||
weapon_info_dict = self.get_weapon_info_template()
|
||
soup = await self._get_soup(url)
|
||
weapon_content = soup.find("div", {"class": "wrappercont"})
|
||
data = weapon_content.find("div", {"class": "data_cont_wrapper", "style": "display: block"})
|
||
weapon_info = data.find("table", {"class": "item_main_table"})
|
||
weapon_name = weapon_content.find("div", {"class": "custom_title"}).text.replace("-", "").replace(" ", "")
|
||
weapon_info_dict["name"] = weapon_name
|
||
weapon_info_row = weapon_info.find_all("tr")
|
||
for weapon_info_ in weapon_info_row:
|
||
content = weapon_info_.find_all("td")
|
||
if len(content) == 3: # 第一行会有三个td，其中一个td是武器图片
|
||
weapon_info_dict["source_img"] = self.ROOT_URL + content[0].find("img",
|
||
{"class": "itempic lazy"})["data-src"]
|
||
weapon_info_dict["type"]["name"] = content[2].text
|
||
weapon_info_dict["type"]["icon"] = self.get_weapon_type(content[2].text)
|
||
elif len(content) == 2:
|
||
if content[0].text == "Rarity":
|
||
weapon_info_dict["star"]["value"] = len(
|
||
content[1].find_all("div", {"class": "sea_char_stars_wrap"}))
|
||
elif content[0].text == "Special (passive) Ability":
|
||
weapon_info_dict["passive_ability"]["name"] = content[1].text
|
||
elif content[0].text == "Special (passive) Ability Description":
|
||
weapon_info_dict["passive_ability"]["description"] = content[1].text
|
||
elif content[0].text == "In-game Description":
|
||
weapon_info_dict["description"] = content[1].text
|
||
elif content[0].text == "Secondary Stat":
|
||
weapon_info_dict["secondary"]["name"] = content[1].text
|
||
|
||
stat_table = data.find("span", {"class": "item_secondary_title"},
|
||
string=" Stat Progression ").find_next_sibling()
|
||
stat_table_row = stat_table.find_all("tr")
|
||
for stat_table_ in stat_table_row:
|
||
content = stat_table_.find_all("td")
|
||
# 通过等级判断
|
||
if content[0].text == "1":
|
||
weapon_info_dict["atk"]["min"] = int(content[1].text)
|
||
weapon_info_dict["secondary"]["min"] = float(content[2].text)
|
||
elif content[0].text == "80+":
|
||
item_hrefs = content[3].find_all("a")
|
||
for item_href in item_hrefs:
|
||
item_id = get_id_form_url(item_href["href"])
|
||
ascension = self.get_ascension(str(item_id))
|
||
if ascension.get("name") is not None:
|
||
weapon_info_dict["materials"]["ascension"] = ascension
|
||
monster = self.get_monster(str(item_id))
|
||
if monster.get("name") is not None:
|
||
weapon_info_dict["materials"]["monster"] = monster
|
||
elite = self.get_elite(str(item_id))
|
||
if elite.get("name") is not None:
|
||
weapon_info_dict["materials"]["elite"] = elite
|
||
elif content[0].text == "90":
|
||
weapon_info_dict["atk"]["max"] = int(content[1].text)
|
||
weapon_info_dict["secondary"]["max"] = float(content[2].text)
|
||
|
||
return weapon_info_dict
|
||
|
||
def get_ascension(self, item_id: str):
|
||
return self._ascension_json.get(item_id, {})
|
||
|
||
def get_monster(self, item_id: str):
|
||
return self._monster_json.get(item_id, {})
|
||
|
||
def get_elite(self, item_id: str):
|
||
return self._elite_json.get(item_id, {})
|
||
|
||
def get_weapon_type(self, weapon_type: str):
|
||
return self.WEAPON_TYPE_MAPPING.get(weapon_type, "")
|