利用Selenium多用戶同時開啟多線程登錄博客園

做自動化測試的朋友經常會遇到多終端同時做某一類的操作，最常見的當然要從登錄開始做起，下面介紹利用selenium在單個瀏覽器開啟多個線程，同時做用戶登錄的操作，后面會介紹多進程方式，請小伙伴們繼續關注，下面開始上demo代碼

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time    : 2020/5/11 22:01
# @Author  : Tang Yiwei
# @Email   : [e-mail]
# @File    : test_mult_user_login.py
# @Software: PyCharm

import threading
from time import sleep,time
from selenium import webdriver
from multiprocessing import current_process


def user_login(username,password):
    """
    單終端多線程登錄
    :param username:
    :param password:
    :return:
    """
    url = "https://account.cnblogs.com/signin"
    driver = webdriver.Chrome(executable_path="E:\PyCharmPro\svn\mbh\drivers\chromedriver.exe")
    procName = current_process().name
    print("當前進程:",procName)
    sleep(1)
    start = time()
    print("session_id: " + driver.session_id)
    try:
        driver.get(url)
        driver.maximize_window()
        driver.find_element_by_id("LoginName").send_keys(username)
        sleep(1.5)
        driver.find_element_by_id("Password").send_keys(password)
        sleep(1.5)
        driver.find_element_by_id("submitBtn").click()
    except Exception as e:
        raise e
    finally:
        driver.close()
        driver.quit()
    end = time()
    print("Task run %0.2f seconds." % (end - start))

def list_of_groups(list_info, per_list_len):
    '''
    將列表分割成指定長度的子列表，每個列表長度為當前測試並發數
    :param list_info:   列表，需要進行參數化的總列表
    :param per_list_len:  每個小列表的長度
    :return:
    '''
    list_of_group = zip(*(iter(list_info),) *per_list_len)
    end_list = [list(i) for i in list_of_group] # i is a tuple
    count = len(list_info) % per_list_len
    end_list.append(list_info[-count:]) if count !=0 else end_list
    return end_list

if __name__ == '__main__':
    userinfo = [
        {"username": "xiaoming", "password": "123456"},
        {"username": "dapeng", "password": "11223344"},
        {"username": "zuxiaobin", "password": "9090990"},
        {"username": "wangtao", "password": "4556677"}
    ]
    threads = []
    split_user_list = list_of_groups(userinfo,2)  # 這里只需要修改分割的數量即可實現循環登錄和並發登錄的數量
    print(split_user_list)
    for user_list in split_user_list:
        threads = [threading.Thread(target=user_login, args=(i["username"], i["password"])) for i in user_list]

        for t in threads:
            t.start()

        for t in threads:
            t.join()