Python實現桌布下載與輪換

阿新 • • 發佈：2020-10-21

準備

下載安裝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",win32con.KEY_SET_VALUE)
  win32api.RegSetValueEx(regKey,"WallpaperStyle",win32con.REG_SZ,"2")
  win32api.RegSetValueEx(regKey,"TileWallpaper","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

知識點