python 实现桌面壁纸自动更换

学了python大概两周了，今天做了个小程序，感觉还比较实用。在此记录一下

程序介绍

功能介绍:每隔30分钟，随机更换桌面。桌面资源是在zol网，自动爬取的热门图片


开机自启设置：发送changeBg.exe快捷方式到 C:\用户\用户名\AppData\Roaming\Microsoft\Windows\「开始」菜单\程序\启动文件夹


如果遇到不喜欢的图片，想要立即更换，则需要先杀死程序，再重新启动。程序每次启动时，都会立即更换一张背景图。


关闭程序：本程序是后台程序，不提供用户界面。所以关闭需要打开任务管理器，在任务管理器中找到changeBg的任务，并结束。


人生就像这个壁纸小程序，你永远也不知道下一张的壁纸长什么样。无意间的才叫惊喜

程序大致思路 ：

爬取zol网上的热门图片链接,并存到数据库。每隔三十分钟获取图片链接，下载图片并设置背景图片。

刚开始学python，类的运用还不时很熟练，所以用文件的形式来编写

main 文件 负责各个模块的调用，主要逻辑

import getBgUrls
import fileUtils
import threading
import SetBg
import random
import db
import time

def setBgUrl():
    #睡五秒钟，等待另一个线程爬取图片utl 这样可以运行程序，立即更换桌面
    time.sleep(5)
    print('setBgUrl has executed')
    pic_urls = getBgUrls.pic_urls
    if len(pic_urls) > 0:
        abs_path = fileUtils.save_file(random.choice(pic_urls))
        SetBg.set_wallpaper(abs_path)
    #每小时换一次
    threading.Timer(60*30,setBgUrl).start()


if __name__ == '__main__':
    times =  db.getStartTimes()
    #第一次使用本程序 调用存储到数据库的爬虫
    if times == 0 :
        t1 = threading.Thread(target=getBgUrls.get_pic_urls_to_db).start()
    elif times <= 10:
        #如果数据库的数据url长度为0，证明中途被打断，没有插入成功，需要重新开始
        urls = db.getUrls()
        if len(urls) == 0:
            t1 = threading.Thread(target=getBgUrls.get_pic_urls_to_db).start()
        else:
            # 从数据库获取url
            getBgUrls.pic_urls = urls
    else :
        db.removeDb()
        fileUtils.removeAllPic()
        t1 = threading.Thread(target=getBgUrls.get_pic_urls_to_db).start()
    setBgUrl()

db文件，负责数据库的操作。数据库用的是python自带的sqlite，还挺好用的

import sqlite3
import os
print('db has import')
#程序每使用十次，就重新爬取数据
file_path = r'D:\deskPicc'
if not os.path.exists(file_path):
    os.mkdir(file_path)
conn_main = sqlite3.connect(r'D:\deskPicc\background.db')


cursor = conn_main.execute(r'SELECT count(*) FROM sqlite_master WHERE type="table" AND name="t_config"')


for i in cursor:
    if i[0] == 0:
        #第一次运行程序，创建数据库config表,urls表
        conn_main.execute(r'create table t_config(start_times int)')
        conn_main.execute(r"insert into t_config('start_times') values(0)") #初始化为0
        conn_main.execute(r'create table t_urls(url varchar(200))')
        conn_main.commit()


conn_main.close()


# 拿到数据库链接，因为规定数据库链接和操作必须在统一线程
def getConnection():
    conn = sqlite3.connect(r'D:\deskPicc\background.db')
    return conn


#返回程序使用的次数
def getStartTimes():
    conn = getConnection()
    cursor = conn.execute(r"select * from t_config")
    for i in cursor:
        times = i[0]
        times = int(times) + 1
        sql = "update t_config set start_times = {}".format(str(times))
        print(sql)
        conn.execute(sql)
        conn.commit()
    conn.close()
    return times - 1


#插入url数据
def insertInto(urls):
    conn = getConnection()
    for i in urls:
        sql = "insert into t_urls('url') values('{}')".format(i)
        conn.execute(sql)
        conn.commit()
        print(sql)
    conn.close()


def getUrls():
    conn = getConnection()
    sql = 'select * from t_urls'
    cursor = conn.execute(sql)
    urls = [i[0] for i in cursor]
    print('getUrls from db : ')
    print(urls)


    conn.commit()
    conn.close()
    return urls


def removeDb():
    conn = getConnection()
    sql1 = 'drop table t_config'
    sql2 = 'drop table t_urls'
    conn.execute(sql1)
    conn.execute(sql2)


    conn.commit()
    conn.close()

getBgUrls文件主要负责解析网站，获取图片链接的url

import Http
from win32api import GetSystemMetrics
import db
pic_urls = []  # 存储所有的图片url
#解析网站，获取图片url
def get_pic_urls_to_db():
    global pic_urls
    base_url = r'http://desk.zol.com.cn'
    hot_url = r'http://desk.zol.com.cn/pc/hot_1.html' #所有，按下载量排序
    #系统分辨率
    width = GetSystemMetrics(0)
    height = GetSystemMetrics(1)
    width_height = str(width)+'x'+str(height)

    soup = Http.visitUrl(hot_url)
    for i in soup.select(r'li.photo-list-padding > a'):
        url = base_url + i.get('href')
        soup = Http.visitUrl(url)
        for tag_a in soup.select(r'#showImg > li > a'):
            # 找到适合自己电脑分辨率的图片链接
            soup = Http.visitUrl(base_url + tag_a.get('href'))
            for i in soup.select('#tagfbl > a'):
                if width_height in i.get('href'):
                    image_html = base_url + i.get('href')
                    #print('image_html : ',image_html)
                    soup = Http.visitUrl(image_html)
                    tag_imgs = soup.select('body > img')
                    if len(tag_imgs) > 0:
                        pic_urls.append(tag_imgs[0].get('src'))
                        print('has get image url',tag_imgs[0].get('src'))

    db.insertInto(pic_urls)

fileUtils文件提供把图片保存到本地的方法

import Http
import os

file_path = r'D:\deskPicc'
file_pathes = []

def save_file(url):
    if not os.path.exists(file_path):
        os.mkdir(file_path)
    os.chdir(file_path)
    response = Http.visitUrl_response(url)
    content = response.content
    #截取url
    file_name = url.split(r'/')[-1]
    with open(file_name,'wb') as f:
        if not os.path.exists(r'/'+file_name):
            f.write(content)
            abs_path = file_path+'\\'+ file_name
            file_pathes.append(abs_path)
            print(abs_path,'has saved as a file')
            return abs_path
        else:
            print(file_name,'hax exist')

def removeAllPic():
    for i in os.listdir(file_path):
        if 'background.db' in i:
            continue
        else :
            abs_path = file_path + '\\' + i
            print('remove : ', abs_path)
            os.remove(abs_path)

setBg文件提供设置桌面背景的方法

import win32api, win32con, win32gui

def set_wallpaper(img_path):
    # 打开指定注册表路径
    reg_key = win32api.RegOpenKeyEx(win32con.HKEY_CURRENT_USER, "Control Panel\\Desktop", 0, win32con.KEY_SET_VALUE)
    # 最后的参数:2拉伸,0居中,6适应,10填充,0平铺
    win32api.RegSetValueEx(reg_key, "WallpaperStyle", 0, win32con.REG_SZ, "2")
    # 最后的参数:1表示平铺,拉伸居中等都是0
    win32api.RegSetValueEx(reg_key, "TileWallpaper", 0, win32con.REG_SZ, "0")
    # 刷新桌面
    win32gui.SystemParametersInfo(win32con.SPI_SETDESKWALLPAPER, img_path, win32con.SPIF_SENDWININICHANGE)

http文件提供访问url的方法

from bs4 import BeautifulSoup
import requests
import time

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/59.0.3071.115 Safari/537.36'
}
#访问url，返回soup 方便控制访问频率，避免输入验证码
def visitUrl(url):
    wb_data = requests.get(url, headers=headers)
    #print('webEncoding is ',wb_data.encoding)
   # wb_data.encoding = 'gbk'
   # print('webEncoding has set gbk')
    soup = BeautifulSoup(wb_data.text, 'lxml')
    time.sleep(0.2)
    return soup

def visitUrl_response(url):
    return requests.get(url, headers=headers)

本次程序命名有很多不规范，有时间再改

源码及地址 http://download.csdn.net/detail/felixzh123/9910229