def scrape_submissions_for_users(*user_ids, from_days=0, to_days=100000):
log.info(f"Scraping submissions for users {user_ids}...")
from_date = datetime.now() - timedelta(days=from_days)
to_date = datetime.now() - timedelta(days=to_days)
log.info(f'Dates between {to_date} and {from_date}...')
handle_dict = {}
for user in user_ids:
judge_id, handle = user.split('/', 1)
handles = __expand_handle(judge_id, handle)
handle_dict[judge_id] = handle_dict.get(judge_id, []) + handles
for judge_id, handles in handle_dict.items():
handles = list(set(handles))
log.info(f"Scraping user submissions from judge '{judge_id}':")
log.info(f'Handles: {handles}')
scraper = scrapers.create_scraper(judge_id)
# Get submissions as list of generators.
for handle in handles:
try:
submissions = scraper.scrape_submissions_for_user(handle)
submissions = itertools.takewhile(
lambda x: x['submitted_on'] >= to_date, submissions)
queries.write_submissions(submissions)
except NotImplementedError:
log.warning(
f'Scraping submissions not implemented for {scraper.__class__.__name__}.')
return
except Exception as ex:
log.exception(ex)
def scrape_submissions_for_tasks(*tasks, from_days=0, to_days=100000):
log.info(f"Scraping submissions for tasks {tasks}...")
from_date = datetime.now() - timedelta(days=from_days)
to_date = datetime.now() - timedelta(days=to_days)
log.info(f'Dates between {to_date} and {from_date}...')
task_dict = {}
for task in tasks:
judge_id, task_id = task.split('/', 1)
task_ids = __expand_task(judge_id, task_id)
task_dict[judge_id] = task_dict.get(judge_id, []) + task_ids
for judge_id, task_ids in task_dict.items():
task_ids = list(set(task_ids))
log.info(f"Scraping task submissions from judge '{judge_id}':")
log.info(f'Task ids: {task_ids}')
scraper = scrapers.create_scraper(judge_id)
for task_id in task_ids:
try:
submissions = scraper.scrape_submissions_for_task(task_id)
submissions = itertools.takewhile(
lambda x: x['submitted_on'] >= to_date, submissions)
queries.write_submissions(submissions)
except NotImplementedError:
log.warning(
f'Scraping submissions not implemented for {scraper.__class__.__name__}.')
break
except Exception as ex:
log.exception(ex)
def scrape_recent_submissions(*judge_ids, to_days=1):
to_date = datetime.now() - timedelta(days=to_days)
for judge_id in judge_ids:
try:
scraper = scrapers.create_scraper(judge_id)
submissions = scraper.scrape_recent_submissions()
submissions = itertools.takewhile(
lambda x: x['submitted_on'] >= to_date, submissions)
queries.write_submissions(submissions)
except Exception as ex:
log.error(
f"Exception while fetching recent submissions for {judge_id}")
log.exception(ex)
def scrape_task_info(task):
log.info(f"Scraping task info for task '{task}'...")
judge_id, task_id = task.split('/', 1)
task_ids = __expand_task(judge_id, task_id)
scraper = scrapers.create_scraper(judge_id)
task_infos = []
log.info(f"Task ids: {task_ids}")
for task_id in task_ids:
try:
task_info = scraper.scrape_task_info(task_id)
if task_info is None:
log.warning(
f"Did not find task info for '{task_id}'. Skipping...")
continue
log.debug(task_info)
log.info(
f"Successfully scraped '{task_id}' [{task_info['title']}]...")
try:
statement_info = scraper.scrape_task_statement(task_id)
task_info.update(statement_info)
except NotImplementedError:
log.warning(
f"Could not get statement of task {task_id}: not implemented."
)
except Exception as ex:
log.warning(f"Could not get statement of task {task_id}: {ex}")
task_infos.append(task_info)
except NotImplementedError:
log.warning(
f'Scraping tasks not implemented for {scraper.__class__.__name__}.'
)
return
except Exception as ex:
log.exception(ex)
queries.write_tasks(task_infos)
def scrape_handle_info(handle):
log.info(f"Scraping info for handle '{handle}'...")
judge_id, handle_id = handle.split('/', 1)
handles = __expand_handle(judge_id, handle_id)
log.info(f"Handles: {handles}")
scraper = scrapers.create_scraper(judge_id)
user_infos = []
for handle in handles:
try:
user_info = scraper.scrape_user_info(handle)
log.info(f"Successfully scraped user info for '{handle}'")
log.debug(user_info)
if user_info:
user_infos.append(user_info)
except NotImplementedError:
log.warning(
f'Scraping handles not implemented for {scraper.__class__.__name__}.')
return
except Exception as ex:
log.exception(ex)
queries.write_handles(user_infos)