mirror of
https://github.com/dylanaraps/pywal.git
synced 2025-01-25 07:19:27 +01:00
149 lines
4.1 KiB
Python
149 lines
4.1 KiB
Python
"""Set the wallpaper."""
|
|
import ctypes
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import urllib.parse
|
|
|
|
from .settings import CACHE_DIR, HOME, OS
|
|
from . import util
|
|
|
|
|
|
def get_desktop_env():
|
|
"""Identify the current running desktop environment."""
|
|
desktop = os.environ.get("XDG_CURRENT_DESKTOP")
|
|
if desktop:
|
|
return desktop
|
|
|
|
desktop = os.environ.get("DESKTOP_SESSION")
|
|
if desktop:
|
|
return desktop
|
|
|
|
desktop = os.environ.get("GNOME_DESKTOP_SESSION_ID")
|
|
if desktop:
|
|
return "GNOME"
|
|
|
|
desktop = os.environ.get("MATE_DESKTOP_SESSION_ID")
|
|
if desktop:
|
|
return "MATE"
|
|
|
|
desktop = os.environ.get("SWAYSOCK")
|
|
if desktop:
|
|
return "SWAY"
|
|
|
|
return None
|
|
|
|
|
|
def xfconf(path, img):
|
|
"""Call xfconf to set the wallpaper on XFCE."""
|
|
util.disown(["xfconf-query", "--channel", "xfce4-desktop",
|
|
"--property", path, "--set", img])
|
|
|
|
|
|
def set_wm_wallpaper(img):
|
|
"""Set the wallpaper for non desktop environments."""
|
|
if shutil.which("feh"):
|
|
util.disown(["feh", "--bg-fill", img])
|
|
|
|
elif shutil.which("nitrogen"):
|
|
util.disown(["nitrogen", "--set-zoom-fill", img])
|
|
|
|
elif shutil.which("bgs"):
|
|
util.disown(["bgs", "-z", img])
|
|
|
|
elif shutil.which("hsetroot"):
|
|
util.disown(["hsetroot", "-fill", img])
|
|
|
|
elif shutil.which("habak"):
|
|
util.disown(["habak", "-mS", img])
|
|
|
|
elif shutil.which("display"):
|
|
util.disown(["display", "-backdrop", "-window", "root", img])
|
|
|
|
else:
|
|
print("error: No wallpaper setter found.")
|
|
return
|
|
|
|
|
|
def set_desktop_wallpaper(desktop, img):
|
|
"""Set the wallpaper for the desktop environment."""
|
|
desktop = str(desktop).lower()
|
|
|
|
if "xfce" in desktop or "xubuntu" in desktop:
|
|
# XFCE requires two commands since they differ between versions.
|
|
xfconf("/backdrop/screen0/monitor0/image-path", img)
|
|
xfconf("/backdrop/screen0/monitor0/workspace0/last-image", img)
|
|
|
|
elif "muffin" in desktop or "cinnamon" in desktop:
|
|
util.disown(["gsettings", "set",
|
|
"org.cinnamon.desktop.background",
|
|
"picture-uri", "file://" + urllib.parse.quote(img)])
|
|
|
|
elif "gnome" in desktop:
|
|
util.disown(["gsettings", "set",
|
|
"org.gnome.desktop.background",
|
|
"picture-uri", "file://" + urllib.parse.quote(img)])
|
|
|
|
elif "mate" in desktop:
|
|
util.disown(["gsettings", "set", "org.mate.background",
|
|
"picture-filename", img])
|
|
|
|
elif "sway" in desktop:
|
|
util.disown(["swaymsg", "output", "*", "bg", img, "fill"])
|
|
|
|
else:
|
|
set_wm_wallpaper(img)
|
|
|
|
|
|
def set_mac_wallpaper(img):
|
|
"""Set the wallpaper on macOS."""
|
|
db_file = "Library/Application Support/Dock/desktoppicture.db"
|
|
db_path = os.path.join(HOME, db_file)
|
|
subprocess.call(["sqlite3", db_path, "update data set value = '%s'" % img])
|
|
|
|
# Kill the dock to fix issues with cached wallpapers.
|
|
# macOS caches wallpapers and if a wallpaper is set that shares
|
|
# the filename with a cached wallpaper, the cached wallpaper is
|
|
# used instead.
|
|
subprocess.call(["killall", "Dock"])
|
|
|
|
|
|
def set_win_wallpaper(img):
|
|
"""Set the wallpaper on Windows."""
|
|
# There's a different command depending on the architecture
|
|
# of Windows. We check the PROGRAMFILES envar since using
|
|
# platform is unreliable.
|
|
if "x86" in os.environ["PROGRAMFILES"]:
|
|
ctypes.windll.user32.SystemParametersInfoW(20, 0, img, 3)
|
|
else:
|
|
ctypes.windll.user32.SystemParametersInfoA(20, 0, img, 3)
|
|
|
|
|
|
def change(img):
|
|
"""Set the wallpaper."""
|
|
if not os.path.isfile(img):
|
|
return
|
|
|
|
desktop = get_desktop_env()
|
|
|
|
if OS == "Darwin":
|
|
set_mac_wallpaper(img)
|
|
|
|
elif OS == "Windows":
|
|
set_win_wallpaper(img)
|
|
|
|
else:
|
|
set_desktop_wallpaper(desktop, img)
|
|
|
|
print("wallpaper: Set the new wallpaper.")
|
|
|
|
|
|
def get(cache_dir=CACHE_DIR):
|
|
"""Get the current wallpaper."""
|
|
current_wall = os.path.join(cache_dir, "wal")
|
|
|
|
if os.path.isfile(current_wall):
|
|
return util.read_file(current_wall)[0]
|
|
|
|
return "None"
|