python-selenium抓图片

from selenium import webdriver
from time import sleep
import win32api, os
import win32con, requests


def get_conan():
    num = 1
    # 循环120次
    for k in range(1, 121):
        driver = webdriver.Chrome("chromedriver.exe")
        driver.get("https://ac.qq.com/ComicView/index/id/623654/cid/%s" % k)
        driver.maximize_window()
        sleep(2)
        # 一直按下键
        for i in range(1000):
            win32api.keybd_event(40, 0, 0, 0)
            win32api.keybd_event(40, 0, win32con.KEYEVENTF_KEYUP, 0) 
            sleep(0.1)

        # 取出title在处理一下
        js1 = """return document.title"""
        title = driver.execute_script(js1)
        title1 = "第%s集" % num + title.split(" ")[1].split("-")[0]
        os.makedirs(title1)
        js = """
                var arr= [];for(var i=0;i<document.getElementsByClassName("loaded").length;i++){
               arr.push(document.getElementsByClassName("loaded")[i].src)};  return arr;
           """
        res_img = driver.execute_script(js)
        for index in range(len(res_img)):
            res = requests.get(res_img[index], timeout=50)
            with open(r"%s\%s.jpg" % (title1, index), "wb") as f:
                f.write(res.content)
        driver.quit()
        num += 1

        
get_conan()