package-tracking/package_tracking/__main__.py

import logging
import yaml

from . import parcelsapp
from . import secrets
from . import http

logger = logging.getLogger(__name__)

def main_cli(parcelsapp_client: parcelsapp.ParcelsAppClient):
    tracking_ids = [
            '00157128965207138207',
            '00057151273127784840',
    ]
    shipment_statuses = parcelsapp_client.get_tracking_status(tracking_ids)
    print(yaml.dump(shipment_statuses['shipments']))


def main():
    logging.basicConfig()
    logger.setLevel('INFO')

    parcelsapp_client: parcelsapp.ParcelsAppClient =  parcelsapp.ParcelsAppClient(secrets.PARCELS_API_KEY)

    if True:
        main_cli(parcelsapp_client)
    else:
        http.initialize_server()


if __name__ == '__main__':
    main()