from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait(3)
#全屏顯示
driver.maximize_window()

#指定瀏覽器大小
driver.set_window_size(600,400)

#前進
driver.forward()
#后退
driver.back()

#刷新頁面
driver.refresh()

img_dir = "C:\\測試\\img.png"
driver.save_screenshot(img_dir)

from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.implicitly_wait(3)
driver.get("https://www.baidu.com")
driver.find_element(By.ID,"kw").send_keys("單擊事件")
element = driver.find_element(By.CSS_SELECTOR,'#su')
ActionChains(driver).click(element).perform()

#鼠標懸停
ActionChains(driver).move_to_element(element).perform()

#鼠標右擊擊事件
ActionChains(driver).context_click(element).perform()

#鼠標雙擊事件
ActionChains(driver).double_click(element).perform()

#鼠標拖動事件
tag = driver.find_element(By.CSS_SELECTOR,"#div1")
ActionChains(driver).drag_and_drop(element,tag).perform()

from selenium import webdriver
from selenium.webdriver.common.by import By
# 引入Keys模塊
from selenium.webdriver.common.keys import Keys
driver = webdriver.Chrome()
driver.get("https://www.baidu.com")
#鍵盤按鍵操作
element=driver.find_element(By.ID,"kw")
element.send_keys("Enter回車事件")
element.send_keys(Keys.ENTER)

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.wait import WebDriverWait

driver = webdriver.Chrome()
#隱式等待
driver.implicitly_wait(3)
driver.get("https://www.baidu.com")
#顯示等待
WebDriverWait(driver,5,1).until(expected_conditions.visibility_of_element_located(By.ID,"kw"))

# !/usr/bin/python3
# -*- coding:utf-8 -*-
"""
@author: JHC000abc@gmail.com
@file: demo.py
@time: 2024/1/30 10:46
@desc:

"""
import json
import os
import time
import datetime
import random
import traceback

from DrissionPage import WebPage, ChromiumOptions, SessionOptions


class DrissionPageDemo(object):
    """

    """

    def __init__(self):
        """

        """

    def get_page(self, ua=None, incognito=False, time_out=60, headless=False, cookies=None):
        """

        :param ua:
        :param incognito:
        :return:
        """
        self.page = None
        co = ChromiumOptions()
        so = SessionOptions()
        if cookies:
            so.set_cookies(cookies)
        if ua:
            co.set_user_agent(user_agent=ua)
        co.incognito(incognito)
        co.set_argument('--window-size', '800,600')
        co.auto_port(True)
        co.headless(headless)
        co.ignore_certificate_errors(True)
        co.mute(True)
        co.set_timeouts(page_load=time_out)

        self.page = WebPage(driver_or_options=co, session_or_options=so)

        return self.page

    def save_result(self, html):
        """

        :param html:
        :return:
        """
        os.makedirs("result", exist_ok=True)
        file = "result/result_{}.html".format(
            datetime.datetime.now().strftime("%Y-%m-%d %H-%M-%S"))

        with open(file, "w", encoding="utf-8")as fp:
            fp.write(html)

    def read_file(self, file):
        """

        :param file:
        :return:
        """
        res = []
        with open(file, "r", encoding="utf-8")as fp:
            for i in fp:
                res.append(i.strip())
        return res

    def read_json_file(self, file):
        """

        :param file:
        :return:
        """
        with open(file, "r", encoding="utf-8")as fp:
            return json.loads(fp.read())

    def check_load_over(self):
        """

        :return:
        """

    def process(self):
        """

        :return:
        """
        url_list = self.read_file("url.list")
        ua_list = self.read_file("ua.list")
        cookie_list = self.read_file("cookie.list")
        settings = self.read_json_file("settings.json")
        time_out = settings["time_out"]
        wait = settings["wait"]
        save = settings["save"]
        incognito = settings["incognito"]
        headless = settings["headless"]
        while True:
            try:
                if ua_list:
                    ua = random.choices(ua_list)[0]
                else:
                    ua = None
                if cookie_list:
                    cookie = random.choices(cookie_list)
                else:
                    cookie = None

                self.get_page(ua, incognito, time_out, headless, cookie)

                url = random.choices(url_list)[0]
                print(f"url : {url}")

                self.page.get(url, retry=3, interval=3)
                if save:
                    html = self.page.html
                    self.save_result(html)

                self.page.quit()

                if len(wait) == 1:
                    _sleep = wait[0]
                else:
                    _sleep = random.randint(wait[0], wait[1])

                while _sleep >= 0:
                    print(f"SLEEP : {_sleep}", end='\r')
                    time.sleep(1)
                    _sleep -= 1
            except BaseException:
                print(traceback.print_exc())
                os.system("taskkill /F /IM chrome.exe")


if __name__ == '__main__':
    dp = DrissionPageDemo()
    dp.process()