python3必应壁纸爬虫练手

我不是女神ヾ 2024-04-19 16:28 79阅读 0赞

修复一个必应壁纸下载工具  
资源站和之前同学一样, bing.ioliu.cn  
打包好可以直接运行的版本在链接里  
主要是练手 有机会再做更新了:)

老规矩，  
先pip install 这几个第三方库

import re
    import os
    import requests
    from time import sleep

全部源码如下：

import re
    import os
    import requests
    from time import sleep
    
    headers = {
        "User-Agent": ("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:64.0) "
                       "Gecko/20100101 Firefox/64.0")
    }
    
    def get_index(resolution, index=1):
        url = f"https://bing.ioliu.cn/ranking?p={index}"
        res = requests.get(url, headers=headers)
        urls = re.findall('pic=(.*?)\\.jpg', res.text)
        _old_resolution = urls[1].split("_")[-1]
        return {url.split("/")[-1].replace(_old_resolution, resolution): url.replace(_old_resolution, resolution) + ".jpg"
                for url in urls}
    
    def download_pic(pics):
        if os.path.exists('必应壁纸'):
            pass
        else:
            os.mkdir('必应壁纸')
            print('目录创建成功')
        try:
            for pic_name, pic_url in pics.items():
                res = requests.get(pic_url, headers=headers)
                with open(f"必应壁纸\\{pic_name}.jpg", mode="wb") as f:
                    f.write(res.content)
                print(f"{pic_name} 下载完成")
        except Exception as e:
            print("下载出错", e)
    
    def input_index():
        print("必应壁纸下载工具, 本工具未经资源站授权.")
        print("仅做学习和交流之用, 随时有可能停止维护.")
        print("目前资源站收容页数为87,当前仅提供1920x1080分辨率下载")
        while True:
            sleep(0.1)
            index = input("请输入要下载的页数(Max=87):")
            try:
                if index == "Q":
                    exit()
                index = 87 if int(index) > 87 else int(index)
                return index
            except ValueError:
                print("请输入数字, 或输入Q退出!")
    
    def main():
        index = input_index()
        i = 1
        while i <= index:
            print(f"当前第{i}页,共需要下载{index}页")
            pics = get_index("1920x1080", i)
            download_pic(pics)
            i += 1
        print("下载完成,将在3秒后关闭...")
        sleep(1)
        print("2")
        sleep(1)
        print("1")
        sleep(1)
        print("0")
    
    if __name__ == '__main__':
        main()