clear_node.py

#!/usr/bin/env python3

import shutil
import os
import pyorient
import argparse
import re


def pathList(*pathes):
    return {os.path.expanduser(path) for path in pathes}

TARGET_DIRS = pathList(
    "~/.sovrin",
    "~/.plenum"
)

WHITE_LIST = pathList(
    "~/.sovrin/sovrin_config.py",
    "~/.plenum/plenum_config.py",
    "~/.sovrin/.*/role",
    "~/.plenum/.*/role",
    "~/.sovrin/.*log"
)

ORIENTDB_HOST = "localhost"
ORIENTDB_PORT = 2424
ORIENTDB_USER = "root"
ORIENTDB_PASSWORD = "password"


def clean_orientdb():
    client = pyorient.OrientDB(ORIENTDB_HOST, ORIENTDB_PORT)
    client.connect(ORIENTDB_USER, ORIENTDB_PASSWORD)
    names = [n for n in client.db_list().oRecordData['databases'].keys()]
    for nm in names:
        try:
            client.db_drop(nm)
        except:
            continue


def clean_files(full):
    if full:
        for dir in TARGET_DIRS:
            if os.path.exists(dir):
                shutil.rmtree(dir)
        return

    files_to_keep = [re.compile(pattern) for pattern in WHITE_LIST]
    def isOk(path):
         return any(pattern.match(path) for pattern in files_to_keep)

    for dir in TARGET_DIRS:
        for root, dirs, files in os.walk(dir):
            if not isOk(root):
                for file in files:
                    fullName = os.path.join(root, file)
                    if not isOk(fullName):
                        os.remove(fullName)
                if not os.listdir(root):
                    os.rmdir(root)
            else:
                dirs[:] = []

if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description='Removes node data and configuration')
    parser.add_argument('--full',
                        action="store_true",
                        help="remove configs and logs too")
    args = parser.parse_args()
    clean_files(args.full)
    try:
        clean_orientdb()
    except Exception as ex:
        print("Cleaning of data from orientdb failed:", ex)