diff options
Diffstat (limited to 'Tools/scripts/win_add2path.py')
| -rw-r--r-- | Tools/scripts/win_add2path.py | 29 |
1 files changed, 14 insertions, 15 deletions
diff --git a/Tools/scripts/win_add2path.py b/Tools/scripts/win_add2path.py index 1c9aedc..876bfb2 100644 --- a/Tools/scripts/win_add2path.py +++ b/Tools/scripts/win_add2path.py @@ -10,28 +10,27 @@ Licensed to PSF under a Contributor Agreement. import sys import site import os -import winreg +import _winreg -HKCU = winreg.HKEY_CURRENT_USER +HKCU = _winreg.HKEY_CURRENT_USER ENV = "Environment" PATH = "PATH" -DEFAULT = "%PATH%" +DEFAULT = u"%PATH%" def modify(): pythonpath = os.path.dirname(os.path.normpath(sys.executable)) scripts = os.path.join(pythonpath, "Scripts") appdata = os.environ["APPDATA"] if hasattr(site, "USER_SITE"): - usersite = site.USER_SITE.replace(appdata, "%APPDATA%") - userpath = os.path.dirname(usersite) + userpath = site.USER_SITE.replace(appdata, "%APPDATA%") userscripts = os.path.join(userpath, "Scripts") else: userscripts = None - with winreg.CreateKey(HKCU, ENV) as key: + with _winreg.CreateKey(HKCU, ENV) as key: try: - envpath = winreg.QueryValueEx(key, PATH)[0] - except OSError: + envpath = _winreg.QueryValueEx(key, PATH)[0] + except WindowsError: envpath = DEFAULT paths = [envpath] @@ -40,19 +39,19 @@ def modify(): paths.append(path) envpath = os.pathsep.join(paths) - winreg.SetValueEx(key, PATH, 0, winreg.REG_EXPAND_SZ, envpath) + _winreg.SetValueEx(key, PATH, 0, _winreg.REG_EXPAND_SZ, envpath) return paths, envpath def main(): paths, envpath = modify() if len(paths) > 1: - print("Path(s) added:") - print('\n'.join(paths[1:])) + print "Path(s) added:" + print '\n'.join(paths[1:]) else: - print("No path was added") - print("\nPATH is now:\n%s\n" % envpath) - print("Expanded:") - print(winreg.ExpandEnvironmentStrings(envpath)) + print "No path was added" + print "\nPATH is now:\n%s\n" % envpath + print "Expanded:" + print _winreg.ExpandEnvironmentStrings(envpath) if __name__ == '__main__': main() |