all
或者全部
: /avatars all
、全部练度查询
+ diff --git a/plugins/genshin/avatar_list.py b/plugins/genshin/avatar_list.py
new file mode 100644
index 0000000..ab84e4b
--- /dev/null
+++ b/plugins/genshin/avatar_list.py
@@ -0,0 +1,200 @@
+"""练度统计"""
+from typing import Iterable, List, Optional, Sequence
+
+from arkowrapper import ArkoWrapper
+from enkanetwork import Assets as EnkaAssets, EnkaNetworkAPI
+from genshin import Client
+from genshin.models import CalculatorCharacterDetails, CalculatorTalent, Character
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup, InputFile, Message, Update, User
+from telegram.constants import ChatAction, ParseMode
+from telegram.ext import CallbackContext, filters
+
+from core.base.assets import AssetsService
+from core.baseplugin import BasePlugin
+from core.config import config
+from core.cookies.error import CookiesNotFoundError
+from core.cookies.services import CookiesService
+from core.plugin import Plugin, handler
+from core.template import TemplateService
+from core.user.error import UserNotFoundError
+from metadata.genshin import AVATAR_DATA, NAMECARD_DATA
+from modules.wiki.base import Model
+from utils.decorators.error import error_callable
+from utils.decorators.restricts import restricts
+from utils.helpers import get_genshin_client
+from utils.log import logger
+
+
+class AvatarListPlugin(Plugin, BasePlugin):
+ def __init__(
+ self, cookies_service: CookiesService, assets_service: AssetsService, template_service: TemplateService
+ ) -> None:
+ self.cookies_service = cookies_service
+ self.assets_service = assets_service
+ self.template_service = template_service
+ self.enka_client = EnkaNetworkAPI(lang="chs", agent=config.enka_network_api_agent)
+ self.enka_assets = EnkaAssets(lang="chs")
+
+ async def get_user_client(self, user: User, message: Message, context: CallbackContext) -> Optional[Client]:
+ try:
+ return await get_genshin_client(user.id)
+ except UserNotFoundError: # 若未找到账号
+ if filters.ChatType.GROUPS.filter(message):
+ buttons = [[InlineKeyboardButton("点我私聊", url=f"https://t.me/{context.bot.username}?start=set_uid")]]
+ reply_msg = await message.reply_text(
+ "未查询到您所绑定的账号信息,请先私聊派蒙绑定账号", reply_markup=InlineKeyboardMarkup(buttons)
+ )
+ self._add_delete_message_job(context, reply_msg.chat_id, reply_msg.message_id, 30)
+ self._add_delete_message_job(context, message.chat_id, message.message_id, 30)
+ else:
+ await message.reply_text("未查询到您所绑定的账号信息,请先私聊派蒙绑定账号")
+ except CookiesNotFoundError:
+ if filters.ChatType.GROUPS.filter(message):
+ buttons = [[InlineKeyboardButton("点我私聊", url=f"https://t.me/{context.bot.username}?start=set_uid")]]
+ reply_msg = await message.reply_text(
+ "此功能需要绑定cookie
后使用,请先私聊派蒙绑定账号",
+ reply_markup=InlineKeyboardMarkup(buttons),
+ parse_mode=ParseMode.HTML,
+ )
+ self._add_delete_message_job(context, reply_msg.chat_id, reply_msg.message_id, 30)
+ self._add_delete_message_job(context, message.chat_id, message.message_id, 30)
+ else:
+ await message.reply_text("此功能需要绑定cookie
后使用,请先私聊派蒙进行绑定", parse_mode=ParseMode.HTML)
+
+ async def get_avatars_data(self, characters: Sequence[Character], client: Client, max_length: int = None):
+ avatar_datas: List[AvatarData] = []
+ for num, character in enumerate(characters):
+ if num == max_length: # 若已经有 max_length 个角色
+ break
+ detail = await client.get_character_details(character)
+ if character.id == 10000005: # 针对男草主
+ talents = []
+ for talent in detail.talents:
+ if "普通攻击" in talent.name:
+ talent.Config.allow_mutation = True
+ # noinspection Pydantic
+ talent.group_id = 1131
+ if talent.type in ["attack", "skill", "burst"]:
+ talents.append(talent)
+ else:
+ talents = [t for t in detail.talents if t.type in ["attack", "skill", "burst"]]
+ buffed_talents = []
+ for constellation in filter(lambda x: x.pos in [3, 5], character.constellations[: character.constellation]):
+ if result := list(
+ filter(lambda x: all([x.name in constellation.effect]), talents) # pylint: disable=W0640
+ ):
+ buffed_talents.append(result[0].type)
+ avatar_datas.append(
+ AvatarData(
+ avatar=character,
+ detail=detail,
+ icon=(await self.assets_service.avatar(character.id).side()).as_uri(),
+ weapon=(
+ await self.assets_service.weapon(character.weapon.id).__getattr__(
+ "icon" if character.weapon.ascension < 2 else "awaken"
+ )()
+ ).as_uri(),
+ skills=[
+ SkillData(skill=s, buffed=s.type in buffed_talents)
+ for s in sorted(talents, key=lambda x: ["attack", "skill", "burst"].index(x.type))
+ ],
+ )
+ )
+ return avatar_datas
+
+ async def get_final_data(self, client: Client, characters: Sequence[Character], update: Update):
+ try:
+ response = await self.enka_client.fetch_user(client.uid)
+ namecard = (await self.assets_service.namecard(response.player.namecard.id).navbar()).as_uri()
+ avatar = (await self.assets_service.avatar(response.player.icon.id).icon()).as_uri()
+ nickname = response.player.nickname
+ rarity = {k: v["rank"] for k, v in AVATAR_DATA.items()}[str(response.player.icon.id)]
+ except Exception as e: # pylint: disable=W0703
+ logger.debug(f"enka 请求失败: {e}")
+ choices = ArkoWrapper(characters).filter(lambda x: x.friendship == 10) # 筛选出好感满了的角色
+ if not choices: # 若没有满好感角色、则以好感等级排序
+ choices = ArkoWrapper(characters).sort(lambda x: x.friendship, reverse=True)
+ namecard_choices = ( # 找到与角色对应的满好感名片ID
+ ArkoWrapper(choices)
+ .map(lambda x: next(filter(lambda y: y["name"].split(".")[0] == x.name, NAMECARD_DATA.values()), None))
+ .filter(lambda x: x)
+ .map(lambda x: x["id"])
+ )
+ namecard = (await self.assets_service.namecard(namecard_choices[0]).navbar()).as_uri()
+ avatar = (await self.assets_service.avatar(cid := choices[0].id).icon()).as_uri()
+ nickname = update.effective_user.full_name
+ rarity = {k: v["rank"] for k, v in AVATAR_DATA.items()}[str(cid)]
+ return namecard, avatar, nickname, rarity
+
+ @handler.command("avatars", filters.Regex(r"^/avatars\s*(?:(\d+)|(all))?$"))
+ @handler.message(filters.Regex(r"^(全部)?练度统计$"))
+ @restricts(30)
+ @error_callable
+ async def avatar_list(self, update: Update, context: CallbackContext):
+ user = update.effective_user
+ message = update.effective_message
+
+ args = context.match
+
+ all_avatars = any(["all" in args.groups(), "全部" in args.groups()]) # 是否发送全部角色
+
+ logger.info(f"用户 {user.full_name}[{user.id}] [bold]练度统计[/bold]: all={all_avatars}", extra={"markup": True})
+
+ client = await self.get_user_client(user, message, context)
+ if not client:
+ return
+
+ notice = await message.reply_text("派蒙需要收集整理数据,还请耐心等待哦~")
+ await message.reply_chat_action(ChatAction.TYPING)
+
+ characters = await client.get_genshin_characters(client.uid)
+
+ avatar_datas: List[AvatarData] = await self.get_avatars_data(characters, client, None if all_avatars else 20)
+
+ namecard, avatar, nickname, rarity = await self.get_final_data(client, characters, update)
+
+ render_data = {
+ "uid": client.uid, # 玩家uid
+ "nickname": nickname, # 玩家昵称
+ "avatar": avatar, # 玩家头像
+ "rarity": rarity, # 玩家头像对应的角色星级
+ "namecard": namecard, # 玩家名片
+ "avatar_datas": avatar_datas, # 角色数据
+ "has_more": len(characters) != len(avatar_datas), # 是否显示了全部角色
+ }
+
+ await message.reply_chat_action(ChatAction.UPLOAD_DOCUMENT if all_avatars else ChatAction.UPLOAD_PHOTO)
+
+ image = await self.template_service.render(
+ "genshin/avatar_list/main.html",
+ render_data,
+ viewport={"width": 1040, "height": 500},
+ full_page=True,
+ query_selector=".container",
+ )
+ self._add_delete_message_job(context, notice.chat_id, notice.message_id, 5)
+ if all_avatars and len(characters) > 20:
+ await message.reply_document(InputFile(image, filename="练度统计.png"))
+ else:
+ await message.reply_photo(image)
+
+ logger.info(
+ f"用户 {user.full_name}[{user.id}] [bold]练度统计[/bold]发送{'文件' if all_avatars else '图片'}成功",
+ extra={"markup": True},
+ )
+
+
+class SkillData(Model):
+ """天赋数据"""
+
+ skill: CalculatorTalent
+ buffed: bool = False
+ """是否得到了命座加成"""
+
+
+class AvatarData(Model):
+ avatar: Character
+ detail: CalculatorCharacterDetails
+ icon: str
+ weapon: str
+ skills: Iterable[SkillData]
diff --git a/poetry.lock b/poetry.lock
index def33e8..4ea6e87 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -159,7 +159,7 @@ python-versions = ">=3.5"
dev = ["cloudpickle", "coverage[toml] (>=5.0.2)", "furo", "hypothesis", "mypy (>=0.900,!=0.940)", "pre-commit", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins", "sphinx", "sphinx-notfound-page", "zope.interface"]
docs = ["furo", "sphinx", "sphinx-notfound-page", "zope.interface"]
tests = ["cloudpickle", "coverage[toml] (>=5.0.2)", "hypothesis", "mypy (>=0.900,!=0.940)", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins", "zope.interface"]
-tests-no-zope = ["cloudpickle", "coverage[toml] (>=5.0.2)", "hypothesis", "mypy (>=0.900,!=0.940)", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins"]
+tests_no_zope = ["cloudpickle", "coverage[toml] (>=5.0.2)", "hypothesis", "mypy (>=0.900,!=0.940)", "pympler", "pytest (>=4.3.0)", "pytest-mypy-plugins"]
[[package]]
name = "backports.zoneinfo"
@@ -234,7 +234,7 @@ optional = false
python-versions = ">=3.6.0"
[package.extras]
-unicode-backport = ["unicodedata2"]
+unicode_backport = ["unicodedata2"]
[[package]]
name = "click"
@@ -366,7 +366,7 @@ python-versions = ">=3.7"
[[package]]
name = "genshin"
version = "1.2.4"
-description = ""
+description = "An API wrapper for Genshin Impact."
category = "main"
optional = false
python-versions = ">=3.8"
@@ -386,7 +386,7 @@ geetest = ["rsa"]
type = "git"
url = "https://github.com/thesadru/genshin.py"
reference = "HEAD"
-resolved_reference = "d6aa54384cbcb260adab1b221d0f8673ac0caf4a"
+resolved_reference = "7b3a4a71bfdf84d9f1bf984e91c0bcf73f9dfa7f"
[[package]]
name = "greenlet"
@@ -960,19 +960,19 @@ aiomysql = ["aiomysql", "greenlet (!=0.4.17)"]
aiosqlite = ["aiosqlite", "greenlet (!=0.4.17)", "typing_extensions (!=3.10.0.1)"]
asyncio = ["greenlet (!=0.4.17)"]
asyncmy = ["asyncmy (>=0.2.3,!=0.2.4)", "greenlet (!=0.4.17)"]
-mariadb-connector = ["mariadb (>=1.0.1,!=1.1.2)"]
+mariadb_connector = ["mariadb (>=1.0.1,!=1.1.2)"]
mssql = ["pyodbc"]
-mssql-pymssql = ["pymssql"]
-mssql-pyodbc = ["pyodbc"]
+mssql_pymssql = ["pymssql"]
+mssql_pyodbc = ["pyodbc"]
mypy = ["mypy (>=0.910)", "sqlalchemy2-stubs"]
mysql = ["mysqlclient (>=1.4.0)", "mysqlclient (>=1.4.0,<2)"]
-mysql-connector = ["mysql-connector-python"]
+mysql_connector = ["mysql-connector-python"]
oracle = ["cx_oracle (>=7)", "cx_oracle (>=7,<8)"]
postgresql = ["psycopg2 (>=2.7)"]
-postgresql-asyncpg = ["asyncpg", "greenlet (!=0.4.17)"]
-postgresql-pg8000 = ["pg8000 (>=1.16.6,!=1.29.0)"]
-postgresql-psycopg2binary = ["psycopg2-binary"]
-postgresql-psycopg2cffi = ["psycopg2cffi"]
+postgresql_asyncpg = ["asyncpg", "greenlet (!=0.4.17)"]
+postgresql_pg8000 = ["pg8000 (>=1.16.6,!=1.29.0)"]
+postgresql_psycopg2binary = ["psycopg2-binary"]
+postgresql_psycopg2cffi = ["psycopg2cffi"]
pymysql = ["pymysql", "pymysql (<1)"]
sqlcipher = ["sqlcipher3_binary"]
diff --git a/resources/genshin/avatar_list/example.html b/resources/genshin/avatar_list/example.html
new file mode 100644
index 0000000..926597a
--- /dev/null
+++ b/resources/genshin/avatar_list/example.html
@@ -0,0 +1,144 @@
+
+
+
all
或者全部
: /avatars all
、全部练度查询
+ all
或者全部
: /avatars all
、全部练度查询
+