自动更新必应Bing壁纸的脚本

1. # coding: utf-8
2. import requests
3. import time
4. import os
5. import random
7. # 添加请求头
8. g_headers = {
9.     'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
10.                   '(KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36',
11. }
14. def get_bing_image_urls(num):
15.     """
16.     从cn.bing.com下载每日的桌面壁纸
17.     :param num: 最多获取壁纸数
18.     :return: 返回获取的壁纸地址列表
19.     """
20.     # start_url = 'https://cn.bing.com/HPImageArchive.aspx?format=js&idx={}&n=1&nc=1622612516247&pid=hp'
21.     start_url = 'https://cn.bing.com/HPImageArchive.aspx?format=js&idx={}&n=1'
22.     imgs = []
23.     page = 1
24.     while len(imgs) < num and page < num*2:
25.         page += 1
26.         page_url = start_url.format(page)
27.         try:
28.             r = requests.get(page_url, headers=g_headers).json()
29.             img = 'https://cn.bing.com' + r['images'][0]['url']
30.             if imgs.count(img) <= 0:
31.                 imgs.append(img)
32.             else:
33.                 break
34.         except:
35.             pass
36.     return imgs
39. def download_image(url, savepath):
40.     """
41.     根据图片地址下载图片至指定路径
42.     :param url: 图片url地址
43.     :param savepath: 图片保存路径
44.     :return: 下载是否成功
45.     """
46.     try:
47.         r = requests.get(url, headers=g_headers, stream=True)
48.         if r.status_code != 200:
49.             return False
50.         open(savepath, 'wb').write(r.content)
51.         return True
52.     except:
53.         return False
56. def get_image_id(url):
57.     """
58.     从URL地址获取图片的参数id信息
59.     :param url: bing图片url地址
60.     :return: 图片id
61.     """
62.     try:
63.         paras = url[url.find('?')+1:].split('&')
64.         for p in paras:
65.             kv = p.split('=')
66.             if kv[0] == 'id':
67.                 return kv[1].strip()
68.     except:
69.         pass
70.     return url
73. def set_desktop_image(img):
74.     """
75.     将图片设置为桌面壁纸
76.     :param img: 图片
77.     :return:
78.     """
79.     import win32gui
80.     import win32con
81.     import win32api
82.     key = win32api.RegOpenKeyEx(win32con.HKEY_CURRENT_USER,
83.                                 "Control Panel\\Desktop", 0, win32con.KEY_SET_VALUE)
84.     win32api.RegSetValueEx(key, "WallpaperStyle", 0, win32con.REG_SZ, "2")
85.     # 2拉伸适应桌面,0桌面居中
86.     win32api.RegSetValueEx(key, "TileWallpaper", 0, win32con.REG_SZ, "0")
87.     win32gui.SystemParametersInfo(win32con.SPI_SETDESKWALLPAPER, os.path.abspath(img), 1 + 2)
90. def auto_change_desktop_image(img_dir, order='rand'):
91.     """
92.     根据指定的壁纸目录，定时更新桌面壁纸，当前日期变化后会退出
93.     :param img_dir: 壁纸目录
94.     :param order: 图片更新顺序，默认为顺序，rand为随机
95.     :return:
96.     """
97.     files = os.listdir(img_dir)
98.     images = []
99.     for file in files:
100.         temp = os.path.join(img_dir, file)
101.         if os.path.isfile(temp):
102.             images.append(temp)
103.     index = 1
104.     if order == 'rand':
105.         index = random.randint(0, len(images) - 1)
106.     index = index % len(images)
107.     set_desktop_image(images[index])
108.     print(time.time(), ' 已设置桌面壁纸为:', images[index])
110. if __name__ == '__main__':
111.     # 获取每日壁纸
112.     urls = get_bing_image_urls(20)
113.     save_dir = './' + time.strftime('%Y%m%d', time.localtime(time.time()))
114.     if not os.path.exists(save_dir):
115.         os.makedirs(save_dir)
116.     for url in urls:
117.         name = get_image_id(url)
118.         save_path = save_dir + '/' + name
119.         print('下载', url, '成功' if download_image(url, save_path) else '失败')
120.     # 自动更新壁纸
121.     auto_change_desktop_image(save_dir, 'rand')