import requests
import os
import yaml

CONF_FOLDER = "conf"

# creates the "conf" dir if it doesn't exist
if not os.path.exists(CONF_FOLDER):
    os.makedirs(CONF_FOLDER)
    

exists = os.path.isfile(CONF_FOLDER + "/configs.yml")
if exists:
    print(f'opening {CONF_FOLDER + "/configs.yml"} ')
    with open(CONF_FOLDER + "/configs.yml", "r") as f:
        config_data = yaml.load(f, Loader=yaml.FullLoader)
    DATASET_SLUG = config_data["dataset_slug"]
    API_KEY = config_data["api_key"]
    API_BASE = config_data["api_base"]
    API_ENV = config_data["api_env"]
    API_FUNCTION = config_data["api_function"]
    print(f'establishing session ')
    headers = {"X-API-KEY": API_KEY, "Content-Type": "application/json"}

    session = requests.Session()
    session.headers.update({"X-API-KEY": API_KEY})

    file = os.listdir('./upload')

    if len(file) > 1:
        print ("ERROR: Multiple files within upload directory. A single file is uploaded each time. Please remove additional files from upload directory")
    elif len(file) == 0:
        print ("ERROR: No files to be uploaded. Check upload directory.")
    else:
        print (f"opening {'./upload/'+file[0]}")
        filehandle = open('./upload/'+file[0], 'rb')
        url=API_ENV+API_BASE+DATASET_SLUG+API_FUNCTION
        response = session.post(url, data={},files = {'file':filehandle})
        print ("SUCCESS: uploaded succeeded")
else:
    print("ERROR: Please setup your configs.yml file within conf folder.")