PC/setup_nt/uninstall.py

"""Uninstaller for Windows NT 3.5 and Windows 95.

Actions:

1. Remove our entries from the Registry:
   - Software\Python\PythonCore\<winver>
   - Software\Microsoft\Windows\CurrentVersion\Uninstall\Python<winver>
   (Should we also remove the entry for .py and Python.Script?)

2. Remove the installation tree -- this is assumed to be the directory
   whose path is both os.path.dirname(sys.argv[0]) and sys.path[0]

"""

import sys
import nt
import os
import win32api
import win32con

def rmkey(parent, key, level=0):
    sep = "    "*level
    try:
        handle = win32api.RegOpenKey(parent, key)
    except win32api.error, msg:
        print sep + "No key", `key`
        return
    print sep + "Removing key", key
    while 1:
        try:
            subkey = win32api.RegEnumKey(handle, 0)
        except win32api.error, msg:
            break
        rmkey(handle, subkey, level+1)
    win32api.RegCloseKey(handle)
    win32api.RegDeleteKey(parent, key)
    print sep + "Done with", key

roothandle = win32con.HKEY_LOCAL_MACHINE
pythonkey = "Software\\Python\\PythonCore\\" + sys.winver
rmkey(roothandle, pythonkey)
uninstallkey = \
 "Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\Python"+sys.winver
rmkey(roothandle, uninstallkey)

def rmtree(dir, level=0):
    sep = "    "*level
    print sep+"rmtree", dir
    for name in os.listdir(dir):
        if level == 0 and \
           os.path.normcase(name) == os.path.normcase("uninstall.bat"):
            continue
        fn = os.path.join(dir, name)
        if os.path.isdir(fn):
            rmtree(fn, level+1)
        else:
            try:
                os.remove(fn)
            except os.error, msg:
                print sep+"  can't remove", `fn`, msg
            else:
                print sep+"  removed", `fn`
    try:
        os.rmdir(dir)
    except os.error, msg:
        print sep+"can't remove directory", `dir`, msg
    else:
        print sep+"removed directory", `dir`    

pwd = os.getcwd()
scriptdir = os.path.normpath(os.path.join(pwd, os.path.dirname(sys.argv[0])))
pathdir = os.path.normpath(os.path.join(pwd, sys.path[0]))
if scriptdir == pathdir:
    rmtree(pathdir)
else:
    print "inconsistend script directory, not removing any files."
    print "script directory =", `scriptdir`
    print "path directory =", `pathdir`