import os
import random
import subprocess
from libqtile.log_utils import logger
from libqtile.widget import base
[docs]
class Wallpaper(base._TextBox):
defaults = [
("directory", "~/Pictures/wallpapers/", "Wallpaper Directory"),
("wallpaper", None, "Wallpaper"),
(
"wallpaper_command",
["feh", "--bg-fill"],
"Wallpaper command. If None, the"
"wallpaper will be painted without the use of a helper.",
),
(
"random_selection",
False,
"If set, use random initial wallpaper and randomly cycle through the wallpapers.",
),
("label", None, "Use a fixed label instead of image name."),
(
"option",
"fill",
"How to fit the wallpaper when wallpaper_command isNone. None, 'fill' or 'stretch'.",
),
]
def __init__(self, **config):
base._TextBox.__init__(self, "empty", **config)
self.add_defaults(Wallpaper.defaults)
self.index = 0
self.images = []
self.get_wallpapers()
if self.random_selection: # Random selection after reading all files
self.index = random.randint(0, len(self.images) - 1)
self.add_callbacks({"Button1": self.set_wallpaper})
def _configure(self, qtile, bar):
base._TextBox._configure(self, qtile, bar)
if not self.bar.screen.wallpaper:
self.set_wallpaper()
def get_path(self, file):
return os.path.join(os.path.expanduser(self.directory), file)
def get_wallpapers(self):
try:
# get path of all files in the directory
self.images = list(
filter(
os.path.isfile,
map(self.get_path, os.listdir(os.path.expanduser(self.directory))),
)
)
except OSError as e:
logger.exception("I/O error(%s): %s", e.errno, e.strerror)
def set_wallpaper(self):
if len(self.images) == 0:
if self.wallpaper is None:
self.text = "empty"
return
else:
self.images.append(self.wallpaper)
if self.random_selection:
self.index = random.randint(0, len(self.images) - 1)
else:
self.index += 1
self.index %= len(self.images)
cur_image = self.images[self.index]
if self.label is None:
self.text = os.path.basename(cur_image)
else:
self.text = self.label
if self.wallpaper_command:
self.wallpaper_command.append(cur_image)
subprocess.call(self.wallpaper_command)
self.wallpaper_command.pop()
else:
self.qtile.paint_screen(self.bar.screen, cur_image, self.option)
self.draw()