csh/main.py

from sys import argv
from threading import Thread

from cheat import Cheat, sleep

if __name__ == "__main__":
    c_id = []
    args = {}

    if (len(argv) > 1):
        # User wanna use offline offsets
        if "--offline" in argv:
            args["offline"] = True

        # User gave a list of cheats
        # Will bypass the interactive selection
        c_id = [int(j) for j in [i for i in argv if i.startswith(
            "--list=")][0][7:].split(",")]

    # Load cheat class
    c = Cheat(**args)

    # Interactive selection
    if (c_id == []):
        # Cheat list
        print("You can run multiples cheat at once, separate your choices with a space.")
        print("Available cheats:")
        for idx, cheat in enumerate(c.cheats_list):
            print(f"#{idx + 1} - {cheat}")

        while c_id == []:
            try:
                response = [int(i) for i in input("Enter ID: ").split(" ")]
                for i in response:
                    if (i > len(c.cheats_list) or i <= 0):
                        raise IndexError
                    c_id.append(i - 1)
            except KeyboardInterrupt:
                print("Bye")
                exit(1)
            except:
                c_id = []
                print("Invalid ID.")

    # Instanciate and run threads
    for fn in [c.cheats_list[i] for i in c_id]:
        print(f"Running {fn}...")
        t = Thread(target=getattr(c, fn))
        t.daemon = True
        t.start()

    # Don't close the main thread as cheats are daemons
    try:
        while True:
            sleep(1000000)
    except KeyboardInterrupt:
        for fn in [c.cheats_list[i] for i in c_id]:
            try:
                unloader = getattr(c, f"{fn}_unload")
            except:
                pass
            else:
                print(f"Unload {fn}...")
                unloader()