selenium自动化代码写好之后领导觉得重复代码比较多，想把重复的定位元素写在json中读取

selenium自动化代码写好之后领导觉得重复代码比较多，想把重复的定位元素写在json中读取

就是说以前如果三条都是定位的ID元素写在代码中，现在只在代码中写一条，循环读取执行三条代码

再就是如何判断执行click（）和sendkeys（），大神们，好几天没睡了，求帮忙，如果谁遇到过类似的帮忙贴一段代码给点参考思路

from test_case.BaseTestCase import BaseTestCase
import time
import unittest2
from func.csvFileManager2 import reader

class OfficesTeachTest(BaseTestCase):


    def test_jonin(self):
        self.driver.get("http://rst2demo.yueeasy.com/htmp_dev_master/pages/main/login/login.jsp")
        self.driver.find_element_by_id("account").send_keys("jy119113")
        self.driver.find_element_by_id("password").send_keys("123456")
        self.driver.find_element_by_class_name("btn-login").click()



    def test_officesteach(self):

        time.sleep(5)
        self.driver.find_element_by_link_text("入科教育").click()
        self.driver.find_element_by_link_text("组织入科教育").click()
        tables = reader("officesteach_test_cases.csv")
        for row in tables:
            self.driver.find_element_by_xpath('//*[@id="frmEduRecord"]/div[1]/div[5]/div/button[2]').click()
            time.sleep(2)
            # 定位增加的入科日期6
            self.driver.execute_script('document.getElementsByClassName("form-control")[6].removeAttribute("readonly")')
            self.driver.find_elements_by_class_name("form-control")[6].send_keys(row[0])
            # 定位增加的教育日期7
            time.sleep(2)
            self.driver.execute_script('document.getElementsByClassName("form-control")[7].removeAttribute("readonly")')
            self.driver.find_elements_by_class_name("form-control")[7].send_keys(row[1])

            # 教育时间
            time.sleep(2)
            self.driver.execute_script('document.getElementById("eduRecordTime").removeAttribute("readonly")')
            self.driver.find_element_by_id("eduRecordTime").clear()
            self.driver.find_element_by_id("eduRecordTime").send_keys(row[2])
            time.sleep(2)
            self.driver.find_element_by_id("employeeNames").click()
            time.sleep(2)
            self.driver.find_element_by_xpath('//*[@id="emp_table"]/tbody/tr[2]/td[1]/input').click()
            time.sleep(2)
            self.driver.find_element_by_xpath('//*[@id="employee_div"]/div/div[3]/button').click()
            time.sleep(2)
            self.driver.find_element_by_xpath('//*[@id="edu_record_info_form"]/table/tbody/tr[6]/td[2]/input').send_keys(row[3])
            self.driver.find_element_by_name("eduRecord.eduRecordContent").send_keys(row[4])
            time.sleep(2)
            self.driver.find_element_by_name("eduRecord.remark").send_keys(row[5])
            time.sleep(2)
            self.driver.find_element_by_id("info_btn").click()
            time.sleep(2)
        query=self.driver.find_element_by_css_selector(".btn.btn-primary").get_attribute("title")
        self.assertEqual("查询",query)


if __name__ == '__main__':
    unittest2.main()

举个例子，这里面写了很多id定位元素或者xpath定位，能不能写一条就可以，其他的循环读取执行

可以进一步封装代码, 将公共部分代码单独抽离出来定义, 在定位时, 尽量使用find_element灵活去定位