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

refactor: cli

Browse Source
This commit is contained in:
eneller
2025-04-16 18:22:26 +02:00
parent 3e8060e075
commit a0eb5928d6
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/uulm_utils/main.py
View File

@@ -1,3 +1,4 @@
from typing import cast
import asyncclick as click import asyncclick as click
import questionary import questionary
from playwright.async_api import Page, async_playwright from playwright.async_api import Page, async_playwright
@@ -40,6 +41,10 @@ def browser_options(f):
f = click.option('--headless', '-h', is_flag=True, help='Dont show the browser window')(f) f = click.option('--headless', '-h', is_flag=True, help='Dont show the browser window')(f)
return f return f
def fcfs_options(f):
f = click.argument('target_times', nargs=-1, type=click.DateTime( ['%H:%M','%H:%M:%S']), required=True)(f)
return f
@click.group() @click.group()
@click.option('--debug', '-d', is_flag=True, help='Set the log level to DEBUG') @click.option('--debug', '-d', is_flag=True, help='Set the log level to DEBUG')
@click.option('--log-level', '-l', type=click.Choice(logging.getLevelNamesMapping().keys()),default = 'INFO') @click.option('--log-level', '-l', type=click.Choice(logging.getLevelNamesMapping().keys()),default = 'INFO')
@@ -80,7 +85,7 @@ async def campusonline(username, password, headless):
return return
@cli.command() @cli.command()
@click.argument('target_times', nargs=-1, type=click.DateTime( ['%H:%M:%S']), required=True) @fcfs_options
@browser_options @browser_options
@click.option('--offset', '-o', type=int, default=10, help='How many seconds before and after the target time to send') @click.option('--offset', '-o', type=int, default=10, help='How many seconds before and after the target time to send')
async def coronang(target_times, username, password, headless, offset): async def coronang(target_times, username, password, headless, offset):
@@ -107,7 +112,7 @@ async def coronang(target_times, username, password, headless, offset):
while True: while True:
server_str = await page.locator("css=#mblock_innen").inner_text() server_str = await page.locator("css=#mblock_innen").inner_text()
server_time = datetime.strptime(server_str.split().pop(), "%H:%M:%S") server_time = datetime.strptime(server_str.split().pop(), "%H:%M:%S")
dtime = target_time -server_time dtime = cast(timedelta, target_time -server_time )
dtime_before = dtime - before_seconds dtime_before = dtime - before_seconds
logger.debug('Server Time: %s, delta: %s', server_time.time(), dtime_before) logger.debug('Server Time: %s, delta: %s', server_time.time(), dtime_before)
# window started? # window started?
@@ -155,7 +160,7 @@ async def coronang(target_times, username, password, headless, offset):
return return
@cli.command() @cli.command()
@click.argument('target_times', nargs=-1, type=click.DateTime( ["%H:%M:%S"]), required=True) @fcfs_options
@click.option('--target_course', '-t', multiple=True, required=True, help='Unique course name to register for. Can be passed multiple times') @click.option('--target_course', '-t', multiple=True, required=True, help='Unique course name to register for. Can be passed multiple times')
@browser_options @browser_options
@click.option('--offset', '-o', type=int, default=10, help='How many seconds before and after the target time to send') @click.option('--offset', '-o', type=int, default=10, help='How many seconds before and after the target time to send')