Add missing price properties

PokemonTCG · Mar 26, 2021 · 77b926e · 77b926e
1 parent bd2a415
commit 77b926e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pokemontcgsdk/querybuilder.py b/pokemontcgsdk/querybuilder.py
@@ -18,6 +18,13 @@ def find(self, id):
         """
         url = "{}/{}/{}".format(__endpoint__, self.type.RESOURCE, id)
         response = RestClient.get(url)['data']
+
+        # Transform json keys into names that are safe for python properties
+        if response.get('tcgplayer', {}).get('prices', {}).get('1stEditionNormal'):
+            response['tcgplayer']['prices']['firstEditionNormal'] = response['tcgplayer']['prices'].pop('1stEditionNormal')
+        if response.get('tcgplayer', {}).get('prices', {}).get('1stEditionHolofoil'):
+            response['tcgplayer']['prices']['firstEditionHolofoil'] = response['tcgplayer']['prices'].pop('1stEditionHolofoil')
+
         return from_dict(self.type, response)
 
     def where(self, **kwargs):

diff --git a/pokemontcgsdk/tcgplayer.py b/pokemontcgsdk/tcgplayer.py
@@ -14,6 +14,9 @@ class TCGPrices():
     normal: Optional[TCGPrice]
     holofoil: Optional[TCGPrice]
     reverseHolofoil: Optional[TCGPrice]
+    firstEditionHolofoil: Optional[TCGPrice]
+    firstEditionNormal: Optional[TCGPrice]
+
 
 @dataclass
 class TCGPlayer():