python操作注册表

dickrong

因为系统经常重装，所以想能不能有个办法可以很快的将开发环境从一个机器搬到另一个地方。
需要处理的有几部分数据：
1，代码和文档拷贝
   因为代码大部分都在svn中，因此代码的拷贝只需要考虑拷贝部分没有纳入svn的即可。
   文档因为相当部分没有在svn中因此需要拷贝。
2，开发工具的安装或拷贝
   尽量采用绿色版的，直接拷贝到指定的目录即可。
3，环境变量的设置
   根据开发工具的拷贝位置设置环境变量。

拷贝的问题很好解决，手工拷贝也不麻烦，关键是环境变量设置有点恼火，因此打算采用python来处理。

下面是简单示例，打算下周再根据实际情况完善这个代码。
# -*- coding: utf-8 -*-

__author__="nxf"
__date__ ="$2010-8-13 15:28:39$"

from winreg import *

PATH = "RM";
#追加值到指定value_name的值中
def append_value(key,value_name,value):
    old_value,type_id = QueryValueEx(key,value_name)
#    确保原来的值中没有包含新加入的值
    if (old_value.find(value) == -1):
        SetValueEx(key,value_name,0,REG_SZ,old_value+";"+value)
        print ("append value("+value+") to "+value_name);
def create_mvn_env():
    print ("create_mvn_env.......")
    env =  OpenKeyEx(HKEY_LOCAL_MACHINE,
                    "SYSTEM\\ControlSet001\\Control\\Session Manager\\Environment",
                    0,
                    KEY_SET_VALUE|KEY_READ
                    )
    SetValueEx(env,"M2_HOME",0,REG_SZ,"D:\\SDK\\apache-maven")
    SetValueEx(env,"M2_RESPOSITY",0,REG_SZ,"e:\\lib")
    append_value(env,PATH,'%M2_HOME%\\bin')
    print ("create_mvn_env ok!")

def display_reg():
    env =  OpenKey(HKEY_LOCAL_MACHINE,
                    "SYSTEM\\ControlSet001\\Control\\Session Manager\\Environment"
                    )
    try:
        i = 0
        while 1:
            name,value,type = EnumValue(env,i)
            if (name == PATH):
                print (value)
                break
            i += 1
    except WindowsError:
        print
if __name__ == "__main__":
    create_mvn_env();
    display_reg();

需要注意的是，我发现CreateKey()是创建“目录”，要设置“叶值”需要使用Ex结尾的代码。