cpython/Tools/scripts/win_add2path.py at pythoncapi · pythoncapi/cpython · GitHub

History

58 lines (48 loc) · 1.62 KB

Raw

"""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 = "%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()

cpython/Tools/scripts/win_add2path.py at pythoncapi · pythoncapi/cpython · GitHub

Uh oh!

Expand file tree

Breadcrumbs

win_add2path.py

Latest commit

History

Breadcrumbs

win_add2path.py

File metadata and controls