diff options
Diffstat (limited to 'Tools/scripts/win_add2path.py')
| -rw-r--r-- | Tools/scripts/win_add2path.py | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/Tools/scripts/win_add2path.py b/Tools/scripts/win_add2path.py index 876bfb2..9259b44 100644 --- a/Tools/scripts/win_add2path.py +++ b/Tools/scripts/win_add2path.py @@ -10,12 +10,12 @@ 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 = u"%PATH%" +DEFAULT = "%PATH%" def modify(): pythonpath = os.path.dirname(os.path.normpath(sys.executable)) @@ -27,9 +27,9 @@ def modify(): else: userscripts = None - with _winreg.CreateKey(HKCU, ENV) as key: + with winreg.CreateKey(HKCU, ENV) as key: try: - envpath = _winreg.QueryValueEx(key, PATH)[0] + envpath = winreg.QueryValueEx(key, PATH)[0] except WindowsError: envpath = DEFAULT @@ -39,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() |