import argparse
import json

import requests
import yaml


def main():
    parser = argparse.ArgumentParser(description='Upload a directory into a data library')
    parser.add_argument("-u", "--url", dest="url", required=True, help="Galaxy URL")
    parser.add_argument("-a", "--api", dest="api_key", required=True, help="API Key")
    parser.add_argument('target', metavar='FILE', type=str,
                        help='file describing data library to fetch')
    args = parser.parse_args()
    with open(args.target) as f:
        target = yaml.safe_load(f)

    histories_url = args.url + "/api/histories"
    new_history_response = requests.post(histories_url, data={'key': args.api_key})

    fetch_url = args.url + '/api/tools/fetch'
    payload = {
        'key': args.api_key,
        'targets': json.dumps([target]),
        'history_id': new_history_response.json()["id"]
    }

    response = requests.post(fetch_url, data=payload)
    print(response.content)


if __name__ == '__main__':
    main()