style: update head icon model

PaiGramTeam · Mar 28, 2024 · d2f1956 · d2f1956
1 parent 21eb9a9
commit d2f1956
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/models/head_icon.py b/models/head_icon.py
@@ -12,14 +12,17 @@ class HeadIcon(BaseModel):
     icons: List[str]
 
     @property
-    def icon(self) -> str:
-        for icon in self.icons:
-            if icon:
-                return icon
+    def webp(self) -> str:
+        return self.icons[0]
+
+    @property
+    def png(self) -> str:
+        return self.icons[1]
 
 
 # 原始数据
 
+
 class ForHash(BaseModel):
     Hash: str