better selection

This commit is contained in:
eneller
2025-04-13 18:48:54 +02:00
parent 6b744f52b4
commit 4788d1ae97

101
main.py
View File

@@ -1,3 +1,4 @@
from enum import Enum
import asyncclick as click import asyncclick as click
import questionary import questionary
from playwright.async_api import async_playwright, Playwright from playwright.async_api import async_playwright, Playwright
@@ -6,62 +7,76 @@ import re
import asyncio import asyncio
from time import sleep from time import sleep
Selection = Enum('Selection', ['TREE_WALK', 'TREE_LEAF', 'ITEM_SELECTED'])
async def selection_or_walk(options): async def selection_or_walk(options):
options = list(map(str.strip, options))
walkstr = 'Walk the Tree from here' walkstr = 'Walk the Tree from here'
options.append(walkstr) options = [questionary.Choice(title=walkstr, checked=True)] + [questionary.Choice(opt.strip()) for opt in options]
selection = await questionary.select(choices=options, message='Select one of the following options').ask_async() selection = await questionary.select(choices=options, message='Select one of the following options').ask_async()
if selection == walkstr: return None if selection == walkstr: return Selection.TREE_WALK , None
return selection return Selection.ITEM_SELECTED, selection
@click.group()
@click.option('--username','-u')
@click.option('--password','-p')
@click.option('--headful', is_flag=True)
@click.pass_context
async def cli(ctx, username, password, headful):
ctx.ensure_object(dict)
ctx.obj['USERNAME'] = username
ctx.obj['PASSWORD'] = password
ctx.obj['HEADLESS'] = not headful
async def run( async with async_playwright() as playwright:
playwright: Playwright, browser = await playwright.chromium.launch(headless=headful)
username, context = await browser.new_context()
password, page = await context.new_page()
headless:bool, ctx.obj['PAGE'] = page
) -> None: yield # Subcommands run here
if not username: username = await questionary.text('Enter your kiz username:').ask_async() await page.close()
if not password: password = await questionary.password('Enter your kiz password:').ask_async() await context.close()
browser = await playwright.chromium.launch(headless=headless) await browser.close()
context = await browser.new_context()
page = await context.new_page() @cli.command()
@click.pass_context
async def campusonline(ctx):
username = ctx.obj['USERNAME']
password = ctx.obj['PASSWORD']
page = ctx.obj['PAGE']
click.echo("Running campusonline...")
await page.goto("https://campusonline.uni-ulm.de") await page.goto("https://campusonline.uni-ulm.de")
await page.get_by_role("textbox", name="Benutzerkennung").click() await page.get_by_role("textbox", name="Benutzerkennung").click()
await page.get_by_role("textbox", name="Benutzerkennung").fill("dkp11") await page.get_by_role("textbox", name="Benutzerkennung").fill(username)
await page.get_by_role("textbox", name="Passwort").fill("zy7Av5rickeyyy") await page.get_by_role("textbox", name="Passwort").fill(password)
await page.get_by_role("button", name="Anmelden").click() await page.get_by_role("button", name="Anmelden").click()
await page.get_by_role("link", name="Studium").click() await page.get_by_role("link", name="Studium").click()
await page.get_by_role("link", name="Modulbeschreibungen ansehen").click() await page.get_by_role("link", name="Modulbeschreibungen ansehen").click()
options = await page.locator('css=li.treelist').all_inner_texts() options = await page.locator('css=li.treelist').all_inner_texts()
selection = await selection_or_walk(options) selection = await selection_or_walk(options)
print(selection) print(selection)
sleep(20) sleep(2)
exit(0) click.echo("Finished campusonline.")
await page.get_by_text("Modulbeschreibungen Bitte wä").click()
await page.get_by_role("link", name="Module für Abschluss: Master of Science").click()
await page.get_by_text("Modulbeschreibungen Sie").click()
await page.get_by_role("link", name="Studiengang: Informatik").click()
await page.get_by_role("link", name="Studium gemäß Prüfungsordnung: 2022").click()
await page.get_by_role("link", name="Wahlpflichtbereich").click()
await page.get_by_role("link", name="Kernbereich Informatik").click()
await page.get_by_role("link", name="Kernbereich Praktische").click()
await page.get_by_role("link", name="Data Mining").click()
await page.get_by_role("link", name="Beschreibung anzeigen").click()
sleep(20)
await page.close()
# --------------------- @cli.command()
await context.close() @click.pass_context
await browser.close() async def coronang(ctx):
username = ctx.obj['USERNAME']
@click.command() password = ctx.obj['PASSWORD']
@click.option('--username','-u') page = ctx.obj['PAGE']
@click.option('--password','-p') click.echo("Running coronang...")
@click.option('--headful', is_flag=True) await page.goto("https://campusonline.uni-ulm.de/CoronaNG/user/mycorona.html")
async def main(username, password, headful): await page.locator("input[name=\"uid\"]").click()
async with async_playwright() as playwright: await page.locator("input[name=\"uid\"]").fill(username)
await run(playwright, username, password, headless=not headful) await page.locator("input[name=\"password\"]").click()
await page.locator("input[name=\"password\"]").fill(password)
await page.get_by_role("button", name="Anmelden").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("button").click()
await page.get_by_role("table", name="Ihre Beobachtungen. Sie kö").get_by_role("combobox").select_option("5")
await page.get_by_role("cell", name="An Markierten teilnehmen Ausf").get_by_role("button").click()
await page.reload()
click.echo("Finished coronang.")
if __name__ == "__main__": if __name__ == "__main__":
asyncio.run(main()) asyncio.run(cli.main())