GeezerVison is showing github.com — cpython/Tools/scripts/win_add2path.py at 2.7 · python/cpython

Expand file tree

win_add2path.py

Copy path

More file actions

Latest commit

History

57 lines (47 loc) · 1.58 KB

win_add2path.py

Copy path

File metadata and controls

57 lines (47 loc) · 1.58 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

"""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.

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 = 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"):

userpath = site.USER_SITE.replace(appdata, "%APPDATA%")

userscripts = os.path.join(userpath, "Scripts")

else:

userscripts = None

with _winreg.CreateKey(HKCU, ENV) as key:

try:

envpath = _winreg.QueryValueEx(key, PATH)[0]

except WindowsError:

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()