90 lines
4.2 KiB
Python
90 lines
4.2 KiB
Python
|
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
|