# -*- coding: utf-8 -*-

# Run from the command window (Run as Administrator!):

# python registerPython.py

# script to register Python 2.0 or later for use with win32all

# and other extensions that require Python registry settings

#

# written by Joakim Löw for Secret Labs AB / PythonWare

#

# source:

# http://www.pythonware.com/products/works/articles/regpy20.htm

import sys

from _winreg import *

# tweak as necessary

version = sys.version[:3]

installpath = sys.prefix

installpath = "D:\Program Files\Python36"  # 注意这里是python安装路径

regpath = "SOFTWARE\\Python\\Pythoncore\\%s\\" % (version)

installkey = "InstallPath"

pythonkey = "PythonPath"

pythonpath = "%s;%s\\Lib\\;%s\\DLLs\\" % (

    installpath, installpath, installpath

)

def RegisterPy():

    try:

        reg = OpenKey(HKEY_LOCAL_MACHINE, regpath)

    except EnvironmentError:

        try:

            print("No Key Found, attempting to register")

            reg = CreateKey(HKEY_LOCAL_MACHINE, regpath)

            SetValue(reg, installkey, REG_SZ, installpath)

            SetValue(reg, pythonkey, REG_SZ, pythonpath)

            CloseKey(reg)

        except:

            print("*** Unable to register!")

            return

        print("--- Python", version, "is now registered!")

        return

    if (QueryValue(reg, installkey) == installpath and

        QueryValue(reg, pythonkey) == pythonpath):

        CloseKey(reg)

        print("=== Python", version, "is already registered!")

        return

    CloseKey(reg)

    print("*** Unable to register!")

    print("*** You probably have another Python installation!")

if __name__ == "__main__":

    RegisterPy()