main.py

import typer
import time
from artemis.course_page import ArtemisCourse
from artemis import core as Artemis
from utils.browser import WebDriverSingleton
from utils.config import setup
from utils.print import printer
from utils.general_utils import is_tum_ID, is_valid_artemis_course_link, extract_base_url

app = typer.Typer(pretty_exceptions_show_locals=False, pretty_exceptions_enable=False)



@app.command()
def hello(name: str):
    print("Hello World!")

@app.command()
def dl_course(
    download_path: str = typer.Option('', help="Specify Download Location"), 
    username: str = typer.Option(..., prompt=True), 
    password: str = typer.Option(..., prompt=True), 
    link: str = typer.Option(..., prompt=True, help="Course-Link from Artemis")
):
    if is_tum_ID(username) is False:
        printer('Username is not valid TUM-ID. Please try again!')
        return
    if len(password) < 8:
        printer('Password is below 8 characters. Please try again!')
        return
    if is_valid_artemis_course_link(link) is not True:
        printer('Link is not a valid Artemis-Course. Please try again!')
        return
    setup(username=username, password=password, download_dir=download_path)

    printer('Opening Browser..')
    sdriver = WebDriverSingleton.get_instance()
    printer('Browser successfully opened')

    printer('Navigating to Artemis Login Screen')
    sdriver.get(extract_base_url(link))
    Artemis.enable_dark_mode()

    printer('Trying to login with credentials')
    if Artemis.login() == False: 
        printer('Login failed. Please check your credentials and try again!')
        return
    printer('Successfully logged in')

    time.sleep(2)
    course = ArtemisCourse(course_link=link)
    course.open()
    course.download_all_exercises()


if __name__ == "__main__":
    app()