feat: subcommands

main.py

@@ -3,7 +3,6 @@ import asyncclick as click
 import questionary
 from playwright.async_api import async_playwright, Playwright
 
-import re
 import asyncio
 from time import sleep
 
@@ -17,6 +16,16 @@ async def selection_or_walk(options):
     if selection == walkstr: return Selection.TREE_WALK , None
     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.option('--username','-u')
 @click.option('--password','-p')
@@ -24,26 +33,17 @@ async def selection_or_walk(options):
 @click.pass_context
 async def cli(ctx, username, password, headful):
     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
 
-    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()
 @click.pass_context
 async def campusonline(ctx):
     username = ctx.obj['USERNAME']
     password = ctx.obj['PASSWORD']
-    page = ctx.obj['PAGE']
+    headless = ctx.obj['HEADLESS']
+    async for page, browser, context in run_playwright(headless):
         click.echo("Running campusonline...")
         await page.goto("https://campusonline.uni-ulm.de")
         await page.get_by_role("textbox", name="Benutzerkennung").click()
@@ -63,7 +63,8 @@ async def campusonline(ctx):
 async def coronang(ctx):
     username = ctx.obj['USERNAME']
     password = ctx.obj['PASSWORD']
-    page = ctx.obj['PAGE']
+    headless = ctx.obj['HEADLESS']
+    async for page, browser, context in run_playwright(headless):
         click.echo("Running coronang...")
         await page.goto("https://campusonline.uni-ulm.de/CoronaNG/user/mycorona.html")
         await page.locator("input[name=\"uid\"]").click()