main.py

"""Implementation of anonymizer service."""
import argparse
import nlp_ws
from src.worker_old import Worker


def get_args():
    """Gets command line arguments."""
    parser = argparse.ArgumentParser(description="anonymizer")

    subparsers = parser.add_subparsers(dest="mode")
    subparsers.required = True

    subparsers.add_parser("service", help="Run as a service")

    return parser.parse_args()


def main():
    """Runs the program."""
    args = get_args()

    generators = {
        "service": lambda: nlp_ws.NLPService.main(Worker),
    }

    gen_fn = generators.get(args.mode, lambda: None)
    gen_fn()


if __name__ == "__main__":
    main()