fix: station avatar image

PaiGramTeam · Mar 27, 2024 · 66cbb25 · 66cbb25
1 parent b37691c
commit 66cbb25
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/models/material.py b/models/material.py
@@ -20,7 +20,7 @@ class YattaMaterial(BaseModel):
     """名称"""
     description: str
     """描述"""
-    story: str
+    story: Optional[str] = None
     """故事"""
     rank: int
     """稀有度"""

diff --git a/res_func/avatar.py b/res_func/avatar.py
@@ -47,7 +47,7 @@ def get_third_pic():
         return f'{style[style.find("(") + 1:style.find(")")]}'
 
     third_pic = get_third_pic()
-    text = soup.find("div", {"class": "a6678 a4af5"}).get("style")
+    text = soup.find_all("div", {"class": "a4af5"})[1].get("style")
     four_pic = f'{text[text.find("(") + 2:text.find(")") - 1]}' if text else ""
     first_pic = f'{soup.find("img", {"class": "ac39b a6602"}).get("src")}'
     datas.append(