phantop
/
quatalog-scraper
mirror of https://github.com/quatalog/quatalog.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make everything stderr print

This commit is contained in:
powe97 2024-03-05 19:03:54 -05:00
parent 42dbf3c19a
commit 52fdab6ce6
No known key found for this signature in database
GPG Key ID: 7D1663B10978D1BA
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
transfer_scraper/main.py
View File

@ -51,7 +51,7 @@ def jump_to_page(curr_page, to_page, postback_type, pagination_type):
return 1, page return 1, page
if to_page > num_pages or to_page < 1: if to_page > num_pages or to_page < 1:
raise ValueError(f"to_page was out of range ({to_page} not in [1, {num_pages})") raise ValueError(f"to_page was out of range ({to_page} not in [1, {num_pages}])")
while curr_page != to_page: while curr_page != to_page:
jumpable_pages = { jumpable_pages = {
int(x.get_attribute("href").split("'")[3][5:]): x int(x.get_attribute("href").split("'")[3][5:]): x
@ -91,7 +91,7 @@ def scrape_page(page_num):
By.CSS_SELECTOR, "a[id^=gdvInstWithEQ_btnCreditFromInstName_]" By.CSS_SELECTOR, "a[id^=gdvInstWithEQ_btnCreditFromInstName_]"
) )
) )
print(f"Scraping page {page_num}, found {num_institutions} links") print(f"Scraping page {page_num}, found {num_institutions} links", file=sys.stderr)
return [scrape_institution(i) for i in range(0, num_institutions)] return [scrape_institution(i) for i in range(0, num_institutions)]
@ -112,7 +112,7 @@ def scrape_institution(index):
inst_name, inst_city = normalize_title(inst_name), normalize_title(inst_city) inst_name, inst_city = normalize_title(inst_name), normalize_title(inst_city)
inst_link.click() inst_link.click()
wait(EC.staleness_of(inst_link)) wait(EC.staleness_of(inst_link))
print(f"Scraping {inst_name} ({inst_city}, {inst_state})") print(f"Scraping {inst_name} ({inst_city}, {inst_state})", file=sys.stderr)
# Add all courses # Add all courses
try: try:
@ -121,9 +121,19 @@ def scrape_institution(index):
) )
except NoSuchElementException: except NoSuchElementException:
num_pages = 1 num_pages = 1
for i in range(1, num_pages + 1):
jump_to_page(max(1, i - 1), i, "gdvCourseEQ", "lblCourseEQPaginationInfo") try:
driver.find_element(By.ID, "gdvCourseEQ_cbxHeaderCheckAll").click() for i in range(1, num_pages + 1):
jump_to_page(max(1, i - 1), i, "gdvCourseEQ", "lblCourseEQPaginationInfo")
driver.find_element(By.ID, "gdvCourseEQ_cbxHeaderCheckAll").click()
except NoSuchElementException:
# Institution has no data
return {
"institution": inst_name,
"city": inst_city,
"state": inst_state,
"courses": [],
}
# Open list # Open list
driver.find_element(By.ID, "btnAddToMyEQList").click() driver.find_element(By.ID, "btnAddToMyEQList").click()
@ -210,13 +220,13 @@ def main():
global driver global driver
if len(sys.argv) != 3: if len(sys.argv) != 3:
print(f"USAGE: python {sys.argv[0]} <page number to scrape> <output file>") print(f"USAGE: python {sys.argv[0]} <page number to scrape> <output file>", file=sys.stderr)
return 1 return 1
PAGE_NUM_TO_SCRAPE = int(sys.argv[1]) PAGE_NUM_TO_SCRAPE = int(sys.argv[1])
OUT_FILENAME = sys.argv[2] OUT_FILENAME = sys.argv[2]
print(f"Setting up selenium Firefox emulator") print(f"Setting up selenium Firefox emulator", file=sys.stderr)
options = webdriver.FirefoxOptions() options = webdriver.FirefoxOptions()
options.add_argument("--headless") options.add_argument("--headless")
@ -226,7 +236,7 @@ def main():
driver = webdriver.Firefox(options=options) driver = webdriver.Firefox(options=options)
print(f"Connecting to the TES Public View") print(f"Connecting to the TES Public View", file=sys.stderr)
driver.get( driver.get(
"https://tes.collegesource.com/publicview/TES_publicview01.aspx?rid=f080a477-bff8-46df-a5b2-25e9affdd4ed&aid=27b576bb-cd07-4e57-84d0-37475fde70ce" "https://tes.collegesource.com/publicview/TES_publicview01.aspx?rid=f080a477-bff8-46df-a5b2-25e9affdd4ed&aid=27b576bb-cd07-4e57-84d0-37475fde70ce"
) )