# Add Python DLL directory to PATH

from _winreg import *
import os
import sys

def install():
    path = r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment'
    reg = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
    key = OpenKey(reg, path, 0, KEY_ALL_ACCESS)

    current_path, type_id = QueryValueEx(key, "PATH")
    dll_path = os.path.join(sys.prefix, 'DLLs')

    if not dll_path.upper() in current_path.upper().split(";"):
        new_path = current_path + ';' + dll_path
        SetValueEx(key, "PATH", 0, REG_EXPAND_SZ, new_path)

    CloseKey(key)
    CloseKey(reg)

if __name__=='__main__':
    if len(sys.argv) == 2 and sys.argv[1] == '-install':
        install()