Python操作IHTMLDocument2用于自动化测试

有些软件的界面采用Win32窗口嵌套一个IE控件，用Spy++只能识别出一个Internet Explorer_Server控件。常用的几个API函数无法取到IE控件里面的内容，更无法对里面的控件进行操作，所以这给自动化带来了麻烦。本文将讲述如何使用Python获取IHTMLDocument2接口，用于自动化测试。

相应的Python实现代码如下：

<a></a>

#!/usr/bin/env python

#coding:utf-8

__author__ = 'CoderZh'

import sys

# Important for multithreading

sys.coinit_flags = 0 # pythoncom.COINIT_MULTITHREADED

import win32com

import win32com.client

import win32gui

import win32con

import pythoncom

def getIEServer(hwnd, ieServer):

    if win32gui.GetClassName(hwnd) == 'Internet Explorer_Server':

        ieServer.append(hwnd)

if __name__ == '__main__':

    #pythoncom.CoInitializeEx(0) # not use this for multithreading

    mainHwnd = win32gui.FindWindow('windowclass', 'windowtitle')

    if mainHwnd:

        ieServers = []

        win32gui.EnumChildWindows(mainHwnd, getIEServer, ieServers)

        if len(ieServers) &gt; 0:

            ieServer = ieServers[0]

            msg = win32gui.RegisterWindowMessage('WM_HTML_GETOBJECT')

            ret, result = win32gui.SendMessageTimeout(ieServer, msg, 0, 0, win32con.SMTO_ABORTIFHUNG, 1000)

            ob = pythoncom.ObjectFromLresult(result, pythoncom.IID_IDispatch, 0)

            doc = win32com.client.dynamic.Dispatch(ob)

            print doc.url

            doc.all['id'].click()

    #pythoncom.CoUninitialize()

IHTMLDocument2是线程安全的，默认情况下不能在多线程中使用，否则会抛异常。但是在具体使用过程中，又必须使用多线程。解决办法就是上面的代码中的：

这句必须在开头的时候设定，同时，不要再显示调用pythoncom.CoInitializeEx(0)和 pythoncom.CoUninitialize()。

基本能够满足自动化测试的需要，可以在此基础上封装出更易使用的自动化UI测试框架。