def set_env(name, value, current=True):
"""Set HKCU/HKLM environment variables"""
root = (
winreg.HKEY_CURRENT_USER
if current
else winreg.HKEY_LOCAL_MACHINE
)
key = winreg.OpenKey(root, "Environment")
try:
_x, key_type = winreg.QueryValueEx(key, name)
except WindowsError:
key_type = winreg.REG_EXPAND_SZ
key = winreg.OpenKey(
root, "Environment", 0, winreg.KEY_SET_VALUE
)
winreg.SetValueEx(key, name, 0, key_type, value)
from win32gui import SendMessageTimeout
from win32con import (
HWND_BROADCAST,
WM_SETTINGCHANGE,
SMTO_ABORTIFHUNG,
)
SendMessageTimeout(
HWND_BROADCAST,
WM_SETTINGCHANGE,
0,
"Environment",
SMTO_ABORTIFHUNG,
5000,
)
def register(target, current=True):
"""Register a Python distribution in Windows registry"""
root = (winreg.HKEY_CURRENT_USER if current else winreg.HKEY_LOCAL_MACHINE)
# Extensions
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".py"),
"",
0,
winreg.REG_SZ,
"Python.File",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".pyw"),
"",
0,
winreg.REG_SZ,
"Python.NoConFile",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".pyc"),
"",
0,
winreg.REG_SZ,
"Python.CompiledFile",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".pyo"),
"",
0,
winreg.REG_SZ,
"Python.CompiledFile",
)
# MIME types
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".py"),
"Content Type",
0,
winreg.REG_SZ,
"text/plain",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C % ".pyw"),
"Content Type",
0,
winreg.REG_SZ,
"text/plain",
)
# Verbs
python = utils.get_python_executable(target) # PyPy !
pythonw = python[:-4] + 'w.exe'
spyder = osp.abspath(osp.join(target, os.pardir, 'Spyder.exe'))
if not osp.isfile(spyder):
spyder = '%s" "%s\Scripts\spyder' % (
pythonw,
target,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("", "open")),
"",
0,
winreg.REG_SZ,
'"%s" "%%1" %%*' % python,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("NoCon", "open")),
"",
0,
winreg.REG_SZ,
'"%s" "%%1" %%*' % pythonw,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("Compiled", "open")),
"",
0,
winreg.REG_SZ,
'"%s" "%%1" %%*' % python,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("", EWI)),
"",
0,
winreg.REG_SZ,
'"%s" "%s\Lib\idlelib\idle.pyw" -n -e "%%1"' % (pythonw, target),
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("NoCon", EWI)),
"",
0,
winreg.REG_SZ,
'"%s" "%s\Lib\idlelib\idle.pyw" -n -e "%%1"' % (pythonw, target),
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("", EWS)),
"",
0,
winreg.REG_SZ,
'"%s" "%%1"' % spyder,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_C2 % ("NoCon", EWS)),
"",
0,
winreg.REG_SZ,
'"%s" "%%1"' % spyder,
)
# Drop support
handler = "{60254CA5-953B-11CF-8C96-00AA00B8708C}"
for ftype in ("", "NoCon", "Compiled"):
winreg.SetValueEx(
winreg.CreateKey(root, KEY_DROP1 % ftype),
"",
0,
winreg.REG_SZ,
handler,
)
# Icons
dlls = osp.join(target, 'DLLs')
winreg.SetValueEx(
winreg.CreateKey(root, KEY_I % ""),
"",
0,
winreg.REG_SZ,
r'%s\py.ico' % dlls,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_I % "NoCon"),
"",
0,
winreg.REG_SZ,
r'%s\py.ico' % dlls,
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_I % "Compiled"),
"",
0,
winreg.REG_SZ,
r'%s\pyc.ico' % dlls,
)
# Descriptions
winreg.SetValueEx(
winreg.CreateKey(root, KEY_D % ""),
"",
0,
winreg.REG_SZ,
"Python File",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_D % "NoCon"),
"",
0,
winreg.REG_SZ,
"Python File (no console)",
)
winreg.SetValueEx(
winreg.CreateKey(root, KEY_D % "Compiled"),
"",
0,
winreg.REG_SZ,
"Compiled Python File",
)
# PythonCore entries
short_version = utils.get_python_infos(target)[0]
long_version = utils.get_python_long_version(target)
key_core = (KEY_S1 % short_version) + r'\%s'
winreg.SetValueEx(
winreg.CreateKey(root, key_core % 'InstallPath'),
"",
0,
winreg.REG_SZ,
target,
)
winreg.SetValueEx(
winreg.CreateKey(root, key_core % r'InstallPath\InstallGroup'),
"",
0,
winreg.REG_SZ,
"Python %s" % short_version,
)
winreg.SetValueEx(
winreg.CreateKey(root, key_core % 'Modules'),
"",
0,
winreg.REG_SZ,
"",
)
winreg.SetValueEx(
winreg.CreateKey(root, key_core % 'PythonPath'),
"",
0,
winreg.REG_SZ,
r"%s\Lib;%s\DLLs" % (target, target),
)
winreg.SetValueEx(
winreg.CreateKey(
root,
key_core % r'Help\Main Python Documentation',
),
"",
0,
winreg.REG_SZ,
r"%s\Doc\python%s.chm" % (target, long_version),
)
# Create start menu entries for all WinPython launchers
for path, desc, fname in _get_shortcut_data(target, current=current):
utils.create_shortcut(path, desc, fname)
# Register the Python ActiveX Scripting client (requires pywin32)
axscript = osp.join(
target,
'Lib',
'site-packages',
'win32comext',
'axscript',
'client',
'pyscript.py',
)
if osp.isfile(axscript):
subprocess.call('"%s" "%s"' % (python, axscript), cwd=target)
else:
print(
'Unable to register ActiveX: please install pywin32',
file=sys.stderr,
)
