X-Git-Url: https://code.delx.au/bg-scripts/blobdiff_plain/5f784db95af686ef4d6e7ec97937477010bf6fb8..94a833d3bd9534b5e8b1487c78ddf7ba8bc0d3a9:/wallchanger.py diff --git a/wallchanger.py b/wallchanger.py index b20c068..74be97b 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 -from logging import debug, info, warning +import commands, sys, os, os.path, time +import logging +try: + import PIL, PIL.Image +except ImportError: + PIL = None __all__ = ("init", "set_image") @@ -13,15 +17,19 @@ __all__ = ("init", "set_image") changers = [] def set_image(filename): - info("Setting image: %s" % filename) + logging.info("Setting image: %s", filename) for changer in changers: if not changer.set_image(filename): - warning("Failed to set background: wallchanger.set_image(%s), changer=%s" % (filename, changer)) + logging.warning("Failed to set background: wallchanger.set_image(%s), changer=%s", filename, changer) def init(*args, **kwargs): """Desktop Changer factory""" - debug("Testing for OSX (NonX11)") + 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: changers.append(OSXChanger(*args, **kwargs)) @@ -31,19 +39,19 @@ 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 commands.getstatusoutput("ps ax -o command|grep -q '^/.*X11 .* %s'" % os.environ['DISPLAY'])[0] != 0: # X11 is not running for this display return - debug("Testing for KDE") + logging.debug("Testing for KDE") if commands.getstatusoutput("xwininfo -name 'KDE Desktop'")[0] == 0: changers.append(KDEChanger(*args, **kwargs)) - debug("Testing for Gnome") - if commands.getstatusoutput("xwininfo -name 'gnome-session'")[0] == 0: + logging.debug("Testing for Gnome") + if commands.getstatusoutput("xwininfo -name 'gnome-settings-daemon'")[0] == 0: changers.append(GnomeChanger(*args, **kwargs)) - debug("Testing for WMaker") + logging.debug("Testing for WMaker") if commands.getstatusoutput("xlsclients | grep -qi wmaker")[0] == 0: changers.append(WMakerChanger(*args, **kwargs)) @@ -54,14 +62,51 @@ def init(*args, **kwargs): class BaseChanger(object): name = "undefined" def __init__(self, background_color='black', permanent=False, convert=False): - info('Determined the window manager is "%s"' % self.name) + logging.info('Determined the window manager is "%s"', self.name) self.background_color = background_color self.permanent = permanent self.convert = convert + try: + import subprocess + except ImportError: + self._runProgram = self._runProgram_command + else: + self._runProgram = self._runProgram_subprocess + + def _runProgram_subprocess(self, cmd): + import subprocess + return subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + + # A simple implementation of subprocess for python2.4 + def _runProgram_command(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/' @@ -80,14 +125,14 @@ class WMakerChanger(BaseChanger): 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] - debug("""Convert command: '"%s"'""" % '" "'.join(cmd)) - return output_name, subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + logging.debug("""Convert command: '"%s"'""", '" "'.join(cmd)) + return output_name, self._runProgram(cmd) def set_image(self, file): if self.convert: file, convert_status = self.convert_image_format(file) if convert_status: - debug('Convert failed') + logging.debug('Convert failed') cmd = ["wmsetbg", "-b", self.background_color, # Sets the background colour to be what the user specified "-S", # 'Smooth' (WTF?) @@ -99,8 +144,8 @@ class WMakerChanger(BaseChanger): if self.permanent: cmd += ["-u"] # update the wmaker database cmd += [file] - debug('''WMaker bgset command: "'%s'"''' % "' '".join(cmd)) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + logging.debug('''WMaker bgset command: "'%s'"''', "' '".join(cmd)) + return not self._runProgram(cmd) class OSXChanger(BaseChanger): name = "Mac OS X" @@ -109,7 +154,6 @@ class OSXChanger(BaseChanger): def __init__(self, *args, **kwargs): BaseChanger.__init__(self, *args, **kwargs) - self.fix_desktop_plist() def remove_old_image_cache(self): """Cleans up any old temp images""" @@ -126,12 +170,9 @@ class OSXChanger(BaseChanger): 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(file, format='PNG', extension='.png') except ImportError: - debug('Could not load PIL, going to try just copying the image') + 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) @@ -150,24 +191,68 @@ class OSXChanger(BaseChanger): # Store the plist again (Make sure we write it out atomically -- Don't want to break finder) desktop_plist.writeToFile_atomically_(self._DesktopPlistLocation, True) except ImportError: - debug('Could not import the Foundation module, you may have problems with dual screens') + logging.debug('Could not import the Foundation module, you may have problems with dual screens') def set_image(self, filename): + self.fix_desktop_plist() if self.convert: filename, ret = self.convert_image_format(filename) if not ret: - debug("Convert failed") + logging.debug("Convert failed") return False cmd = """osascript -e 'tell application "finder" to set desktop picture to posix file "%s"'""" % filename - debug(cmd) + logging.debug(cmd) return not commands.getstatusoutput(cmd)[0] +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 GnomeChanger(BaseChanger): name = "Gnome" def set_image(self, file): cmd = ['gconftool-2', '--type', 'string', '--set', '/desktop/gnome/background/picture_filename', file] - debug(cmd) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + logging.debug(cmd) + return not self._runProgram(cmd) class KDEChanger(BaseChanger): name = "KDE" @@ -182,15 +267,14 @@ class KDEChanger(BaseChanger): cmds.append(['dcop', 'kdesktop', 'KBackgroundIface', 'configure']) for cmd in cmds: - debug(cmd) - if subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() != 0: + logging.debug(cmd) + if self._runProgram(cmd) != 0: return False return True def main(filename): - import logging logging.basicConfig(level=logging.DEBUG, format="%(levelname)s: %(message)s") init() set_image(filename)