refactoring and fixes

src/giveaway.py

@@ -0,0 +1,90 @@
+import re
+import log
+
+logger = log.get_logger(__name__)
+
+
+class Giveaway:
+
+    def __init__(self, soup_item):
+        self.soup_item = soup_item
+        self.game_name = None
+        self.game_id = None
+        self.pinned = False
+        self.game_cost = None
+        self.game_entries = None
+        self.copies = None
+        self.time_remaining_string = None
+        self.time_remaining_in_minutes = None
+        self.time_created_string = None
+        self.time_created_in_minutes = None
+
+        self.game_name = soup_item.find('a', {'class': 'giveaway__heading__name'}).text
+        self.game_id = soup_item.find('a', {'class': 'giveaway__heading__name'})['href'].split('/')[2]
+        pin_class = soup_item.parent.parent.get("class")
+        self.pinned = pin_class is not None and len(pin_class) == 1 and pin_class[0].find('pinned') != -1
+        self.game_cost, self.copies = self.determine_cost_and_copies(self.soup_item, self.game_name, self.game_id)
+        self.game_entries = int(soup_item.select('div.giveaway__links span')[0].text.split(' ')[0].replace(',', ''))
+        times = soup_item.select('div span[data-timestamp]')
+        self.time_remaining_string = times[0].text
+        self.time_remaining_in_minutes = self.determine_time_in_minutes(self.time_remaining_string)
+        self.time_created_string = times[1].text
+        self.time_created_in_minutes = self.determine_time_in_minutes(self.time_created_string)
+
+    # this isn't exact because 'a week' could mean 8 days or 'a day' could mean 27 hours
+    def determine_time_in_minutes(self, string_time):
+        if not string_time:
+            logger.error(f"Could not determine time from string {string_time}")
+            return None
+        match = re.search('(?P<number>[0-9]+) (?P<time_unit>(hour|day|minute|second|week))', string_time)
+        if match:
+            number = int(match.group('number'))
+            time_unit = match.group('time_unit')
+            if time_unit == 'hour':
+                return number * 60
+            elif time_unit == 'day':
+                return number * 24 * 60
+            elif time_unit == 'minute':
+                return number
+            elif time_unit == 'second':
+                return 1
+            elif time_unit == 'week':
+                return number * 7 * 24 * 60
+            else:
+                logger.error(f"Unknown time unit displayed in giveaway: {string_time}")
+                return None
+        else:
+            return None
+
+    def determine_cost_and_copies(self, item, game_name, game_id):
+        item_headers = item.find_all('span', {'class': 'giveaway__heading__thin'})
+        if len(item_headers) == 1:  # then no multiple copies
+            game_cost = item_headers[0].getText().replace('(', '').replace(')', '').replace('P', '')
+            if not re.search('^[0-9]+$', game_cost):
+                txt = f"Unable to determine cost of {game_name} with id {game_id}. Cost string: {item_headers[0]}"
+                logger.error(txt)
+                return None, None
+            game_cost = int(game_cost)
+            return game_cost, 1
+        elif len(item_headers) == 2:  # then multiple copies
+            game_cost = item_headers[1].getText().replace('(', '').replace(')', '').replace('P', '')
+            if not re.search('^[0-9]+$', game_cost):
+                txt = f"Unable to determine cost of {game_name} with id {game_id}. Cost string: {item_headers[1].getText()}"
+                logger.error(txt)
+                return None, None
+            game_cost = int(game_cost)
+
+            match = re.search('(?P<copies>[0-9]+) Copies', item_headers[0].getText(), re.IGNORECASE)
+            if match:
+                num_copies_str = match.group('copies')
+                num_copies = int(num_copies_str)
+                return game_cost, num_copies
+            else:
+                txt = f"It appears there are multiple copies of {game_name} with id {game_id}, but we could not " \
+                      f"determine that. Copy string: {item_headers[0].getText()}"
+                logger.error(txt)
+                return game_cost, 1
+        else:
+            txt = f"Unable to determine cost or num copies of {game_name} with id {game_id}."
+            logger.error(txt)
+            return None, None