近日,发现数据批量入错,由于没有数据库权限,之前应用界面重复操作,量有200多个,尝试使用selenium自动化一把
- 给chrome安装katalon automation recorder(google应用商店,过墙)
- 界面操作录制脚本,export 为python脚本
- 修改python脚本:如加上时间等待(sleep),修正katalon不支持的python语法(# ERROR: Caught exception [ERROR: Unsupported command [selectFrame | relative=parent | ]]),主要是切换frame
参考文章:Python selenium —— 深刻解析及操作frame、iframe
# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
class UntitledTestCase(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.base_url = "https://www.katalon.com/"
self.verificationErrors = []
self.accept_next_alert = True
def test_untitled_test_case(self):
driver = self.driver
driver.get("http://www.公司内部地址.com/")
driver.find_element_by_id("txtUserName").clear()
driver.find_element_by_id("txtUserName").send_keys("帐号")
driver.find_element_by_id("txtPassword").clear()
driver.find_element_by_id("txtPassword").send_keys("帐号密码")
driver.find_element_by_id("btnLogin").click()
driver.find_element_by_link_text(u"企业数据").click()
# ERROR: Caught exception [ERROR: Unsupported command [selectWindow | win_ser_1 | ]]
time.sleep(10)
driver.switch_to_window(driver.window_handles[1])#弹出窗口切换
driver.find_element_by_id("hxzc_img").click()
# ERROR: Caught exception [ERROR: Unsupported command [selectWindow | win_ser_2 | ]]
# ERROR: Caught exception [ERROR: Unsupported command [selectFrame | index=0 | ]]
time.sleep(10)
driver.switch_to_window(driver.window_handles[2])#弹出窗口切换
driver.switch_to.frame(0)
driver.find_element_by_xpath(u"//td[@οnclick=\"init('承包单元管理')\"]").click()
# ERROR: Caught exception [ERROR: Unsupported command [selectFrame | relative=parent | ]]
driver.switch_to.parent_frame()#frame切换
time.sleep(0.1)
# ERROR: Caught exception [ERROR: Unsupported command [selectFrame | index=1 | ]]
driver.switch_to.frame(1)#frame切换
dy=['844070110040011','844070110040009','844070110040010']
for cbdycode in dy:
print cbdycode
driver.find_element_by_id("unitCode").click()
time.sleep(0.1)
driver.find_element_by_id("unitCode").clear()
time.sleep(0.1)
driver.find_element_by_id("unitCode").send_keys(cbdycode)
driver.find_element_by_xpath(u"//input[@value='搜索']").click()
time.sleep(1)
driver.find_element_by_xpath("(//input[@type='checkbox'])[2]").click()
time.sleep(0.3)
driver.find_element_by_id("modBtn").click()
time.sleep(2)
driver.find_element_by_id("mgmtType1").click()
time.sleep(0.3)
driver.find_element_by_id("mgmtType2").click()
time.sleep(0.3)
Select(driver.find_element_by_id("mgmtType2")).select_by_visible_text(u"实体渠道承包经营")
time.sleep(0.3)
driver.find_element_by_id("mgmtType2").click()
time.sleep(0.3)
driver.find_element_by_id("mgmtType").click()
time.sleep(0.3)
Select(driver.find_element_by_id("mgmtType")).select_by_visible_text(u"营维合一小区承包经营")
time.sleep(0.3)
driver.find_element_by_id("mgmtType").click()
time.sleep(0.3)
driver.find_element_by_id("itNumber").click()
time.sleep(0.3)
driver.find_element_by_id("itNumber").clear()
time.sleep(0.3)
driver.find_element_by_id("itNumber").send_keys("XQJM2018032700656")
time.sleep(0.3)
driver.find_element_by_link_text(u"保存").click()
time.sleep(0.3)
self.assertEqual(u"保存成功", self.close_alert_and_get_its_text())
time.sleep(1)
def is_element_present(self, how, what):
try: self.driver.find_element(by=how, value=what)
except NoSuchElementException as e: return False
return True
def is_alert_present(self):
try: self.driver.switch_to_alert()
except NoAlertPresentException as e: return False
return True
def close_alert_and_get_its_text(self):
try:
alert = self.driver.switch_to_alert()
alert_text = alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally: self.accept_next_alert = True
def tearDown(self):
self.driver.quit()
self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
unittest.main()