X-Git-Url: https://code.delx.au/bg-scripts/blobdiff_plain/5f9578e04b4023f580643c388cfa7a3f2baa7990..f74c8d7f40c4db696091ad7cad1705c4968a5d57:/wallchanger.py diff --git a/wallchanger.py b/wallchanger.py index 6509d8f..40cac72 100755 --- a/wallchanger.py +++ b/wallchanger.py @@ -1,11 +1,16 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 + # Copyright 2008 Greg Darke # Copyright 2008 James Bunton # 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 subprocess, sys, os, os.path, time import logging +try: + import PIL, PIL.Image +except ImportError: + PIL = None __all__ = ("init", "set_image") @@ -13,194 +18,274 @@ __all__ = ("init", "set_image") changers = [] def set_image(filename): - logging.info("Setting image: %s", filename) - for changer in changers: - if not changer.set_image(filename): - logging.warning("Failed to set background: wallchanger.set_image(%s), changer=%s", filename, changer) + logging.info("Setting image: %s", filename) + for changer in changers: + if not changer.set_image(filename): + logging.warning("Failed to set background: wallchanger.set_image(%s), changer=%s", filename, changer) + +def get_displays(): + for line in subprocess.check_output("xrandr").decode("utf-8").split("\n"): + tokens = line.split() + if len(tokens) < 2 or tokens[1] != "connected": + continue + yield tokens[0] + +def call_shell(cmd): + return subprocess.getstatusoutput(cmd)[0] == 0 def init(*args, **kwargs): - """Desktop Changer factory""" - - logging.debug("Testing for OSX (NonX11)") - if commands.getstatusoutput("ps ax -o command -c|grep -q WindowServer")[0] == 0: - changers.append(OSXChanger(*args, **kwargs)) - - if 'DISPLAY' not in os.environ or os.environ['DISPLAY'].startswith('/tmp/launch'): - # X11 is not running - return - 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: - # X11 is not running for this display - return - - logging.debug("Testing for KDE") - if commands.getstatusoutput("xwininfo -name 'KDE Desktop'")[0] == 0: - changers.append(KDEChanger(*args, **kwargs)) - - logging.debug("Testing for Gnome") - if commands.getstatusoutput("xwininfo -name 'gnome-session'")[0] == 0: - changers.append(GnomeChanger(*args, **kwargs)) - - logging.debug("Testing for WMaker") - if commands.getstatusoutput("xlsclients | grep -qi wmaker")[0] == 0: - changers.append(WMakerChanger(*args, **kwargs)) - - if len(changers) == 0: - raise Exception("Unknown window manager") + """Desktop Changer factory""" + + classes = [] + + if sys.platform == "win32": + classes.append(WIN32Changer) + return + + logging.debug("Testing for OSX (NonX11)") + if call_shell("ps ax -o command -c|grep -q WindowServer"): + classes.append(OSXChanger) + + if 'DISPLAY' not in os.environ or os.environ['DISPLAY'].startswith('/tmp/launch'): + # X11 is not running + return + else: + if os.uname()[0] == 'Darwin': + # Try to detect if the X11 server is running on OSX + if call_shell("ps ax -o command|grep -q '^/.*X11 .* %s'" % os.environ['DISPLAY']): + # X11 is not running for this display + return + + logging.debug("Testing for XFCE4") + if call_shell("xwininfo -name 'xfdesktop'"): + classes.append(Xfce4Changer) + + logging.debug("Testing for Gnome") + if call_shell("xwininfo -name 'gnome-settings-daemon'"): + if call_shell("gsettings get org.gnome.desktop.background picture-uri"): + classes.append(Gnome3Changer) + else: + classes.append(Gnome2Changer) + + logging.debug("Testing for xloadimage") + if call_shell("which xloadimage"): + classes.append(XLoadImageChanger) + + if len(classes) == 0: + raise Exception("Unknown window manager") + + for klass in classes: + changers.append(klass(*args, **kwargs)) class BaseChanger(object): - name = "undefined" - def __init__(self, background_color='black', permanent=False, convert=False): - logging.info('Determined the window manager is "%s"', self.name) - self.background_color = background_color - self.permanent = permanent - self.convert = convert - - def set_image(self, filename): - raise NotImplementedError() - -class WMakerChanger(BaseChanger): - name = "WindowMaker" - _ConvertedWallpaperLocation = '/tmp/wallpapers_wmaker/' - 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 convert_image_format(self, file): - """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] - logging.debug("""Convert command: '"%s"'""", '" "'.join(cmd)) - return output_name, subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() - - def set_image(self, file): - if self.convert: - file, convert_status = self.convert_image_format(file) - if convert_status: - logging.debug('Convert failed') - cmd = ["wmsetbg", - "-b", self.background_color, # Sets the background colour to be what the user specified - "-S", # 'Smooth' (WTF?) - "-e", # Center the image on the screen (only affects when the image in no the in the correct aspect ratio -### "-a", # scale the image, keeping the aspect ratio - "-u", # Force this to be the default background - "-d" # dither - ] - if self.permanent: - cmd += ["-u"] # update the wmaker database - cmd += [file] - logging.debug('''WMaker bgset command: "'%s'"''', "' '".join(cmd)) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() + name = "undefined" + def __init__(self, background_color='black', convert=False): + logging.info('Determined the window manager is "%s"', self.name) + self.background_color = background_color + self.convert = convert + + 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 XLoadImageChanger(BaseChanger): + name = "xloadimage" + _ConvertedWallpaperLocation = '/tmp/wallpapers_xloadimage/' + 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 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", filename, output_name] + logging.debug("""Convert command: '"%s"'""", '" "'.join(cmd)) + return output_name, subprocess.call(cmd) == 0 + + def set_image(self, filename): + if self.convert: + filename, convert_status = self.convert_image_format(filename) + if convert_status: + logging.debug('Convert failed') + cmd = [ + "xloadimage", + "-onroot", + "-fullscreen", + "-border", "black", + filename, + ] + logging.debug('''WMaker bgset command: "'%s'"''', "' '".join(cmd)) + return subprocess.call(cmd) == 0 class OSXChanger(BaseChanger): - name = "Mac OS X" - _ConvertedWallpaperLocation = '/tmp/wallpapers/' - _DesktopPlistLocation = os.path.expanduser('~/Library/Preferences/com.apple.desktop.plist') - - 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""" - 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 convert_image_format(self, file): - """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 - 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) - return output_name, True - - def fix_desktop_plist(self): - """Removes the entry in the desktop plist file that specifies the wallpaper for each monitor""" - try: - import Foundation - desktop_plist = Foundation.NSMutableDictionary.dictionaryWithContentsOfFile_(self._DesktopPlistLocation) - # Remove all but the 'default' entry - for k in desktop_plist['Background'].keys(): - if k == 'default': - continue - desktop_plist['Background'].removeObjectForKey_(k) - # 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: - logging.debug('Could not import the Foundation module, you may have problems with dual screens') - - def set_image(self, filename): - if self.convert: - filename, ret = self.convert_image_format(filename) - if not ret: - logging.debug("Convert failed") - return False - cmd = """osascript -e 'tell application "finder" to set desktop picture to posix file "%s"'""" % filename - logging.debug(cmd) - return not commands.getstatusoutput(cmd)[0] - -class GnomeChanger(BaseChanger): - name = "Gnome" - def set_image(self, file): - cmd = ['gconftool-2', '--type', 'string', '--set', '/desktop/gnome/background/picture_filename', file] - logging.debug(cmd) - return not subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr, stdin=None).wait() - -class KDEChanger(BaseChanger): - name = "KDE" - def set_image(self, file): - cmds = [] - for group in ('Desktop0', 'Desktop0Screen0'): - base = ['kwriteconfig', '--file', 'kdesktoprc', '--group', group, '--key'] - cmds.append(base + ['Wallpaper', file]) - cmds.append(base + ['UseSHM', '--type', 'bool', 'true']) - cmds.append(base + ['WallpaperMode', 'ScaleAndCrop']) - cmds.append(base + ['MultiWallpaperMode', 'NoMulti']) - - 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: - return False - - return True - + name = "Mac OS X" + _ConvertedWallpaperLocation = '/tmp/wallpapers/' + _DesktopPlistLocation = os.path.expanduser('~/Library/Preferences/com.apple.desktop.plist') + + def __init__(self, *args, **kwargs): + BaseChanger.__init__(self, *args, **kwargs) + + 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 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: + 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(filename)) + shutil.copyfile(filename, output_name) + return output_name, True + + def fix_desktop_plist(self): + """Removes the entry in the desktop plist file that specifies the wallpaper for each monitor""" + try: + import Foundation + desktop_plist = Foundation.NSMutableDictionary.dictionaryWithContentsOfFile_(self._DesktopPlistLocation) + # Remove all but the 'default' entry + for k in desktop_plist['Background'].keys(): + if k == 'default': + continue + desktop_plist['Background'].removeObjectForKey_(k) + # 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: + 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: + logging.debug("Convert failed") + return False + cmd = """osascript -e 'tell application "finder" to set desktop picture to posix file "%s"'""" % filename + logging.debug(cmd) + return not subprocess.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 Gnome2Changer(BaseChanger): + name = "Gnome" + def set_image(self, filename): + cmd = ['gconftool-2', '--type', 'string', '--set', '/desktop/gnome/background/picture_filename', filename] + logging.debug(cmd) + return subprocess.call(cmd) == 0 + +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 subprocess.call(cmd) == 0 + +class Xfce4Changer(BaseChanger): + name = "XFCE4" + def set_image(self, filename): + for display in get_displays(): + cmd = [ + "xfconf-query", + "-c", "xfce4-desktop", + "-p", "/backdrop/screen0/monitor%s/workspace0/last-image" % display, + "-s", filename, + ] + logging.debug(cmd) + if subprocess.call(cmd) != 0: + return False + return True def main(filename): - logging.basicConfig(level=logging.DEBUG, format="%(levelname)s: %(message)s") - init() - set_image(filename) + logging.basicConfig(level=logging.DEBUG, format="%(levelname)s: %(message)s") + init() + set_image(filename) if __name__ == "__main__": - try: - filename = sys.argv[1] - except: - print >>sys.stderr, "Usage: %s filename" % sys.argv[0] - sys.exit(1) + try: + filename = sys.argv[1] + except: + print("Usage: %s filename" % sys.argv[0], file=sys.stderr) + sys.exit(1) - main(filename) + main(filename)