nixfiles/musicutil/__main__.py

#!/usr/bin/env python3

import argparse

from musicutil.commands.stats_command import StatsCommand, add_stats_command, get_stats_args

from .commands.genhtml_command import GenHTMLCommand, add_genhtml_command, get_genhtml_args
from .commands.process_command import ProcessCommand, add_process_command, get_process_args
from .commands.copy_command import CopyCommand, add_copy_command, get_copy_args
from .commands.transcode_command import TranscodeCommand, add_transcode_command, get_transcode_args

def main():
    parser = argparse.ArgumentParser(
        description="chaos's musicutil")
    subparsers = parser.add_subparsers(dest="subparser_name")

    add_copy_command(subparsers)
    add_process_command(subparsers)
    add_transcode_command(subparsers)
    add_genhtml_command(subparsers)
    add_stats_command(subparsers)

    args = parser.parse_args()

    if args.subparser_name == "process":
        ProcessCommand(get_process_args(args)).run()
    elif args.subparser_name == "copy":
        CopyCommand(get_copy_args(args)).run()
    elif args.subparser_name == "transcode":
        TranscodeCommand(get_transcode_args(args)).run()
    elif args.subparser_name == "genhtml":
        GenHTMLCommand(get_genhtml_args(args)).run()
    elif args.subparser_name == "stats":
        StatsCommand(get_stats_args(args)).run()
    else:
        parser.print_help()

if __name__ == "__main__":
    main()