Python setup Windows desktop wallpaper implementation code
- 2020-04-02 09:46:31
- OfStack
# -*- coding: UTF-8 -*-
from __future__ import unicode_literals
import Image
import datetime
import win32gui,win32con,win32api
import re
from HttpWrapper import SendRequest
StoreFolder = "c:\dayImage"
def setWallpaperFromBMP(imagepath):
k = win32api.RegOpenKeyEx(win32con.HKEY_CURRENT_USER,"Control Panel\Desktop",0,win32con.KEY_SET_VALUE)
win32api.RegSetValueEx(k, "WallpaperStyle", 0, win32con.REG_SZ, "2") #2 Stretch fit desktop ,0 Desktop center
win32api.RegSetValueEx(k, "TileWallpaper", 0, win32con.REG_SZ, "0")
win32gui.SystemParametersInfo(win32con.SPI_SETDESKWALLPAPER,imagepath, 1+2)
def setWallPaper(imagePath):
"""
Given a path to an image, convert it to bmp and set it as wallpaper
"""
bmpImage = Image.open(imagePath)
newPath = StoreFolder + '\mywallpaper.bmp'
bmpImage.save(newPath, "BMP")
setWallpaperFromBMP(newPath)
def getPicture():
url = "http://photography.nationalgeographic.com/photography/photo-of-the-day/"
h = SendRequest(url)
if h.GetSource():
r = re.findall('<div class="download_link"><a href="(.*?)">Download',h.GetSource())
if r:
return SendRequest(r[0]).GetSource()
else:
print " There was an error parsing the image address. Check the regular expression for correctness "
return None
def setWallpaperOfToday():
img = getPicture()
if img:
path = StoreFolder + "\%s.jpg" % datetime.date.today()
f = open(path,"wb")
f.write(img)
f.close()
setWallPaper(path)
setWallpaperOfToday()
print 'Wallpaper set ok!'
The httpwrapper is an HTTP access wrapper that I wrote:
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
#-------------------------------------------------------------------------------
# Name: right http Encapsulation of access
#
# Author: qianlifeng
#
# Created: 10-02-2012
#-------------------------------------------------------------------------------
import base64
import urllib
import urllib2
import time
import re
import sys
class SendRequest:
"""
Page request enhancement class
SendRequest('http://xxx.com',data=dict, type='POST', auth='base',user='xxx', password='xxx')
"""
def __init__(self, url, data=None, method='GET', auth=None, user=None, password=None, cookie = None, **header):
"""
url: The request of url Cannot be null
date: Need to be post Must be a dictionary
method: Get or Post By default, Get
auth: 'base' or 'cookie'
user: Used for base The authenticated user name
password: Used for base Authenticated password
cookie: Attached to the request cookie , generally used for authentication after login
Other header information :
e.g. referer='www.sina.com.cn'
"""
self.url = url
self.data = data
self.method = method
self.auth = auth
self.user = user
self.password = password
self.cookie = cookie
if 'referer' in header:
self.referer = header[referer]
else:
self.referer = None
if 'user-agent' in header:
self.user_agent = header[user-agent]
else:
## self.user_agent = 'Mozilla/5.0 (Windows NT 5.1; rv:8.0) Gecko/20100101 Firefox/8.0'
self.user_agent = 'Mozilla/5.0 (iPhone; U; CPU iPhone OS 3_0 like Mac OS X; en-us) AppleWebKit/528.18 (KHTML, like Gecko) Version/4.0 Mobile/7A341 Safari/528.16'
self.__SetupRequest()
self.__SendRequest()
def __SetupRequest(self):
if self.url is None or self.url == '':
raise 'url Can't be empty !'
# Access mode Settings
if self.method.lower() == 'post':
self.Req = urllib2.Request(self.url, urllib.urlencode(self.data))
elif self.method.lower() == 'get':
if self.data == None:
self.Req = urllib2.Request(self.url)
else:
self.Req = urllib2.Request(self.url + '?' + urllib.urlencode(self.data))
# Set authentication information
if self.auth == 'base':
if self.user == None or self.password == None:
raise 'The user or password was not given!'
else:
auth_info = base64.encodestring(self.user + ':' + self.password).replace('n','')
auth_info = 'Basic ' + auth_info
self.Req.add_header("Authorization", auth_info)
elif self.auth == 'cookie':
if self.cookie == None:
raise 'The cookie was not given!'
else:
self.Req.add_header("Cookie", self.cookie)
if self.referer:
self.Req.add_header('referer', self.referer)
if self.user_agent:
self.Req.add_header('user-agent', self.user_agent)
def __SendRequest(self):
try:
self.Res = urllib2.urlopen(self.Req)
self.source = self.Res.read()
self.code = self.Res.getcode()
self.head_dict = self.Res.info().dict
self.Res.close()
except:
print "Error: HttpWrapper=>_SendRequest ", sys.exc_info()[1]
def GetResponseCode(self):
"""
Gets the status code returned by the server (200 Indicates success ,404 Web page does not exist )
"""
return self.code
def GetSource(self):
"""
Get the web source code, need to decode after use
"""
if "source" in dir(self):
return self.source
return u''
def GetHeaderInfo(self):
"""
u' Get the response header information '
"""
return self.head_dict
def GetCookie(self):
"""
Gets what the server returns Cookie , generally used for subsequent operations of login
"""
if 'set-cookie' in self.head_dict:
return self.head_dict['set-cookie']
else:
return None
def GetContentType(self):
"""
Get the return type
"""
if 'content-type' in self.head_dict:
return self.head_dict['content-type']
else:
return None
def GetCharset(self):
"""
Try to get the code for the page
If you don't get it back None
"""
contentType = self.GetContentType()
if contentType is not None:
index = contentType.find("charset")
if index > 0:
return contentType[index+8:]
return None
def GetExpiresTime(self):
"""
Get page expiration time
"""
if 'expires' in self.head_dict:
return self.head_dict['expires']
else:
return None
def GetServerName(self):
"""
Get the server name
"""
if 'server' in self.head_dict:
return self.head_dict['server']
else:
return None
__all__ = [SendRequest,]
if __name__ == '__main__':
b = SendRequest("http://www.baidu.com")
print b.GetSource()