added uninstall

diff --git a/setup.py b/setup.py
index 4b286cb..90f852a 100755 (executable)
--- a/setup.py
+++ b/setup.py
@@ -12,8 +12,61 @@ scripts = ['geni/gimport.py', 'geni/plc.py', 'cmdline/sfi.py']
 package_dirs = ['geni', 'geni/util', 'geni/methods']
 data_files = [('/etc/geni/', ['geni/aggregates.xml', 'geni/registries.xml', 'geni/util/geni_config', 'cmdline/configSfi.sh'])]
 
+def list_dir(directory):
+    # list the files in this directory
+    try:
+        files = os.listdir(directory)
+        add_directory_path = lambda path: directory + os.sep + path
+        # add absolute path to files
+        files = map(add_directory_path, files)
+        # separate files from directories
+        files_only = lambda path: os.path.isfile(path)
+        directories_only = lambda path: os.path.isdir(path)
+        directories = filter(directories_only, files)
+        files = filter(files_only, files)
+        for d in directories:
+            try:
+                (fs, ds) = list_dir(d)
+                files.extend(fs)
+            except: 
+                print "Error listing", d
+        return (files, directories)
+    except:
+        return ([], [])
+        
+        
+if sys.argv[1] in ['uninstall', 'remove', 'delete']:
+    python_path = sys.path
+    site_packages_only = lambda path: path.endswith('site-packages') 
+    site_packages_path = filter(site_packages_only, python_path)
+    add_geni_path = lambda path: path + os.sep + 'geni'
+    site_packages_path = map(add_geni_path, site_packages_path) 
+    remove_dirs = ['/etc/geni/'] + site_packages_path
+    remove_files = ['/usr/bin/gimport.py', '/usr/bin/plc.py', '/usr/bin/sfi.py']
+    # update list of files/directories to remove
+    for directory in remove_dirs:
+        files, dirs = list_dir(directory)
+        remove_files.extend(list(set(files)))
+        remove_dirs.extend(dirs)
+        remove_dirs.reverse()
 
-setup(name='geniwrapper', 
+    # remove files   
+    for filepath in remove_files:
+        print "removing", filepath, " ... ",
+        try: 
+            os.remove(filepath)
+            print "success"
+        except: print "failed"
+    # remove directories 
+    for directory in remove_dirs: 
+        print "removing", directory, " ... ",
+        try: 
+            os.rmdir(directory)
+            print "success"
+        except: print "failed"
+ 
+else:
+    setup(name='geniwrapper', 
       version = version,
       packages = package_dirs, 
       data_files = data_files,