allris-scraper/src/twomartens/allrisscraper/main.py

import argparse

from twomartens.allrisscraper import internal
from twomartens.allrisscraper import public


def main():
    parser = argparse.ArgumentParser(description="Scrape the ALLRis website")
    subparsers = parser.add_subparsers(help="sub-command help", required=True)
    oparl_parser = subparsers.add_parser("oparl", help="scrapes the public website")
    oparl_parser.add_argument("--include-organizations", action="store_true", dest="include_organizations")
    oparl_parser.add_argument("--include-meetings", action="store_true", dest="include_meetings")
    oparl_parser.set_defaults(function=public.main)
    internal_parser = subparsers.add_parser("internal", help="scrapes the internal website")
    internal_parser.set_defaults(function=internal.main)
    
    args = parser.parse_args()
    args.function(args)


if __name__ == "__main__":
    main()