play.py

from utilities.watcher import watch, names as server_names
import asyncio
import argparse

commands = '''

commands:
    update      Autoupdate the library by fetching the latest episodes form to_update.json
    download    Run the download menu to download a range of episodes
    latest      Prints the latest updates on the website 
    check       Checks for any available updates for download
    
'''

parser = argparse.ArgumentParser(description="Play anime directly" + commands, formatter_class=argparse.RawDescriptionHelpFormatter)

parser.add_argument(
    "--list",
    "-L",
    action="store_true",
    help="Optional switch to list available player servers (not ext) with their index",
)

# Optional positional argument
parser.add_argument("name", type=str, nargs="*", help="Name of the anime to search")

# Optional argument
parser.add_argument("--ep", "-e", type=int, help="Episode number. Default is the latest episode.")

parser.add_argument("--url", "-u", type=str, help="Optional kickass anime url")

parser.add_argument(
    "--opt",
    "-o",
    type=int,
    help="Optional way to select search result number.",
)

parser.add_argument(
    "--ext",
    action="store_true",
    help="Optional switch to play only from ext servers (its faster if it works). Needs no arguments",
)

parser.add_argument(
    "--stop",
    "-s",
    action="store_true",
    help="Optional switch to stop the script after searching without playing anything. Will also display the url of the anime.",
)

parser.add_argument(
    "--encode",
    action="store_true",
    help="Optional switch to print ffmpeg command to encode the stream",
)

def parse_server_name(arg):
    if arg:
        try:
            index = int(arg)
            if index < len(server_names):
                return server_names[index]
            else:
                print('Priority Index out of range')
                return ''
        except:
            return arg.upper()
    else:
        return ''

parser.add_argument(
    "--custom_server",
    "-c",
    help="Name of the player/server you want or the INDEX number in the watch_config priority dictionary. Overrides the current priority.",
    type = parse_server_name,
    nargs='?',
    # default=''
)

args = parser.parse_args()
episode = args.ep
link = args.url
query = " ".join(args.name) if args.name else None
opt = args.opt
ext_only = args.ext
custom_server = args.custom_server
to_list = args.list
stop = args.stop
encode = args.encode

if to_list:
    print('Possible player servers are:')
    for j, i in enumerate(server_names):
        print(j, i)
    exit()


if __name__ == '__main__':
    try:
        asyncio.run(
            watch(episode, link=link, query=query, option_number=opt, ext_only=ext_only, custom_server=custom_server, stop=stop, encode=encode)
        )
    except RuntimeError:
        print('Exit')