Python開發專屬壁紙下載與輪換程序

本人對於壁紙一直偏佛系，不愛特意去找一堆壁紙。因此用Python簡單地搞了一個自動下載壁紙，定時隨機輪換的功能來自娛自樂，順便分享給大家。
原文鏈接：https://www.codernice.top/articles/9f79c65e.html

准備

下載安裝Python3

官網下載即可，選擇合適的版本：https://www.python.org/downloads/
安裝一直下一步即可，記得勾選添加到環境變量。

安裝pypiwin32

執行設置壁紙操作需要調用Windows系統的API，需要安裝pypiwin32，控制台執行如下命令：

pip install pypiwin32

工作原理

兩個線程，一個用來下載壁紙，一個用來輪換壁紙。每個線程內部均做定時處理，通過在配置文件中配置的等待時間來實現定時執行的功能。

壁紙下載線程

簡易的爬蟲工具，查詢目標壁紙網站，過濾出有效連接，逐個遍歷下載壁紙。

壁紙輪換線程

遍歷存儲壁紙的目錄，隨機選擇一張壁紙路徑，並使用pypiwin32庫設置壁紙。

部分代碼

線程創建與配置文件讀取

def main():
    # 加載現有配置文件
    conf = configparser.ConfigParser()
    # 讀取配置文件
    conf.read("conf.ini")
    # 讀取配置項目
    search = conf.get('config', 'search')
    max_page = conf.getint('config','max_page')
    loop = conf.getint('config','loop')
    download = conf.getint('config','download')
    
    # 壁紙輪換線程
    t1 = Thread(target=loop_wallpaper,args=(loop,))
    t1.start()

    # 壁紙下載線程
    t2 = Thread(target=download_wallpaper,args=(max_page,search,download))
    t2.start()

遍歷圖片隨機設置壁紙

def searchImage():
    # 獲取壁紙路徑
    imagePath = os.path.abspath(os.curdir) + '\images'
    if not os.path.exists(imagePath):
        os.makedirs(imagePath)
    # 獲取路徑下文件
    files = os.listdir(imagePath)
    # 隨機生成壁紙索引
    if len(files) == 0:
        return
    index = random.randint(0,len(files)-1)
    for i in range(0,len(files)):
        path = os.path.join(imagePath,files[i])
        # if os.path.isfile(path):
        if i == index:
            if path.endswith(".jpg") or path.endswith(".bmp"):
                setWallPaper(path)
            else:
                print("不支持該類型文件")

設置壁紙

def setWallPaper(pic):
    # open register
    regKey = win32api.RegOpenKeyEx(win32con.HKEY_CURRENT_USER,"Control Panel\\Desktop",0,win32con.KEY_SET_VALUE)
    win32api.RegSetValueEx(regKey,"WallpaperStyle", 0, win32con.REG_SZ, "2")
    win32api.RegSetValueEx(regKey, "TileWallpaper", 0, win32con.REG_SZ, "0")
    # refresh screen
    win32gui.SystemParametersInfo(win32con.SPI_SETDESKWALLPAPER,pic, win32con.SPIF_SENDWININICHANGE)

壁紙查詢鏈接過濾

def crawl(page,search):
    # 1\. 進入壁紙查詢頁面
    hub_url = 'https://wallhaven.cc/search?q=' + search + '&sorting=random&page=' + str(page)
    res = requests.get(hub_url)
    html = res.text

    # 2\. 獲取鏈接
    ## 2.1 匹配 'href'
    links = re.findall(r'href=[\'"]?(.*?)[\'"\s]', html)
    print('find links:', len(links))
    news_links = []
    ## 2.2 過濾需要的鏈接
    for link in links:
        if not link.startswith('https://wallhaven.cc/w/'):
            continue
        news_links.append(link)
    print('find news links:', len(news_links))
    # 3\. 遍歷有效鏈接進入詳情
    for link in news_links:
        html = requests.get(link).text
        fing_pic_url(link, html)
    print('下載成功，當前頁碼:'+str(page));

圖片下載

def urllib_download(url):
    #設置目錄下載圖片
    robot = './images/'
    file_name = url.split('/')[-1]
    path = robot + file_name
    if os.path.exists(path):
        print('文件已經存在')
    else:
        url=url.replace('\\','')
        print(url)
        r=requests.get(url,timeout=60)
        r.raise_for_status()
        r.encoding=r.apparent_encoding
        print('准備下載')
        if not os.path.exists(robot):
            os.makedirs(robot)
        with open(path,'wb') as f:
            f.write(r.content)
            f.close()
            print(path+' 文件保存成功')

import部分

import re
import time
import requests
import os
import configparser
import random
import tldextract #pip install tldextract
import win32api, win32gui, win32con
from threading import Thread

完整代碼請查看GitHub：https://github.com/codernice/wallpaper

知識點

threading：多線程，這里用來創建壁紙下載和壁紙輪換兩個線程。
requests：這里用get獲取頁面，並獲取最終的壁紙鏈接
pypiwin32：訪問windows系統API的庫，這里用來設置壁紙。
configparser：配置文件操作，用來讀取線程等待時間和一些下載配置。
os：文件操作，這里用來存儲文件，遍歷文件，獲取路徑等。