# coding=utf-8
"""
    作者：gaojs
    功能：
    新增功能：
    日期：2022/6/9 18:21
"""
from selenium import webdriver
from selenium.webdriver.common.by import By
import threading
import time
import sys
browsers = []
threads = []
num = int(sys.argv[1])
i = 0
for i in range(0, num):
    # chrome_options = webdriver.ChromeOptions()
    # chrome_options.add_argument('--ignore-certificate-errors')
    # chrome_options.add_experimental_option('excludeSwitches', ['enable-logging'])
    browser = webdriver.Chrome()
    browser.implicitly_wait(3)
    browsers.append(browser)
def saml_request(browser):
    global i
    while True:
        try:
            browser.get("https://192.168.120.222/prx/000/http/localhost/login/sp1/default/login/index.html")
            # 点击高级选项
            if i == 0:
                browser.find_element_by_xpath('//button[@id="details-button"]').click()
                # 点击详情连接，跳出隐私连接
                browser.find_element_by_xpath('//a[@id="proceed-link"]').click()
            # 点击下拉框
            browser.find_element_by_xpath('//button[@class="btn dropdown-toggle btn-default"]').click()
            # 选择method
            browser.find_element_by_xpath('//a[@role="option"] //span[contains(text(),"sm1")]').click()
            # 点击sign in 按钮
            browser.find_element_by_xpath('//button[@id="saml_login"]').click()
            # 点击高级选项
            if i == 0:
                browser.find_element_by_xpath('//button[@id="details-button"]').click()
                # 点击详情连接，跳出隐私连接
                browser.find_element_by_xpath('//a[@id="proceed-link"]').click()
            # 跳转idp，等待时间
            time.sleep(2)
            # 登录操作
            browser.find_element_by_name("uname").send_keys("u1")
            browser.find_element_by_name("pwd").send_keys("u1")
            browser.find_element_by_name("submitbutton").click()
            time.sleep(2)
            # 退出登录
            browser.find_element_by_xpath("//a[text()='Logout']").click()
            print("%s %d" % (time.ctime(), i))
            i += 1
        except:
            pass
for i in range(0, num):
    thread = threading.Thread(target=saml_request, args=(browsers[i],))
    # 开启新线程
    thread.start()
    # 添加线程到线程列表
    threads.append(thread)
# 等待所有线程完成
for i in range(0, num):
    threads[i].join()
print('退出主线程')