把下面的代码改成python：public WebElement myFindElement(By by){ try { return driver.findElement(by); }catch (Exception e){ //找不到对应的元素就进入到弹窗的处理机制中来 //弹窗黑名单列表(Map结构保存弹窗信息，其中key为元素在dom中关键信息字符串，value为元素定位表达式) HashMap<String,By> alertMap = new HashMap<>(); alertMap.put("text="我知道了"",MobileBy.AndroidUIAutomator("new UiSelector().text("我知道了")")); alertMap.put("xxx",MobileBy.id("xxx")); alertMap.put("xxx",MobileBy.xpath("xxx")); Set<String> allKeys = alertMap.keySet(); //拿到页面的dom结构信息 String pageSource = driver.getPageSource(); for (String key:allKeys){ if(pageSource.contains(key)){ //点击对应的弹窗元素 driver.findElement(alertMap.get(key)).click(); } } //点完弹窗之后继续再来点击对应的元素 return driver.findElement(by); } }

def myFindElement(by):
try:
return driver.find_element(by)
except:
# 找不到对应的元素就进入到弹窗的处理机制中来
# 弹窗黑名单列表(Map结构保存弹窗信息，其中key为元素在dom中关键信息字符串，value为元素定位表达式)
alertMap = {}
alertMap["text=我知道了"] = MobileBy.AndroidUIAutomator('new UiSelector().text("我知道了")')
alertMap["xxx"] = MobileBy.ID('xxx')
alertMap["xxx"] = MobileBy.XPATH('xxx')
allKeys = alertMap.keys()
# 拿到页面的dom结构信息
pageSource = driver.page_source
for key in allKeys:
if key in pageSource:
# 点击对应的弹窗元素
driver.find_element(alertMap[key]).click()
# 点完弹窗之后继续再来点击对应的元素
return driver.find_element(by)