python+selenium自动化测试（三）

# encoding=utf-8

import time
import sys
import testlogin.login as login

reload(sys)
sys.setdefaultencoding('utf8')


def selectMenu(argument):
    dr = login.driver
    '''选择top导航菜单'''
    # li_0000 = 与我有关，li_0001 = 项目管理，li_0002 = 评审管理，li_0003 = 客户管理，li_0005 = 系统管理
    menulist = {"li_0000", "li_0001", "li_0002", "li_0003", "li_0005"}
    for i in menulist:
        if i == argument:
            dr.find_element_by_id("my_menu")
            dr.find_element_by_id(i).click()
    return "pass"


def selectLeftMenu(argument):
    dr = login.driver
    '''选择left导航菜单'''
    # c_00010101（商机录入），c_00010102（商机查询），c_00010104(商机跟踪)，c_00010107（商机补录），c_00010108（商机状态变动）
    leftmenulist = {"c_00010101", "c_00010102", "c_00010104", "c_00010107", "c_00010108"}
    for i in leftmenulist:
        if i == argument:
            '''若是传入的argument=导航菜单的ID了，我就进行点击事件'''
            time.sleep(1)
            dr.find_element_by_id("my_menu")
            dr.find_element_by_id("li_0001").click()
            time.sleep(1)
            dr.find_element_by_id("CM_mMeun_div")
            dr.find_element_by_xpath(".//*[@id='a_0001']/div[1]/span").click()
            time.sleep(1)
            dr.find_element_by_id(i).click()
            time.sleep(1)
            dr.switch_to_frame("mainConFrame")
    return "pass"


def selecrCss(id, name):
    dr = login.driver
    if id != '':
        js = "var id = document.getElementById(%d);''id.style.border=\"2px solid red\";" % id
        dr.execute_script(js)
    if name != '':
        js = "var id = document.getElementByName(%d);''id.style.border=\"2px solid red\";" % name
        dr.execute_script(js)


'''退出浏览器驱动'''


def quit():
    dr = login.driver
    dr.quit()