#!/usr/bin/env python
# -*- coding=utf-8 -*-
# coding: utf-8
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from common.read_txt import read_txt
# from common.read_csv import read_csv
from common.read_xlsx import read_xlsx
def get_ele_times(driver, times, func):
return WebDriverWait(driver, times).until(func)
def open_browser():
"""
打开浏览器驱动
:return: 返回句柄
"""
driver = webdriver.Chrome()
return driver
def open_url(driver, url):
"""
打开网址
:return:
"""
driver.get(url)
driver.maximize_window()
def find_element(driver, kwargs):
"""
:param driver:浏览器句柄, kwargs必须是字典
1：login_text：登录按钮， //*[@id="head"]/div[1]/div[3]/a[7]
2：user_login_button：用户名登录按钮， //*[text()='用户名登录']
3：user_id：用户名， //*[@name=""userName""]
4：pwd_id：密码， //*[@name="password"]
5：login_id：立即登录按钮， //*[@id="TANGRAM__PSP_10__submit"]
:return：login_text, user_login_button, user_id, pwd_id, login_id
"""
get_ele_times(driver, 3, lambda d: d.find_element_by_xpath(kwargs['login_text'])).click()
get_ele_times(driver, 5, lambda d: d.find_element_by_xpath(kwargs['user_login_button'])).click()
user_id = driver.find_element_by_xpath(kwargs['user_id'])
pwd_id = driver.find_element_by_xpath(kwargs['pwd_id'])
login_id = driver.find_element_by_xpath(kwargs['login_id'])
ele_tuple = (user_id, pwd_id, login_id)
return ele_tuple
def send_values(ele_tuple, account_dict):
"""
:ele_tuple: 元素对象名称组成的元组
:account_dict : 用户要输入的值组成的字典
:return:
"""
list_key = ['user_name', 'pwd']
i = 0
for key in list_key:
ele_tuple[i].clear()
ele_tuple[i].send_keys(account_dict[key])
i = i + 1
ele_tuple[-1].click()
def login_test(ele_dict, account_dict):
"""
:param ele_dict:页面元素xpath值组成的一个字典
:return:
"""
driver = open_browser()
url = 'https://www.baidu.com/'
open_url(driver, url)
ele_tuple = find_element(driver, ele_dict)
send_values(ele_tuple, account_dict)
driver.quit()
if __name__ == '__main__':
"""
login_text0 = '//*[@id="head"]/div[1]/div[3]/a[7]'
user_login_button0 = '//*[@id="TANGRAM__PSP_10__footerULoginBtn"]'
user_id0 = '//*[@name="userName"]'
pwd_id0 = '//*[@name="password"]'
login_id0 = '//*[@id="TANGRAM__PSP_10__submit"]'
ele_dict0 = {'login_text': login_text0, 'user_login_button': user_login_button0,
'user_id': user_id0, 'pwd_id': pwd_id0, 'login_id': login_id0}
user_name0 = 'aaa'
pwd0 = '123'
account_dict0 = {'user_name': user_name0, 'pwd': pwd0}
"""
ele_dict0 = read_txt('..\\test_file\\element.txt')
print(ele_dict0)
xlsx_dict = read_xlsx(path='..\\test_file\\send_values.xlsx', sheet='user_pwd')
for account in xlsx_dict:
print(account)
login_test(ele_dict0, account)