艺图语图片下载

点击查看代码

import requests
import os


# 下图片功能
def down_img(url, zid):
    if not os.path.exists(zid):
        os.makedirs(zid)
    headers = {
        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36 Edg/129.0.0.0"
    }
    resp = requests.get(url, headers=headers)
    file_name = zid + '/' + url.split('/')[-1]
    with open(file_name, 'wb') as f:
        f.write(resp.content)
    print('下载成功')


def down_all(zid):
    url = 'https://www.yituyu.com/ajax_gallery/'
    headers = {
        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36 Edg/129.0.0.0"
    }

    data = {
        "zid": zid
    }
    r = requests.post(url, headers=headers, data=data)
    s = r.json()['data']['content']
    imgs = s.split(',')
    for img in imgs:
        img = f'https://img.yituyu.com/gallery/{zid}/' + img
        # print(img)
        title = img.split('/')[-1]
        print(title)
        down_img(img, zid)


def main():
    zid = input('请输入id:')
    down_all(zid)


main()