Python 下载贴吧图片到指定地址

#coding=utf-8
from urllib import request
import urllib
import re
import chardet

def mkdir(path):
    # 引入模块
    import os
    # 去除首位空格
    path = path.strip()
    # 去除尾部 \ 符号
    path = path.rstrip("\\")
    # 判断路径是否存在
    # 存在     True
    # 不存在   False
    isExists = os.path.exists(path)
    # 判断结果
    if not isExists:
        # 如果不存在则创建目录
        # 创建目录操作函数
        os.makedirs(path)
        print
        path + ' 创建成功'
        return True
    else:
        # 如果目录存在则不创建,并提示目录已存在
        print
        path + ' 目录已存在'
        return False
def getHtml(url):
    page = urllib.request.urlopen(url)
    html = page.read()
    return html
def getImg(html):
    reg = r'src="(.+?\.jpg)" pic_ext'
    imgre = re.compile(reg)
    encode_type = chardet.detect(html)
    html = html.decode(encode_type['encoding'])
    imglist = re.findall(imgre,html)
    x = 0
    for imgurl in imglist:
        urllib.request.urlretrieve(imgurl,r'C:\Users\41174\AppData\Local\Temp\change.py\image\%s.jpg' % x)
        x+=1
    return imglist

# 定义要创建的目录
mkpath = r"C:\Users\41174\AppData\Local\Temp\change.py\image"
# 调用函数
mkdir(mkpath)
html = getHtml("https://tieba.baidu.com/p/2460150866?red_tag=1683736448")
print(getImg(html))

猜你喜欢

转载自blog.csdn.net/weixin_42323337/article/details/83931927