fix(gacha): Manually specify banner type

Wuthery · Jun 10, 2024 · dbca3cd · dbca3cd
1 parent 1ccc8c3
commit dbca3cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kuro/client/components/gacha.py b/kuro/client/components/gacha.py
@@ -55,6 +55,7 @@ async def get_gacha_record(
             errors.raise_from_data(rsp)
 
         return [
-            GachaWeapon(**record) if record["resourceId"] >= 100000 else GachaCharacter(**record)
+            GachaWeapon(**{**record, "cardPoolType": banner}) if record["resourceId"] >= 100000
+            else GachaCharacter(**{**record, "cardPoolType": banner})
             for record in rsp["data"]
         ]