X-Git-Url: https://code.delx.au/bg-scripts/blobdiff_plain/87dee88c4f1702f501f55c8a97da4a3d3e377303..7d7e922b60e1f1885faf676651564c6c66d5dd91:/wallchanger.py diff --git a/wallchanger.py b/wallchanger.py index 5982621..8230f48 100755 --- a/wallchanger.py +++ b/wallchanger.py @@ -4,8 +4,12 @@ # Licensed for distribution under the GPL version 2, check COPYING for details # This is a cross platform/cross window manager way to change your wallpaper -import commands, sys, os, os.path, subprocess, time +import commands, sys, os, os.path, time import logging +try: + import PIL, PIL.Image +except ImportError: + PIL = None __all__ = ("init", "set_image") @@ -18,11 +22,18 @@ def set_image(filename): if not changer.set_image(filename): logging.warning("Failed to set background: wallchanger.set_image(%s), changer=%s", filename, changer) +def check_cmd(cmd): + return commands.getstatusoutput(cmd)[0] == 0 + def init(*args, **kwargs): """Desktop Changer factory""" + if sys.platform == "win32": + changers.append(WIN32Changer(*args, **kwargs)) + return + logging.debug("Testing for OSX (NonX11)") - if commands.getstatusoutput("ps ax -o command -c|grep -q WindowServer")[0] == 0: + if check_cmd("ps ax -o command -c|grep -q WindowServer"): changers.append(OSXChanger(*args, **kwargs)) if 'DISPLAY' not in os.environ or os.environ['DISPLAY'].startswith('/tmp/launch'): @@ -31,20 +42,23 @@ def init(*args, **kwargs): else: if os.uname()[0] == 'Darwin': # Try to detect if the X11 server is running on OSX - if commands.getstatusoutput("ps ax -o command|grep -q '/.*X11 .* %s'" % os.environ['DISPLAY'])[0] != 0: + if check_cmd("ps ax -o command|grep -q '^/.*X11 .* %s'" % os.environ['DISPLAY']): # X11 is not running for this display return logging.debug("Testing for KDE") - if commands.getstatusoutput("xwininfo -name 'KDE Desktop'")[0] == 0: + if check_cmd("xwininfo -name 'KDE Desktop'"): changers.append(KDEChanger(*args, **kwargs)) logging.debug("Testing for Gnome") - if commands.getstatusoutput("xwininfo -name 'gnome-session'")[0] == 0: - changers.append(GnomeChanger(*args, **kwargs)) + if check_cmd("xwininfo -name 'gnome-settings-daemon'"): + if check_cmd("gsettings get org.gnome.desktop.background picture-uri"): + changers.append(Gnome3Changer(*args, **kwargs)) + else: + changers.append(Gnome2Changer(*args, **kwargs)) logging.debug("Testing for WMaker") - if commands.getstatusoutput("xlsclients | grep -qi wmaker")[0] == 0: + if check_cmd("xlsclients | grep -qi wmaker"): changers.append(WMakerChanger(*args, **kwargs)) if len(changers) == 0: @@ -59,9 +73,41 @@ class BaseChanger(object): self.permanent = permanent self.convert = convert + try: + def _exec_cmd(self, cmd): + import subprocess + return subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + + except ImportError: + # A simple implementation of subprocess for python2.4 + def _exec_cmd(self, cmd): + """Runs a program given in cmd""" + return os.spawnvp(os.P_WAIT, cmd[0], cmd) + def set_image(self, filename): raise NotImplementedError() + def convert_image_format(self, filename, format='BMP', allowAlpha=False, extension='.bmp'): + """Convert the image to another format, and store it in a local place""" + if not os.path.exists(filename): + logger.warn('The input file "%s" does not exist, so it will not be converted', filename) + return filename, False + if PIL is None: + logger.warn('PIL could not be found, not converting image format') + return filename, False + + self.remove_old_image_cache() + output_name = os.path.join(self._ConvertedWallpaperLocation, '%s%s' % (time.time(), extension)) + img = PIL.Image.open(filename) + + # Remove the alpha channel if the user doens't want it + if not allowAlpha and img.mode == 'RGBA': + img = img.convert('RGB') + img.save(output_name, format) + + return output_name, True + + class WMakerChanger(BaseChanger): name = "WindowMaker" _ConvertedWallpaperLocation = '/tmp/wallpapers_wmaker/' @@ -75,17 +121,17 @@ class WMakerChanger(BaseChanger): for dirname in dirnames: os.unlink(os.path.join(fullpath, dirname)) - def convert_image_format(self, file): + def convert_image_format(self, filename): """Convert the image to a png, and store it in a local place""" self.remove_old_image_cache() output_name = os.path.join(self._ConvertedWallpaperLocation, '%s.png' % time.time()) - cmd = ["convert", '-resize', '1280', '-gravity', 'Center', '-crop', '1280x800+0+0', file, output_name] + cmd = ["convert", '-resize', '1280', '-gravity', 'Center', '-crop', '1280x800+0+0', filename, output_name] logging.debug("""Convert command: '"%s"'""", '" "'.join(cmd)) - return output_name, subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + return output_name, self._exec_cmd(cmd) - def set_image(self, file): + def set_image(self, filename): if self.convert: - file, convert_status = self.convert_image_format(file) + filename, convert_status = self.convert_image_format(filename) if convert_status: logging.debug('Convert failed') cmd = ["wmsetbg", @@ -98,9 +144,9 @@ class WMakerChanger(BaseChanger): ] if self.permanent: cmd += ["-u"] # update the wmaker database - cmd += [file] + cmd += [filename] logging.debug('''WMaker bgset command: "'%s'"''', "' '".join(cmd)) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + return not self._exec_cmd(cmd) class OSXChanger(BaseChanger): name = "Mac OS X" @@ -120,20 +166,17 @@ class OSXChanger(BaseChanger): for dirname in dirnames: os.unlink(os.path.join(fullpath, dirname)) - def convert_image_format(self, file): + def convert_image_format(self, filename): """Convert the image to a png, and store it in a local place""" self.remove_old_image_cache() output_name = os.path.join(self._ConvertedWallpaperLocation, '%s.png' % time.time()) try: - import PIL, PIL.Image - img = PIL.Image.open(file) - img.save(output_name, "PNG") - return output_name, True + return super(OSXChanger, self).convert_image_format(filename, format='PNG', extension='.png') except ImportError: logging.debug('Could not load PIL, going to try just copying the image') import shutil - output_name = os.path.join(self._ConvertedWallpaperLocation, os.path.basename(file)) - shutil.copyfile(file, output_name) + output_name = os.path.join(self._ConvertedWallpaperLocation, os.path.basename(filename)) + shutil.copyfile(filename, output_name) return output_name, True def fix_desktop_plist(self): @@ -162,20 +205,70 @@ class OSXChanger(BaseChanger): logging.debug(cmd) return not commands.getstatusoutput(cmd)[0] -class GnomeChanger(BaseChanger): +class WIN32Changer(BaseChanger): + name = "Windows" + _ConvertedWallpaperLocation = os.path.join(os.environ.get('APPDATA', os.path.expanduser('~')), 'wallchanger') + + def __init__(self, *args, **kwargs): + BaseChanger.__init__(self, *args, **kwargs) + if not self.convert: + logging.warn('Running on windows, but convert is not set') + + def remove_old_image_cache(self): + """Cleans up any old temp images""" + if not os.path.isdir(self._ConvertedWallpaperLocation): + os.mkdir(self._ConvertedWallpaperLocation) + for fullpath, filenames, dirnames in os.walk(self._ConvertedWallpaperLocation, topdown=False): + for filename in filenames: + os.unlink(os.path.join(fullpath, filename)) + for dirname in dirnames: + os.unlink(os.path.join(fullpath, dirname)) + + def set_image(self, filename): + import ctypes + user32 = ctypes.windll.user32 + + # Taken from the Platform SDK + SPI_SETDESKWALLPAPER = 20 + SPIF_SENDWININICHANGE = 2 + + if self.convert: + filename, ret = self.convert_image_format(filename) + if not ret: + logging.debug("Convert failed") + return False + + # Parameters for SystemParametersInfoA are: + # (UINT uiAction, UINT uiParam, PVOID pvParam, UINT fWinIni) + user32.SystemParametersInfoA( + SPI_SETDESKWALLPAPER, + 0, + filename, + SPIF_SENDWININICHANGE, + ) + return True + +class Gnome2Changer(BaseChanger): name = "Gnome" - def set_image(self, file): - cmd = ['gconftool-2', '--type', 'string', '--set', '/desktop/gnome/background/picture_filename', file] + def set_image(self, filename): + cmd = ['gconftool-2', '--type', 'string', '--set', '/desktop/gnome/background/picture_filename', filename] logging.debug(cmd) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + return not self._exec_cmd(cmd) + +class Gnome3Changer(BaseChanger): + name = "Gnome3" + def set_image(self, filename): + cmd = ['gsettings', 'set', 'org.gnome.desktop.background', 'picture-uri', 'file://'+filename] + logging.debug(cmd) + return not self._exec_cmd(cmd) class KDEChanger(BaseChanger): name = "KDE" - def set_image(self, file): + def set_image(self, filename): cmds = [] for group in ('Desktop0', 'Desktop0Screen0'): base = ['kwriteconfig', '--file', 'kdesktoprc', '--group', group, '--key'] - cmds.append(base + ['Wallpaper', file]) + cmds.append(base + ['Wallpaper', filename]) cmds.append(base + ['UseSHM', '--type', 'bool', 'true']) cmds.append(base + ['WallpaperMode', 'ScaleAndCrop']) cmds.append(base + ['MultiWallpaperMode', 'NoMulti']) @@ -183,7 +276,7 @@ class KDEChanger(BaseChanger): cmds.append(['dcop', 'kdesktop', 'KBackgroundIface', 'configure']) for cmd in cmds: logging.debug(cmd) - if subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() != 0: + if self._exec_cmd(cmd) != 0: return False return True