164 lines
6.2 KiB
Python
164 lines
6.2 KiB
Python
from selenium import webdriver
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.print_page_options import PrintOptions
|
|
from selenium.webdriver.firefox.service import Service as FirefoxService
|
|
from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
|
|
from webdriver_manager.firefox import GeckoDriverManager
|
|
from pypdf import PdfWriter
|
|
import configparser
|
|
import os
|
|
import base64
|
|
import errno
|
|
import glob
|
|
|
|
GECKO_DRIVER = 'gecko'
|
|
CHROME_DRIVER = 'chrome'
|
|
|
|
|
|
# HACK
|
|
os.environ["TMPDIR"] = "./tmp"
|
|
|
|
config = configparser.ConfigParser()
|
|
|
|
config.read('./config.ini')
|
|
output_dir = config['main']['output_directory']
|
|
try:
|
|
os.makedirs(output_dir)
|
|
except OSError as exception:
|
|
files = glob.glob(os.path.join(output_dir, "*.pdf"))
|
|
for f in files:
|
|
os.remove(f)
|
|
if exception.errno != errno.EEXIST:
|
|
raise
|
|
|
|
driver_type = config['main']['driver']
|
|
# TODO only firefox for now
|
|
if driver_type == GECKO_DRIVER:
|
|
opt = webdriver.FirefoxOptions()
|
|
opt.binary_location = "/usr/bin/firefox"
|
|
opt.add_argument("-headless") # Here
|
|
firefox_profile = FirefoxProfile()
|
|
firefox_profile.set_preference("print.enabled", False)
|
|
opt.profile = firefox_profile
|
|
driver = webdriver.Firefox(options=opt, service=FirefoxService(GeckoDriverManager().install()))
|
|
else:
|
|
print(format('unsupported driver type "%s"', driver_type))
|
|
os.exit(1)
|
|
|
|
url = 'https://app.flightschedulepro.com/Account/Login?company='+ config['main']['tenant_id']
|
|
driver.get(url)
|
|
|
|
# wait for stupid cookie modal to load and accept all ¯\_(ツ)_/¯
|
|
try:
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.ID, 'onetrust-accept-btn-handler'))
|
|
)
|
|
cookie_btn = driver.find_element(By.ID, 'onetrust-accept-btn-handler')
|
|
cookie_btn.click()
|
|
except:
|
|
pass
|
|
|
|
# Login
|
|
form = driver.find_element(By.CLASS_NAME, 'account-form')
|
|
username = driver.find_element(By.ID, 'username')
|
|
username.send_keys(config['main']['username'])
|
|
username.submit()
|
|
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.ID, 'password'))
|
|
)
|
|
password = driver.find_element(By.ID, 'password')
|
|
password.send_keys(config['main']['password'])
|
|
|
|
password.submit()
|
|
|
|
# Wait for app to load and naviagte to students "page"
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.CLASS_NAME, 'fsp-main-drawer-content'))
|
|
)
|
|
driver.get('https://app.flightschedulepro.com/App/Students')
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.CLASS_NAME, 'clickable-course'))
|
|
)
|
|
|
|
# Get all the student rows and iterate
|
|
students = driver.find_elements(By.CLASS_NAME, 'clickable-course')
|
|
student_count = len(students)
|
|
|
|
print(f'processing {student_count} students')
|
|
|
|
original_window = driver.current_window_handle
|
|
for i in range(0, student_count):
|
|
# Need to reload student elements after returning to page
|
|
students = driver.find_elements(By.CLASS_NAME, 'clickable-course')
|
|
if len(students) is not student_count:
|
|
raise 'student count changed. aborting...'
|
|
student = students[i]
|
|
student_name = student.find_element(By.XPATH, '//div[@class="student"]/div[@class="bold"]').get_attribute("innerText")
|
|
student_name_no_space = student_name.replace(' ', '_')
|
|
course_td = student.find_element(By.CLASS_NAME, 'course-td')
|
|
course_td.click()
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.XPATH, '//span[text()[contains(., "Sessions")]]'))
|
|
)
|
|
sessions_tab = driver.find_element(By.XPATH, '//span[text()[contains(., "Sessions")]]')
|
|
sessions_tab.click()
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.XPATH, '//fsp-ui-button[@Text="View"]'))
|
|
)
|
|
# Ugh... Sessions Table reloads on modal opening after the first modal is opened. So we have to record
|
|
# the count of sessions initially and iterate while reloading the elements each time a button is clicked
|
|
view_btns = driver.find_elements(By.XPATH, '//fsp-ui-button[@Text="View"]')
|
|
btn_count = len(view_btns)
|
|
pdf_count = 0
|
|
print(f'Downloading {btn_count} sessions for student "{student_name}"')
|
|
for j in range(0, btn_count):
|
|
view_btns = driver.find_elements(By.XPATH, '//fsp-ui-button[@Text="View"]')
|
|
if len(view_btns) is not btn_count:
|
|
raise 'session count changed. aborting...'
|
|
view_btns[j].click()
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_all_elements_located((By.XPATH, '//fsp-ui-button[@Text="Print"]'))
|
|
)
|
|
print_btn = driver.find_element(By.XPATH, '//fsp-ui-button[@Text="Print"]')
|
|
print_btn.click()
|
|
|
|
# Wait for the new window or tab
|
|
WebDriverWait(driver, 10).until(EC.number_of_windows_to_be(2))
|
|
|
|
# Loop through until we find a new window handle
|
|
for window_handle in driver.window_handles:
|
|
if window_handle != original_window:
|
|
driver.switch_to.window(window_handle)
|
|
break
|
|
# Wait for the new tab to finish loading content
|
|
WebDriverWait(driver, 10).until(EC.title_is('Print Training Session'))
|
|
print_options = PrintOptions()
|
|
print_options.orientation = "portrait"
|
|
pdf = driver.print_page(print_options)
|
|
with open(os.path.join(output_dir, student_name_no_space + "{:03d}".format(pdf_count) + '.pdf'), 'wb') as file:
|
|
file.write(base64.decodebytes(pdf.encode('utf-8')))
|
|
driver.close()
|
|
|
|
driver.switch_to.window(original_window)
|
|
WebDriverWait(driver, 10).until(EC.title_is('Flight Schedule Pro'))
|
|
close_btn = driver.find_element(By.XPATH, '//fsp-ui-button[@Text="Close"]')
|
|
close_btn.click()
|
|
pdf_count = pdf_count + 1
|
|
# Merge all Session PDFs for the student
|
|
print(f'Merging PDFs into "{student_name_no_space}.pdf"')
|
|
driver.close()
|
|
writer = PdfWriter()
|
|
|
|
pdfs = [a for a in os.listdir(output_dir) if a.endswith(".pdf") and a.startswith(student_name_no_space)]
|
|
pdfs.sort()
|
|
|
|
for pdf in pdfs:
|
|
writer.append(os.path.join(output_dir, pdf))
|
|
|
|
writer.write(student_name_no_space + '.pdf')
|
|
writer.close()
|
|
|