purge-local-db.py

import os
import shutil
from pymongo import MongoClient


def get_db_handle(db_name, host):
    client = MongoClient(host)
    db_handle = client[db_name]
    return db_handle, client


def get_collection_handle(db_handle,collection_name):
    return db_handle[collection_name]


# Set up database connection
for db_name in ['caper', 'caper-dev']:
    print("Purging " + db_name)
    db_handle, mongo_client = get_db_handle(db_name, 'mongodb://localhost:27017')
    collection_handle = get_collection_handle(db_handle,'projects')
    fs_files = get_collection_handle(db_handle,'fs.files')
    fs_chunks = get_collection_handle(db_handle,'fs.chunks')

# Delete all projects
collection_handle.drop()

# Delete all files
fs_files.drop()
fs_chunks.drop()

# clear the contents of tmp/
folder = 'tmp/'
for filename in os.listdir(folder):
    file_path = os.path.join(folder, filename)
    try:
        if os.path.isfile(file_path) or os.path.islink(file_path):
            os.unlink(file_path)
        elif os.path.isdir(file_path):
            shutil.rmtree(file_path)
    except Exception as e:
        print('Failed to delete %s. Reason: %s' % (file_path, e))