nocturn9x
/
UniVRLessonBooker
mirror of https://github.com/nocturn9x/UniVRLessonBooker.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Updated GIA login with new frontend

This commit is contained in:
Mattia Giambirtone 2022-09-22 13:51:32 +02:00
parent a0c17df7da
commit 5d705bf897
1 changed files with 27 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/main.py
View File

@ -13,8 +13,10 @@
import sys import sys
import json import json
import httpx import httpx
import sniffio
import logging import logging
import asyncio import asyncio
import warnings
import argparse import argparse
from typing import Union from typing import Union
from hashlib import sha256 from hashlib import sha256
@ -149,17 +151,17 @@ async def login_with_gia(
) )
try: try:
logger.debug("Launching headless browser instance") logger.debug("Launching headless browser instance")
browser = await launch(headless=True) browser = await launch(headless=False, executablePath="/usr/bin/google-chrome-beta")
logger.debug("Headless browser instance launched, opening new page") logger.debug("Headless browser instance launched, opening new page")
page = await browser.newPage() page = await browser.newPage()
logger.debug(f"New page opened, going to redirect URL: {result.url}") logger.debug(f"New page opened, going to redirect URL: {result.url}")
await page.goto(str(result.url)) await page.goto(str(result.url))
logger.debug("Page loaded, typing credentials") logger.debug("Page loaded, typing credentials")
# Types username and password with 100ms delay between key presses # Types username and password with 100ms delay between key presses
await page.type('#IDToken1', username, delay=100) await page.type('#form_username.form-control', username, delay=100)
await page.type('#IDToken2', password, delay=100) await page.type('#form_password.form-control', password, delay=100)
logger.debug("Submitting login form") logger.debug("Submitting login form")
await page.click('[type="button"]') await page.click('[type="submit"]')
try: try:
if "failed" in (await page.title()).lower(): if "failed" in (await page.title()).lower():
logger.error("SSO authentication failed: invalid credentials") logger.error("SSO authentication failed: invalid credentials")
@ -179,7 +181,8 @@ async def login_with_gia(
) )
return "" return ""
finally: finally:
await browser.close() if locals().get("browser"):
await browser.close()
async def main(arguments: argparse.Namespace) -> int: async def main(arguments: argparse.Namespace) -> int:
@ -342,13 +345,22 @@ if __name__ == "__main__":
parser.add_argument("-l", "--log-file", help="Tells the script to also write logs on the specified file (relative" parser.add_argument("-l", "--log-file", help="Tells the script to also write logs on the specified file (relative"
" or absolute paths are both accepted). Defaults to no file (i.e. no" " or absolute paths are both accepted). Defaults to no file (i.e. no"
" file logging)", default=None) " file logging)", default=None)
loop = asyncio.get_event_loop() logging.getLogger("asyncio").setLevel(logging.WARNING)
try: logging.getLogger("websockets").setLevel(logging.WARNING)
main_task = asyncio.ensure_future(main(parser.parse_args())) logging.getLogger("httpx").setLevel(logging.WARNING)
for sig in [SIGINT, SIGTERM]: with warnings.catch_warnings():
loop.add_signal_handler(sig, main_task.cancel) warnings.simplefilter('ignore', DeprecationWarning)
sys.exit(loop.run_until_complete(main_task)) loop = asyncio.get_event_loop()
except asyncio.exceptions.CancelledError: try:
print() main_task = asyncio.ensure_future(main(parser.parse_args()))
finally: for sig in [SIGINT, SIGTERM]:
loop.close() loop.add_signal_handler(sig, main_task.cancel)
sys.exit(loop.run_until_complete(main_task))
except (asyncio.exceptions.CancelledError, RuntimeError, sniffio._impl.AsyncLibraryNotFoundError):
print()
finally:
try:
if not loop.is_closed():
loop.close()
except RuntimeError:
pass