def delete_old_models()

in api_service/migrations/versions/2c92fa01c7ef_.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

def delete_old_models():
    # cleanup defunc models?
    session = Session(bind=connection)
    from flask import current_app
    base_path = current_app.config.get('MODEL_STORE', '/model_store')
    model_ids = list(session.query(PathogenModel.id).distinct())
    model_ids = set([r for r, in model_ids])
    for file in glob(f"{base_path}/*.csv"):
        if os.path.basename(file)[:-4] not in model_ids:
            print(f'Removing old model {file}')
            os.remove(file)
    session.commit()