Index: register.py
===================================================================
RCS file: /cvsroot/pywin32/pywin32/com/win32com/server/register.py,v
retrieving revision 1.20
diff -u -r1.20 register.py
--- register.py	8 Mar 2007 23:39:59 -0000	1.20
+++ register.py	18 Jan 2008 04:07:24 -0000
@@ -479,11 +479,43 @@
   flags['debug'] = flags.get('debug',0) or '--debug' in sys.argv
   if unregisterInfo:
     return UnregisterInfoClasses(*classes, **flags)
-  if unregister:
-    UnregisterClasses(*classes, **flags)
-  else:
-    RegisterClasses(*classes, **flags)
+  try:
+    if unregister:
+      UnregisterClasses(*classes, **flags)
+    else:
+      RegisterClasses(*classes, **flags)
+  except win32api.error, exc:
+    # If we are on Vista and have "access denied", retry using
+    # ShellExecuteEx with 'runas' verb to force elevation.
+    if exc[0] != winerror.ERROR_ACCESS_DENIED \
+       or sys.getwindowsversion()[0] < 6:
+      raise
+    from win32com.shell.shell import ShellExecuteEx
+    from win32com.shell import shellcon
+    import win32process, win32event
 
+    if not flags['quiet']:
+      print "Requesting elevation and retrying..."
+    new_params = " ".join(['"' + a + '"' for a in sys.argv])
+    rc = ShellExecuteEx(fMask=shellcon.SEE_MASK_NOCLOSEPROCESS,
+                        lpVerb="runas",
+                        lpFile=win32api.GetShortPathName(sys.executable),
+                        lpParameters=new_params,
+                        lpDirectory=os.getcwd(),
+                        nShow=win32con.SW_SHOW)
+    # Output is lost to the new console which opens, so the
+    # best we can do is get the exit code of the process.
+    hproc = rc['hProcess']
+    win32event.WaitForSingleObject(hproc, win32event.INFINITE)
+    exit_code = win32process.GetExitCodeProcess(hproc)
+    if exit_code:
+      # Even if quiet you get to see this error.
+      print "Error: registration failed (exit code %s)." % exit_code
+      print "Please re-execute this command from an elevated command-prompt"
+      print "to see details about the error"
+    else:
+      if not flags['quiet']:
+        print "Elevated process succeeded."
 
 def RegisterPyComCategory():
   """ Register the Python COM Server component category.