Selenium crawls Baidu pictures

# coding=utf-8
"""  obtain 10 Baidu pictures  """

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
import time, requests


def download_img(kw):
    #  Open the browser 
    browser = webdriver.Chrome()
    time.sleep(3)

    #  The other party visits the corresponding web page 
    url = r'https://image.baidu.com/'
    browser.get(url)
    time.sleep(3)

    #  Enter the corresponding... In the input box key words
    keyword = browser.find_element_by_id('kw')
    keyword.send_keys(kw)  #  Enter key 
    keyword.send_keys(Keys.ENTER)  # enter
    time.sleep(3)

    #  Set the picture size 
    size = browser.find_element_by_id('sizeFilter')
    big_size = browser.find_element_by_xpath('/html/body/div[1]/div[4]/div[2]/div/div[2]/div/div[1]')
    ActionChains(browser).click(size).move_to_element(big_size).click().perform()

    #  Click on the first picture 
    first_pic = browser.find_element_by_xpath('//*[@id="imgid"]/div/ul/li[1]/div/a/img')
    ActionChains(browser).click(first_pic).perform()  #  call perform() When the method is used , The events in the queue will execute in turn .
    time.sleep(5)
    #  Switch to new window in 
    browser.switch_to.window(browser.window_handles[1])
    for i in range(20):
        #  Get photo 
        pic = browser.find_element_by_xpath('//*[@id="currentImg"]')
        src = pic.get_attribute('src')
        r = requests.get(src)
        title = browser.find_element_by_class_name('pic-title')  #  Output picture title at the same time 
        print(i, ' ', title.text)
        if r.status_code == 200:
            #  Save the picture to a file 
            file_name = r'D:\1. learning\python_web\pachong\img\{}.jpg'.format(i)
            with open(file_name, 'wb') as f:
                f.write(r.content)  #  Use of words r.text /  Other formats use  r.content
            #  Switch to the next picture 
            next_btn = browser.find_element_by_class_name('img-next')
            next_btn.click()
            time.sleep(5)
    return


if __name__ == '__main__':
    download_img(' Mickey Mouse ')