添加QQ官方机器人头像获取

StarRailUID/starrailuid_abyss/__init__.py

@@ -63,6 +63,6 @@ async def send_srabyss_info(bot: Bot, ev: Event):
     else:
         floor = None
     await bot.logger.info(f'[sr查询深渊信息]深渊层数: {floor}')
-    im = await draw_abyss_img(user_id, uid, floor, schedule_type)
+    im = await draw_abyss_img(user_id, uid, floor, schedule_type, ev.sender)
     await bot.send(im)
     return None

StarRailUID/starrailuid_abyss/draw_abyss_card.py

@@ -152,6 +152,7 @@ async def draw_abyss_img(
     uid: str,
     floor: Optional[int] = None,
     schedule_type: str = '1',
+    sender,
 ) -> Union[bytes, str]:
     raw_abyss_data = await mys_api.get_srspiral_abyss_info(uid, schedule_type)
 
@@ -185,6 +186,8 @@ async def draw_abyss_img(
     _id = str(qid)
     if _id.startswith('http'):
         char_pic = await get_qq_avatar(avatar_url=_id)
+    elif sender.get('avatar') is not None:
+        char_pic = await get_qq_avatar(avatar_url=sender['avatar'])
     else:
         char_pic = await get_qq_avatar(qid=qid)
     char_pic = await draw_pic_with_ring(char_pic, 250, None, False)

StarRailUID/starrailuid_rogue/__init__.py

@@ -63,7 +63,7 @@ async def send_srabyss_info(bot: Bot, ev: Event):
     else:
         floor = None
     await bot.logger.info(f'[sr查询模拟宇宙信息]模拟宇宙世界数: {floor}')
-    im = await draw_rogue_img(user_id, uid, floor, schedule_type)
+    im = await draw_rogue_img(user_id, uid, floor, schedule_type, ev.sender)
     await bot.send(im)
     return None
 
@@ -90,6 +90,6 @@ async def send_srabyss_locust_info(bot: Bot, ev: Event):
     if uid is None:
         return await bot.send(UID_HINT)
     await bot.logger.info(f'[sr查询寰宇蝗灾信息]uid: {uid}')
-    im = await draw_rogue_locust_img(user_id, uid)
+    im = await draw_rogue_locust_img(user_id, uid, ev.sender)
     await bot.send(im)
     return None

StarRailUID/starrailuid_rogue/draw_rogue_card.py

@@ -250,6 +250,7 @@ async def draw_rogue_img(
     uid: str,
     floor: Optional[int] = None,
     schedule_type: str = '3',
+    sender,
 ) -> Union[bytes, str]:
     raw_rogue_data = await mys_api.get_rogue_info(uid, '3')
 
@@ -325,6 +326,8 @@ async def draw_rogue_img(
     _id = str(qid)
     if _id.startswith('http'):
         char_pic = await get_qq_avatar(avatar_url=_id)
+    elif sender.get('avatar') is not None:
+        char_pic = await get_qq_avatar(avatar_url=sender['avatar'])
     else:
         char_pic = await get_qq_avatar(qid=qid)
     char_pic = await draw_pic_with_ring(char_pic, 250, None, False)
@@ -533,6 +536,7 @@ async def draw_rogue_img(
 async def draw_rogue_locust_img(
     qid: Union[str, int],
     uid: str,
+    sender,
 ) -> Union[bytes, str]:
     raw_rogue_data = await mys_api.get_rogue_locust_info(uid, '3')
 
@@ -608,6 +612,8 @@ async def draw_rogue_locust_img(
     _id = str(qid)
     if _id.startswith('http'):
         char_pic = await get_qq_avatar(avatar_url=_id)
+    elif sender.get('avatar') is not None:
+        char_pic = await get_qq_avatar(avatar_url=sender['avatar'])
     else:
         char_pic = await get_qq_avatar(qid=qid)
     char_pic = await draw_pic_with_ring(char_pic, 250, None, False)

StarRailUID/starrailuid_roleinfo/__init__.py

@@ -43,5 +43,5 @@ async def send_detail_info(bot: Bot, ev: Event):
 
     logger.info(f'[sr查询信息]UID: {uid}')
     await bot.logger.info('开始执行[sr查询信息]')
-    await bot.send(await get_detail_img(user_id, uid))
+    await bot.send(await get_detail_img(user_id, uid, ev.sender))
     return None

StarRailUID/starrailuid_roleinfo/draw_roleinfo_card.py

@@ -70,8 +70,8 @@ async def get_role_img(uid: str) -> Union[bytes, str]:
     return await draw_role_card(uid)
 
 
-async def get_detail_img(qid: Union[str, int], uid: str) -> Union[bytes, str]:
+async def get_detail_img(qid: Union[str, int], uid: str, sender) -> Union[bytes, str]:
-    return await get_detail_card(qid, uid)
+    return await get_detail_card(qid, uid, sender)
 
 
 def _lv(level: int) -> str:
@@ -376,7 +376,7 @@ async def _draw_detail_card(
 
 
 async def get_detail_card(
-    qid: Union[str, int], sr_uid: str
+    qid: Union[str, int], sr_uid: str, sender
 ) -> Union[bytes, str]:
     # 获取角色列表
     avatar_list = await mys_api.get_avatar_info(sr_uid, 1001)
@@ -398,6 +398,8 @@ async def get_detail_card(
     _id = str(qid)
     if _id.startswith('http'):
         char_pic = await get_qq_avatar(avatar_url=_id)
+    elif sender.get('avatar') is not None:
+        char_pic = await get_qq_avatar(avatar_url=sender['avatar'])
     else:
         char_pic = await get_qq_avatar(qid=qid)
     char_pic = await draw_pic_with_ring(char_pic, 250, None, False)