pyfluminus_cli.py

import argparse
from pyfluminus.authorization import vafs_jwt
from pyfluminus import api
from pyfluminus.structs import File, Module
import os
from typing import Dict, List


parser = argparse.ArgumentParser(description="CLI wrapper to pyfluminus")

# Authentication
parser.add_argument('-username', type=str, help="NUSNET username, e.g. e01234")
parser.add_argument('-password', type=str, help="NUSNET password")
parser.add_argument('--env', action="store_true", help="Get username and password from environment variables")

# Other Flags
parser.add_argument("--download_to", type=str, help="Download destination") # if downloading files
parser.add_argument("--ignore", type=str, help="Comma separated list of modules to ignore (e.g. CS1231,CS4321)")
parser.add_argument("--announcements", action="store_true", help="Display announcements")

def download_files(file: File, auth: Dict, download_path: str, verbose=False):
    if not file.directory:
        full_file_path = os.path.join(download_path, file.name)
        if os.path.exists(full_file_path):
            return
        print("- {}".format(full_file_path))
        file.download(auth, download_path, verbose)
        return
    download_path = os.path.join(download_path, file.name)
    if file.children is None:
        file.load_children(auth) 
        if file.children is None: 
            print("Error loading children for file: {}".format(file.name))
            return
    for child in file.children:
        download_files(child, auth, download_path, verbose)
        

if __name__ == "__main__":
    args = parser.parse_args()
    if args.env:
        username = os.environ["LUMINUS_USERNAME"]
        password = os.environ["LUMINUS_PASSWORD"]
    else:
        username = args.username
        password = args.password
    auth = vafs_jwt("nusstu\\" + username, password)

    if 'jwt' not in auth:
        print("Failed to authenticate:", auth['error'])
        exit()
    

    name_res = api.name(auth)
    if not name_res.ok:
        print("Error getting name: ", name_res.error_msg)
    print("Hello {}".format(name_res.data))

    modules_res = api.modules(auth)
    if not modules_res.ok:
        print("Error: ", modules_res.error_msg)
    print("Your are taking the following mods")
    modules = modules_res.data
    for module in modules:
        if module is None:
            print("Error parsing module data")
            continue
        print("- {} {}".format(module.code, module.name))

    ignored_modules: List[str] = [] 
    if args.ignore:
        ignored_modules = args.ignore.split(",")

    if args.announcements:
        print("\n# Announcements")
        actually_ignored_modules = []
        for module in modules:
            if module is None:
                continue
            if module.code in ignored_modules:
                actually_ignored_modules.append(module)
                continue
            print("\n## {}: {}".format(module.code, module.name))
            announcements = module.announcements(auth)
            if announcements is None:
                print("Error retrieving annoucements")
                continue
            for ann in announcements:
                print("### {}".format(ann['title']))
                print("Posted on: {}\n".format(ann['datetime']))
                print(ann["description"])
        if actually_ignored_modules:
            print("Ignored the following module(s)")
            for module in actually_ignored_modules:
                print("- {} {}".format(module.code, module.name))

    if args.download_to:
        print("\n\nDownloading Files to {}".format(args.download_to))
        actually_ignored_modules = []
        for module in modules:
            if module is None:
                continue
            if module.code in ignored_modules:
                actually_ignored_modules.append(module)
                continue
            print("{} {}".format(module.code, module.name))
            module_file = File.from_module(auth, module)
            # TODO set verbose=True for now
            download_files(module_file, auth, args.download_to, True)
        if actually_ignored_modules:
            print("Ignored the following module(s)")
            for module in actually_ignored_modules:
                print("- {} {}".format(module.code, module.name))
        print("\nDONE")