Tools/scripts/win_add2path.py

   1 """Add Python to the search path on Windows
   2
   3 This is a simple script to add Python to the Windows search path. It
   4 modifies the current user (HKCU) tree of the registry.
   5
   6 Copyright (c) 2008 by Christian Heimes <christian@cheimes.de>
   7 Licensed to PSF under a Contributor Agreement.
   8 """
   9
  10 import sys
  11 import site
  12 import os
  13 import _winreg
  14
  15 HKCU = _winreg.HKEY_CURRENT_USER
  16 ENV = "Environment"
  17 PATH = "PATH"
  18 DEFAULT = u"%PATH%"
  19
  20 def modify():
  21     pythonpath = os.path.dirname(os.path.normpath(sys.executable))
  22     scripts = os.path.join(pythonpath, "Scripts")
  23     appdata = os.environ["APPDATA"]
  24     if hasattr(site, "USER_SITE"):
  25         userpath = site.USER_SITE.replace(appdata, "%APPDATA%")
  26         userscripts = os.path.join(userpath, "Scripts")
  27     else:
  28         userscripts = None
  29
  30     with _winreg.CreateKey(HKCU, ENV) as key:
  31         try:
  32             envpath = _winreg.QueryValueEx(key, PATH)[0]
  33         except WindowsError:
  34             envpath = DEFAULT
  35
  36         paths = [envpath]
  37         for path in (pythonpath, scripts, userscripts):
  38             if path and path not in envpath and os.path.isdir(path):
  39                 paths.append(path)
  40
  41         envpath = os.pathsep.join(paths)
  42         _winreg.SetValueEx(key, PATH, 0, _winreg.REG_EXPAND_SZ, envpath)
  43         return paths, envpath
  44
  45 def main():
  46     paths, envpath = modify()
  47     if len(paths) > 1:
  48         print "Path(s) added:"
  49         print '\n'.join(paths[1:])
  50     else:
  51         print "No path was added"
  52     print "\nPATH is now:\n%s\n" % envpath
  53     print "Expanded:"
  54     print _winreg.ExpandEnvironmentStrings(envpath)
  55
  56 if __name__ == '__main__':
  57     main()