windows下一个简单的输入法模块[Python]

新博客地址：http://gorthon.sinaapp.com/

# coding:cp936

# by 书记: bh2#qq.com
# do what you want to do with this script
# pywin32 needed

import win32con
import win32api

class InputMethod(list):
    def __init__(self):
        name = "Keyboard Layout\\Preload"
        _k_id = win32api.RegOpenKey(win32con.HKEY_CURRENT_USER, name)
        
        name = "System\\CurrentControlSet\\Control\\Keyboard Layouts\\"
        _i, _running, _ids = 1, True, list()
        while _running:
            try:
                _id = win32api.RegQueryValueEx(_k_id, str(_i))[0]
                _i += 1
                _k_name = win32api.RegOpenKey(win32con.HKEY_LOCAL_MACHINE, name + _id)
                _name = win32api.RegQueryValueEx(_k_name, 'Layout Text')[0]
                self.append((_id, _name))
                win32api.RegCloseKey(_k_name)
            except:
                _running = False
                win32api.RegCloseKey(_k_id)

    def set(self, _im):
        win32api.LoadKeyboardLayout(_im[0], win32con.KLF_ACTIVATE)

    
if __name__ == '__main__':
    im = InputMethod()
    for i, v in enumerate(im):
        print '【输入法%d】: %s'  % (i, v[1])
    try:
        i = input('设置输入法(输入数字并回车): ')
        im.set(im[i])
        print '成功将输入法设置为【%s】' % im[i][1]
    except:
        print '输入有误'