def get_collection_keys(cleaned_zot_files):
    """
    Gets the collection (folder) id keys for each entry in the cleaned Zotero files
    """
    col_keys = []
    if 'collections' in cleaned_zot_files['data'].keys():
        if len(cleaned_zot_files['data']['collections']) > 1:
            for i in cleaned_zot_files['data']['collections']:
                col_keys.append(i)
        elif len(cleaned_zot_files['data']['collections']) == 1:
            col_keys.append(cleaned_zot_files['data']['collections'][0])
        else: pass

    return col_keys