#

"""
uninstallSimtkPackage.py: Used for removing python wrappers from Simbios library.
"""
__author__ = "Randall J. Radmer"
__version__ = "1.0"

import os, sys

def removeRecursive(dir):
    for file in os.listdir(dir):
        path = os.path.join(dir, file)
        if os.path.isdir(path):
            removeRecursive(path)
        else:
            os.remove(path)
    os.rmdir(dir)

def removePackage(mod, verbose):
        try:
            pathList = mod.__path__
        except AttributeError:
            return
        if len(pathList) > 1:
           raise Exception("more than one item in simtk.__path__")
        simtkInstallPath = pathList[0]
        if os.path.exists(simtkInstallPath):
            if verbose:
                sys.stdout.write('REMOVING "%s"\n' % simtkInstallPath)
            removeRecursive(simtkInstallPath)

def main(verbose=False):
    try:
        import simtk.pyopenmm as pyopenmm
        removePackage(pyopenmm, verbose)
    except ImportError:
        pass


if __name__=='__main__':
    main(verbose=True)