def upload_model()

in upload_models.py [0:0]

• 31 lines of code
• 9 McCabe index (conditional complexity)

def upload_model(model, api_url, models_path, api_key, continue_on_error=False):
    headers = dict(Authorization=f'Bearer {api_key}')
    model_path = os.path.join(models_path, f"{model['filename']}.csv")
    if len(model['rds']) > 0:
        rds_path = os.path.join(models_path, os.path.basename(model['rds']))
    else:
        rds_path = None
    model_data = {
        "id": model['filename'],
        "name": model['name'],
        "query_str": model['queryJSON'],
        "model_type": model['type'],
        "created": model['created'],

    }
    try:
        files = {
            'model': (os.path.basename(model_path), open(model_path, 'rb'), 'text/plain')
        }
        if rds_path and os.path.exists(rds_path):
            files['rds'] = (os.path.basename(rds_path), open(rds_path, 'rb'), 'application/octet-stream')

        r = requests.post(api_url, data=model_data, headers=headers, files=files)
        if r.status_code != 201 and not continue_on_error:
            logger.error(f"Failed to upload: {model['filename']}")
            raise Exception(f"upload failed of {model['filename']}: {r.status_code}: {r.content}")
        elif r.status_code != 201:
            logger.warning(f"Failed to upload: {model['filename']}: {r.status_code}: {r.content}")
    except FileNotFoundError as e:
        if not continue_on_error:
            raise e
        else:
            logger.warning(f"Could not find file from modelDB.tsv for {model['name']}")