cpython/Tools/scripts/win_add2path.py

59 lines
1.6 KiB
Python
Raw Normal View History

"""Add Python to the search path on Windows
This is a simple script to add Python to the Windows search path. It
modifies the current user (HKCU) tree of the registry.
Copyright (c) 2008 by Christian Heimes <christian@cheimes.de>
Licensed to PSF under a Contributor Agreement.
"""
import sys
import site
import os
2008-05-25 04:45:51 -03:00
import winreg
2008-05-25 04:45:51 -03:00
HKCU = winreg.HKEY_CURRENT_USER
ENV = "Environment"
PATH = "PATH"
2010-08-02 19:59:44 -03:00
DEFAULT = "%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)
userscripts = os.path.join(userpath, "Scripts")
else:
userscripts = None
2008-05-25 04:45:51 -03:00
with winreg.CreateKey(HKCU, ENV) as key:
try:
2008-05-25 04:45:51 -03:00
envpath = winreg.QueryValueEx(key, PATH)[0]
except OSError:
envpath = DEFAULT
paths = [envpath]
for path in (pythonpath, scripts, userscripts):
if path and path not in envpath and os.path.isdir(path):
paths.append(path)
envpath = os.pathsep.join(paths)
2008-05-25 04:45:51 -03:00
winreg.SetValueEx(key, PATH, 0, winreg.REG_EXPAND_SZ, envpath)
return paths, envpath
def main():
paths, envpath = modify()
if len(paths) > 1:
2009-11-08 20:49:19 -04:00
print("Path(s) added:")
print('\n'.join(paths[1:]))
else:
2009-11-08 20:49:19 -04:00
print("No path was added")
print("\nPATH is now:\n%s\n" % envpath)
print("Expanded:")
print(winreg.ExpandEnvironmentStrings(envpath))
if __name__ == '__main__':
main()