Python 抓url的图

想到可以用Selenium抓一下url内的图。

from selenium import webdriver
from time import sleep
from urllib import request

driver=webdriver.Chrome()
#这个函数接收URL和相关图片的img标签class属性，返回相关图片的URL list。
def get_pic_url(url,tag_class): 

    driver.get(url)
    pic_urls=[]

    for f in driver.find_elements_by_css_selector(tag_class):
        pic_urls.append(f.get_attribute('src'))

    return pic_urls

url=input('url')
tag_class=input('tag_class')
m=0
for i in get_pic_url(url,tag_class):
    m=m+1
    request.urlretrieve(i, 'F:\\pics\\'+str(m)+'.jpg')
driver.quit()