rewe-vereinscheine-fetcher/main.py

import argparse
from concurrent.futures import ThreadPoolExecutor

from csv_exporter import export_vereine
from verein import Verein
from verein_fetcher import get_verein_ids_near_location, get_verein_for_id


def main():
    parser = argparse.ArgumentParser(
        prog='Rewe Vereinsscheine',
        description='Get the amount of Vereinsscheine for Vereine near a location')
    parser.add_argument('latitude')
    parser.add_argument('longitude')
    parser.add_argument('-f', '--export-file', default='vereine.csv', required=False)
    args = parser.parse_args()

    print("Fetching Vereine")
    verein_ids = get_verein_ids_near_location(args.latitude, args.longitude)

    print("Getting Verein information")
    vereine: [Verein]
    with ThreadPoolExecutor() as executor:
        vereine = list(executor.map(get_verein_for_id, verein_ids))

    print("Exporting Verein information")
    export_vereine(vereine, args.export_file)


if __name__ == "__main__":
    main()