rofi_selector.py

#!/usr/bin/env python3

from auto_walls import State, get_config, set_wallpaper
import subprocess, os

if __name__ == '__main__':

    c = get_config()
    state = State()

    wd = os.path.expanduser(c["wallpapers_dir"])

    # Generate Rofi options with thumbnails
    rofi_options = "\n".join(f"{w.split("/")[-1]}\x00icon\x1f{w}" for w in state.wallpapers)

    # Launch Rofi with thumbnails and passed theme (none if empty)
    rofi_theme =  "-theme " + c["rofi_theme"] if c["rofi_theme"] else ' '
    
    rofi_process = subprocess.Popen(["rofi", "-dmenu", "-i"] + rofi_theme.split() + ["-selected-row", str(state.index)],
                                    stdin=subprocess.PIPE,
                                    stdout=subprocess.PIPE,
                                    stderr=subprocess.PIPE,
                                    text=True)

    # Pass wallpapers with thumbnails to Rofi
    selected_option, _ = rofi_process.communicate(input=rofi_options)
    selected_option = selected_option.strip()

    # Check if a wallpaper was selected
    if selected_option:
        selected_wallpaper_dir = os.path.join(wd, selected_option.split('\x00icon\x1f')[0])  # Extract full wallpaper path
        i = state.wallpapers.index(selected_wallpaper_dir)
        if i != state.index:
            set_wallpaper(c, state, selected_wallpaper_dir, i)