mirror of
https://github.com/python/cpython.git
synced 2024-12-14 12:26:47 +08:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
"""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
|
|
import winreg
|
|
|
|
HKCU = winreg.HKEY_CURRENT_USER
|
|
ENV = "Environment"
|
|
PATH = "PATH"
|
|
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
|
|
|
|
with winreg.CreateKey(HKCU, ENV) as key:
|
|
try:
|
|
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)
|
|
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:]))
|
|
else:
|
|
print("No path was added")
|
|
print("\nPATH is now:\n%s\n" % envpath)
|
|
print("Expanded:")
|
|
print(winreg.ExpandEnvironmentStrings(envpath))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|