def get_models()

in api_service/scripts/download_models.py [0:0]

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

def get_models(url, model_store, model_type=None, only_latest=True):
    response = requests.get(url)
    if response.status_code == 200:
        models = response.json()
        if only_latest:
            n_models = {}
            for model in models:
                if model['query_str'] not in n_models:
                    n_models[model['query_str']] = model
            models = n_models.values()
        for model in models:
            query_obj = json.loads(model['query_str'])
            if model_type is None or query_obj['model_type'] == model_type:
                if os.path.exists(os.path.join(model_store, f"{model['id']}.csv")):
                    metafn = os.path.join(model_store, f"{model['id']}.meta")
                    json.dump(model, open(metafn, 'w'))