forked from p15670423/monkey
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import ctypes
|
||
|
|
import shutil
|
||
|
|
import logging
|
||
|
|
from data import resource_path
|
||
|
|
|
||
|
|
__author__ = 'itamar'
|
||
|
|
|
||
|
|
LOG = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
SPI_SETDESKWALLPAPER = 20
|
||
|
|
SPIF_UPDATEINIFILE = 0x01
|
||
|
|
SPIF_SENDCHANGE = 0x02
|
||
|
|
|
||
|
|
class ChangeDesktopAction(object):
|
||
|
|
def __init__(self, desktop_image):
|
||
|
|
self._desktop_image = resource_path(desktop_image)
|
||
|
|
|
||
|
|
assert os.path.exists(self._desktop_image), "desktop image %s is missing" % (self._desktop_image, )
|
||
|
|
|
||
|
|
def do_action(self):
|
||
|
|
ctypes_desktop_img = ctypes.c_char_p(self._desktop_image)
|
||
|
|
|
||
|
|
# set the image
|
||
|
|
if not bool(ctypes.windll.user32.SystemParametersInfoA(SPI_SETDESKWALLPAPER, 0, ctypes_desktop_img, SPIF_SENDCHANGE | SPIF_UPDATEINIFILE)):
|
||
|
|
LOG.warn("Error setting desktop wallpaper image to '%s' (error %d)",
|
||
|
|
ctypes_desktop_img.value, ctypes.windll.kernel32.GetLastError())
|
||
|
|
else:
|
||
|
|
LOG.debug("Desktop wallpaper image is set to %r", ctypes_desktop_img.value)
|
||
|
|
|
||
|
|
def undo_action(self):
|
||
|
|
if not bool(ctypes.windll.user32.SystemParametersInfoA(SPI_SETDESKWALLPAPER, 0, "" , SPIF_SENDCHANGE | SPIF_UPDATEINIFILE)):
|
||
|
|
LOG.warn("Error resetting desktop wallpaper image (error %d)",
|
||
|
|
ctypes.windll.kernel32.GetLastError())
|
||
|
|
else:
|
||
|
|
LOG.debug("Desktop wallpaper image reset")
|