import requests
import time
import logging
import yaml
from . import secrets

logger = logging.getLogger(__name__)

URL_TRACKING = 'https://parcelsapp.com/api/v3/shipments/tracking'

target_country = 'Denmark'

TRACKING_STATUS_CHECKING_INTERVAL = 1

def request_json(method: str, url: str, **kwargs) -> dict:
    response = requests.request(method=method,url=URL_TRACKING, json={'apiKey': secrets.PARCELS_API_KEY, **kwargs})
    print(response)
    response.raise_for_status()
    json_data = response.json()
    if 'error' in json_data:
        msg = 'Error from endpoint: {}'.format(json_data['error'])
        raise RuntimeError(msg)
    return json_data

def check_tracking_status(uuid: str) -> dict:
    """ Function to check tracking status with UUID"""
    json_data = request_json('GET', URL_TRACKING, uuid=uuid)
    if json_data['done']:
        logger.info('Tracking complete')
        return json_data
    else:
        logger.info('Tracking in progress...')
        time.sleep(TRACKING_STATUS_CHECKING_INTERVAL)
        return check_tracking_status(uuid)

def get_tracking_status(tracking_ids: list[str]):
    shipments = [{'trackingId': id, 'language': 'en', 'country': target_country} for id in tracking_ids]

    # Initiate tracking request

    json_data = request_json('POST', URL_TRACKING, shipments=shipments)
    if json_data.get('done'):
        return json_data
    return check_tracking_status(json_data['uuid'])

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

    tracking_ids = [
            '00157128965207138207',
            '00057151273127784840',
    ]
    shipment_statuses = get_tracking_status(tracking_ids)
    print(yaml.dump(shipment_statuses['shipments']))

if __name__ == '__main__':
    main()