PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-16 04:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Update /material source

Browse Source
This commit is contained in:
LittleMengBot 2023-04-02 12:36:27 +08:00 committed by GitHub
parent e7662240ad
commit d9cb6f6164
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/services/game/services.py
View File

@ -56,7 +56,7 @@ class GameMaterialService(BaseService):
def __init__(self, cache: GameCacheForMaterial, collections: Optional[List[int]] = None): def __init__(self, cache: GameCacheForMaterial, collections: Optional[List[int]] = None):
self._cache = cache self._cache = cache
self._hyperion = Hyperion() self._hyperion = Hyperion()
self._collections = [428421, 1164644] if collections is None else collections self._collections = [428421, 1362644] if collections is None else collections
self._special = ["雷电将军", "珊瑚宫心海", "菲谢尔", "托马", "八重神子", "九条裟罗", "辛焱", "神里绫华"] self._special = ["雷电将军", "珊瑚宫心海", "菲谢尔", "托马", "八重神子", "九条裟罗", "辛焱", "神里绫华"]
async def _get_material_from_hyperion(self, collection_id: int, character_name: str) -> int: async def _get_material_from_hyperion(self, collection_id: int, character_name: str) -> int:
@ -91,11 +91,11 @@ class GameMaterialService(BaseService):
artwork_info = await self._hyperion.get_post_info(2, post_id) artwork_info = await self._hyperion.get_post_info(2, post_id)
image_url_list = artwork_info.image_urls image_url_list = artwork_info.image_urls
if collection_id == 1362644 or character_name in self._special:
image_url_list.pop(0)
await self._cache.set_url_list(character_name, image_url_list) await self._cache.set_url_list(character_name, image_url_list)
if len(image_url_list) == 0: if len(image_url_list) == 0:
return "" return ""
if len(image_url_list) == 1: if len(image_url_list) == 1:
return image_url_list[0] return image_url_list[0]
if character_name in self._special:
return image_url_list[2]
return image_url_list[1] return image_url_list[1]