python3爬取-alpha.wallhaven.cc 全站壁纸

Python admin 4个月前 (05-02) 342次浏览 0个评论

这是前面学习了一段python的时候,写的这个爬虫程序。刚刚在整理硬盘时翻到了,记录一下

# -*- coding:utf-8 -*-
import requests
import time
import os
import random
'''
下载wallhaven全站壁纸
'''
def download():   # 全站下载

    id = 1
    a = input('输入你要下载的壁纸数量:')

    while id <= int(a):
        with open('/wallpaper/' + str(id) + '.jpg', 'wb') as f:
            try:
                url = 'https://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-%s.jpg' % id
                pic = requests.get(url=url, headers=headers)
                print('正在下载第%s个文件,图片地址:%s' % (id, url))
                f.write(pic.content)
                id = id + 1
                print('休息5秒后继续执行。。')
                time.sleep(5)
            except Exception as e:
                print(repr(e))

def randows():     # 随机下载

    id = 1
    a = input('输入随机下载多少个壁纸:')
    while id <= int(a):
        with open('wallpaper/' + str(id) + '.jpg', 'wb') as f:
            try:
                shu = random.randint(1, 668088)
                url = 'https://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-%s.jpg' % shu
                pic = requests.get(url=url, headers=headers)
                print('正在下载第%s个文件,图片地址:%s' % (id, url))
                f.write(pic.content)
                id = id + 1
                print('休息一会')
                time.sleep(5)
            except Exception as e:
                print(repr(e))
    print('你要的小黄图壁纸全部下载完毕!')

def creat_file():

    filePath = 'wallpaper/'
    if not os.path.exists(filePath):  # 如果文件夹不存在,创建文件夹
        os.mkdir(filePath)
        print('成功创建文件夹!')

def kaishi():
    creat_file()

    print(
        '1.随机下载指定数量壁纸\n'
        '2.顺序下载壁纸\n'
        '3.退出程序'
    )
    a = input('输入你要的功能:')
    if a == '1':
        randows()
    elif a == '2':
        download()
    elif a == '3':
        quit()
    else:
        print('输入有误,请重新输入!')
        kaishi()


if __name__ == '__main__':
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.87 Safari/537.36'
    }
    kaishi()

小泉博客|本站的文章和资源来自互联网或者站长的原创,按照 CC BY -NC -SA 3.0 CN
协议发布和共享,转载或引用本站文章应遵循相同协议。如果有侵犯版权的资源请尽快联系站长,我们会在24h内删除有争议的资源。转载请注明原文链接:python3爬取-alpha.wallhaven.cc 全站壁纸
喜欢 (0)
发表我的评论
取消评论
表情 加粗 删除线 居中 斜体 签到