初次尝试Python 爬图片

import urllib.request
import os
 
 
def url_open(url):
    req = urllib.request.Request(url)
    req.add_header('User-Agent','Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 UBrowser/5.7.15702.19 Safari/537.36')
    #这里设置浏览器头
    response = urllib.request.urlopen(req)
    html = response.read()
    return html
 
 
 
def find_img(url):
    bianma = input("请输入网页编码:")
    #这里设置url编码
    html = url_open(url).decode(bianma)
    a = html.find('img src=')
    img_addrs = []
    while a != -1:
        b = html.find('.jpg',a,a+255)
        if b != -1:
            img_addrs.append(html[a+9:b+4])
        else:
            b = a+9
        a = html.find('img src=',b)
        #a b 需要按页面修改find
    for each in img_addrs:
        print(each)
    return img_addrs
 
def save_img(folder,img_addrs):
    for each in img_addrs:
        filename = each.split('/')[-1]
        with open(filename,'wb') as f:
            print('正确')
            img =url_open(each)
            f.write(img)
 
 
 
 
 
 
 
def download_mm(folder='图片'):
    folder = input('请输入目录:')
    try:
        os.mkdir(folder)
    except:
        print('该目录已经存在')
    os.chdir(folder)
    url=input('请输入要下载图片的网址')
    img_addrs=find_img(url)
    save_img(folder,img_addrs)
 
if __name__ == '__main__':
    download_mm()
FEATURED TAGS

FRIENDS
