aswi2020sebela-gitlab/modules/crawler/RemoveDatasetDatabase.py @ 587b1c57

from Utilities.Database import DatabaseLoader


def remove_dataset_database(dataset_name):
    """
    Removes dataset entries from database
    Args:
        dataset_name: name of dataset that has existing configuration file
    """
    # Creating connection
    mydb = DatabaseLoader.create_database_connection()

    # collection where are specified aviable datasets
    collection_datasets = mydb[DatabaseLoader.MONGODB_DATASET_COLLECTION]

    collection_datasets.delete_one({"name": dataset_name})
    print("Removing record from DATASETS collection")


    # Retrieve list of all collections
    collections = mydb.list_collection_names()

    # Drop of all collections
    for name in collections:
        if name.startswith(dataset_name):
            mydb[name].drop()
            print("Dropping: " + name)

    print("Database Cleaned")

print("Zadejte jméno Datasetu který chcete odstranit z databáze:\n")
remove_dataset_database(input())