from selenium import webdriver
import time

d = webdriver.Chrome()
d.maximize_window()  # 窗口最大化###登录某网站
d.get('http://xx.xxx.xx.xx:xxxx/')
d.find_element_by_xpath('//*[@id="userName"]').send_keys('xxx')
d.find_element_by_xpath('//*[@id="userPwd"]').send_keys('xxx')
d.find_element_by_xpath('//*[@id="login"]').click()
time.sleep(2)###切换表单，进入到操作页面
d.find_element_by_xpath('//*[@id="menu_ul"]/li[5]/a').click()
d.switch_to_frame('mainframe2')
d.find_element_by_xpath('//*[@id="nav-accordion"]/li[2]/a').click()
d.switch_to_frame('mainframe')
d.switch_to_frame('vehIframe')###定位到要获取标签的*元素，并使用for循环获取
names = d.find_elements_by_xpath('//*[@id="vehGroupTree_1"]')  #for循环遍历-->重点在这里
lists = []
for i in names:
    a = i.text
    lists.append(a)
    print(a, i.get_attribute("href"))  # 打印遍历标签出来的内容和获取href属性的内容
    print(lists)
    elements = '中国'
    if elements in lists:
      print('元素在列表中，成功')
      self.assertIn(elements, lists)