eneller/uulm-utils
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: subcommands

Browse Source
This commit is contained in:
eneller
2025-04-13 22:13:25 +02:00
parent 4788d1ae97
commit e3b61b0742
1 changed files with 42 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
main.py
View File

@@ -3,7 +3,6 @@ import asyncclick as click
import questionary import questionary
from playwright.async_api import async_playwright, Playwright from playwright.async_api import async_playwright, Playwright
import re
import asyncio import asyncio
from time import sleep from time import sleep
@@ -17,6 +16,16 @@ async def selection_or_walk(options):
if selection == walkstr: return Selection.TREE_WALK , None if selection == walkstr: return Selection.TREE_WALK , None
return Selection.ITEM_SELECTED, selection return Selection.ITEM_SELECTED, selection
async def run_playwright(headless: bool):
async with async_playwright() as playwright:
browser = await playwright.chromium.launch(headless=headless)
context = await browser.new_context()
page = await context.new_page()
yield page, browser, context
await page.close()
await context.close()
await browser.close()
@click.group() @click.group()
@click.option('--username','-u') @click.option('--username','-u')
@click.option('--password','-p') @click.option('--password','-p')
@@ -24,59 +33,51 @@ async def selection_or_walk(options):
@click.pass_context @click.pass_context
async def cli(ctx, username, password, headful): async def cli(ctx, username, password, headful):
ctx.ensure_object(dict) ctx.ensure_object(dict)
ctx.obj['USERNAME'] = username ctx.obj['USERNAME'] = username or await questionary.text('Enter your kiz username:').ask_async()
ctx.obj['PASSWORD'] = password ctx.obj['PASSWORD'] = password or await questionary.password('Enter your kiz password:').ask_async()
ctx.obj['HEADLESS'] = not headful ctx.obj['HEADLESS'] = not headful
async with async_playwright() as playwright:
browser = await playwright.chromium.launch(headless=headful)
context = await browser.new_context()
page = await context.new_page()
ctx.obj['PAGE'] = page
yield # Subcommands run here
await page.close()
await context.close()
await browser.close()
@cli.command() @cli.command()
@click.pass_context @click.pass_context
async def campusonline(ctx): async def campusonline(ctx):
username = ctx.obj['USERNAME'] username = ctx.obj['USERNAME']
password = ctx.obj['PASSWORD'] password = ctx.obj['PASSWORD']
page = ctx.obj['PAGE'] headless = ctx.obj['HEADLESS']
click.echo("Running campusonline...") async for page, browser, context in run_playwright(headless):
await page.goto("https://campusonline.uni-ulm.de") click.echo("Running campusonline...")
await page.get_by_role("textbox", name="Benutzerkennung").click() await page.goto("https://campusonline.uni-ulm.de")
await page.get_by_role("textbox", name="Benutzerkennung").fill(username) await page.get_by_role("textbox", name="Benutzerkennung").click()
await page.get_by_role("textbox", name="Passwort").fill(password) await page.get_by_role("textbox", name="Benutzerkennung").fill(username)
await page.get_by_role("button", name="Anmelden").click() await page.get_by_role("textbox", name="Passwort").fill(password)
await page.get_by_role("link", name="Studium").click() await page.get_by_role("button", name="Anmelden").click()
await page.get_by_role("link", name="Modulbeschreibungen ansehen").click() await page.get_by_role("link", name="Studium").click()
options = await page.locator('css=li.treelist').all_inner_texts() await page.get_by_role("link", name="Modulbeschreibungen ansehen").click()
selection = await selection_or_walk(options) options = await page.locator('css=li.treelist').all_inner_texts()
print(selection) selection = await selection_or_walk(options)
sleep(2) print(selection)
click.echo("Finished campusonline.") sleep(2)
click.echo("Finished campusonline.")
@cli.command() @cli.command()
@click.pass_context @click.pass_context
async def coronang(ctx): async def coronang(ctx):
username = ctx.obj['USERNAME'] username = ctx.obj['USERNAME']
password = ctx.obj['PASSWORD'] password = ctx.obj['PASSWORD']
page = ctx.obj['PAGE'] headless = ctx.obj['HEADLESS']
click.echo("Running coronang...") async for page, browser, context in run_playwright(headless):
await page.goto("https://campusonline.uni-ulm.de/CoronaNG/user/mycorona.html") click.echo("Running coronang...")
await page.locator("input[name=\"uid\"]").click() await page.goto("https://campusonline.uni-ulm.de/CoronaNG/user/mycorona.html")
await page.locator("input[name=\"uid\"]").fill(username) await page.locator("input[name=\"uid\"]").click()
await page.locator("input[name=\"password\"]").click() await page.locator("input[name=\"uid\"]").fill(username)
await page.locator("input[name=\"password\"]").fill(password) await page.locator("input[name=\"password\"]").click()
await page.get_by_role("button", name="Anmelden").click() await page.locator("input[name=\"password\"]").fill(password)
await page.get_by_role("link", name="Beobachtungen & Teilnahmen").click() await page.get_by_role("button", name="Anmelden").click()
await page.get_by_role("table", name="Ihre Beobachtungen. Sie kö").get_by_role("button").click() await page.get_by_role("link", name="Beobachtungen & Teilnahmen").click()
await page.get_by_role("table", name="Ihre Beobachtungen. Sie kö").get_by_role("combobox").select_option("5") await page.get_by_role("table", name="Ihre Beobachtungen. Sie kö").get_by_role("button").click()
await page.get_by_role("cell", name="An Markierten teilnehmen Ausf").get_by_role("button").click() await page.get_by_role("table", name="Ihre Beobachtungen. Sie kö").get_by_role("combobox").select_option("5")
await page.reload() await page.get_by_role("cell", name="An Markierten teilnehmen Ausf").get_by_role("button").click()
click.echo("Finished coronang.") await page.reload()
click.echo("Finished coronang.")
if __name__ == "__main__": if __name__ == "__main__":
asyncio.run(cli.main()) asyncio.run(cli.main())