Adding discount price

Mirio · Nov 13, 2017 · 0372ed6 · 0372ed6
1 parent 33a83f5
commit 0372ed6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
     name='steamstoreprice',
     packages=['steamstoreprice'],
     package_dir={'steamstoreprice': 'steamstoreprice'},
-    version='0.1',
+    version='0.2',
     install_requires=['requests', 'beautifulsoup4'],
     description='Find the price on Steam Store from url',
     author='Alessandro Sbarbati',

diff --git a/steamstoreprice/steamstoreprice.py b/steamstoreprice/steamstoreprice.py
@@ -59,4 +59,7 @@ def getprice(self, url):
         :return: float(price cleaned)
         """
         body_content = self.getpage(self.normalizeurl(url))
-        return self.normalizeprice(body_content.find("div", {"class": "game_purchase_price"}).contents[0])
+        try:
+            return self.normalizeprice(body_content.find("div", {"class": "game_purchase_price"}).contents[0])
+        except AttributeError:
+            return self.normalizeprice(body_content.find("div", {"class": "discount_final_price"}).contents[0])