From 77b926e557b03cbb7bd52b4954bb0a64c6f71b18 Mon Sep 17 00:00:00 2001
From: CptSpaceToaster <CptSpaceToaster@gmail.com>
Date: Fri, 26 Mar 2021 12:29:33 -0400
Subject: [PATCH] Add missing price properties

---
 pokemontcgsdk/querybuilder.py | 7 +++++++
 pokemontcgsdk/tcgplayer.py    | 3 +++
 2 files changed, 10 insertions(+)

diff --git a/pokemontcgsdk/querybuilder.py b/pokemontcgsdk/querybuilder.py
index f672209..1beb56d 100644
--- 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
index 3689fb4..7698ebf 100644
--- 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():