fix: pins

Author: MalikAza

pyZUnivers/pins.py

@@ -1,5 +1,5 @@
 from .pack import Pack
-from typing import Union, List
+from typing import Literal, Union, List
 
 class UserPin:
     """
@@ -11,20 +11,19 @@ class UserPin:
         type (str): The type of the pin.
         rarity (int): The rarity of the pin.
         identifier (int): The identifier of the pin.
-        description (str): The description of the pin.
-        reference (str): The reference of the pin.
         pack (Pack): The pack of the pin.
         image_urls (List[str]): The image urls of the pin.
+        shiny_level (Literal['Normal', 'Golden', 'Shiny']): The shiny level of the pin.
         score (int): The score of the pin.
-        score_golden (int): The score of the golden pin.
         is_recyclable (bool): Whether the pin is recyclable.
         is_tradable (bool): Whether the pin is tradable.
         is_counting (bool): Whether the pin is counting.
         is_craftable (bool): Whether the pin is craftable.
         is_invocable (bool): Whether the pin is invocable.
         is_goldable (bool): Whether the pin is goldable.
         is_upgradable (bool): Whether the pin is upgradable.
-        is_golden (bool): Whether the pin is golden.
+        is_golden (bool): Whether the pin is golden or not.
+        is_shiny (bool): Whether the pin is shiny or not.
     """
 
     def __init__(self, payload) -> None:
@@ -51,14 +50,6 @@ def rarity(self) -> int:
     def identifier(self) -> int:
         return self.__item['identifier']
 
-    @property
-    def description(self) -> Union[str, None]:
-        return self.__item['description']
-    
-    @property
-    def reference(self) -> Union[str, None]:
-        return self.__item['reference']
-    
     @property
     def pack(self) -> Pack:
         return Pack(self.__item['pack'])
@@ -68,12 +59,20 @@ def image_urls(self) -> List[str]:
         return [x for x in self.__item['urls']]
 
     @property
-    def score(self) -> int:
-        return self.__item['score']
-    
+    def shiny_level(self) -> Literal['Normal', 'Golden', 'Shiny']:
+        self.__shiny_index = self.__payload['shinyLevel']
+
+        match self.__shiny_index:
+            case 0:
+                return 'Normal'
+            case 1:
+                return 'Golden'
+            case 2:
+                return 'Shiny'
+
     @property
-    def score_golden(self) -> int:
-        return self.__item['scoreGolden']
+    def score(self) -> int:
+        return self.__item['scores'][f'{self.__shiny_index}']
 
     @property
     def is_recyclable(self) -> bool:
@@ -105,4 +104,8 @@ def is_upgradable(self) -> bool:
 
     @property
     def is_golden(self) -> bool:
-        return self.__payload['isGolden']
+        return self.__shiny_index == 1
+    
+    @property
+    def is_shiny(self) -> bool:
+        return self.__shiny_index == 2

tests/test_user_overview.py

@@ -23,14 +23,12 @@ def test_pins(self) -> None:
             self.assertIsInstance(pin.type, str)
             self.assertIsInstance(pin.rarity, int)
             self.assertIsInstance(pin.identifier, int)
-            self.assertIsInstance(pin.description, Union[str, None])
-            self.assertIsInstance(pin.reference, Union[str, None])
             self.assertIsInstance(pin.pack, Pack)
             self.assertIsInstance(pin.image_urls, List)
             for url in pin.image_urls:
                 self.assertIsInstance(url, str)
+            self.assertIsInstance(pin.shiny_level, str)
             self.assertIsInstance(pin.score, int)
-            self.assertIsInstance(pin.score_golden, int)
             self.assertIsInstance(pin.is_recyclable, bool)
             self.assertIsInstance(pin.is_tradable, bool)
             self.assertIsInstance(pin.is_counting, bool)
@@ -39,6 +37,7 @@ def test_pins(self) -> None:
             self.assertIsInstance(pin.is_goldable, bool)
             self.assertIsInstance(pin.is_upgradable, bool)
             self.assertIsInstance(pin.is_golden, bool)
+            self.assertIsInstance(pin.is_shiny, bool)
 
     def test_invocations_before_pity(self) -> None:
         self.assertIsInstance(self.overview_powaza.invocations_before_pity, int)