Shinonome/caelestia-cli
1
0
Fork 1
mirror of https://github.com/caelestia-dots/cli.git synced 2026-06-06 07:19:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

wallpaper: cache smart mode

Browse Source
This commit is contained in:
2 * r + 2 * t
2025-06-12 21:51:59 +10:00
parent c043a14ca2
commit a97de9d430
2 changed files with 25 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/caelestia/utils/paths.py
+1 -1
View File
@@ -26,7 +26,7 @@ wallpapers_dir = Path.home() / "Pictures/Wallpapers"
wallpaper_path_path = c_state_dir / "wallpaper/path.txt"
wallpaper_link_path = c_state_dir / "wallpaper/current"
wallpaper_thumbnail_path = c_state_dir / "wallpaper/thumbnail.jpg"
thumbnail_cache_dir = c_cache_dir / "thumbnails"
wallpapers_cache_dir = c_cache_dir / "wallpapers"
def compute_hash(path: Path | str) -> str:
src/caelestia/utils/wallpaper.py
+24 -11
View File
@@ -10,10 +10,10 @@ from caelestia.utils.hypr import message
from caelestia.utils.material import get_colours_for_image
from caelestia.utils.paths import (
compute_hash,
thumbnail_cache_dir,
wallpaper_link_path,
wallpaper_path_path,
wallpaper_thumbnail_path,
wallpapers_cache_dir,
)
from caelestia.utils.scheme import Scheme, get_scheme
from caelestia.utils.theme import apply_colours
@@ -55,8 +55,8 @@ def get_wallpapers(args: Namespace) -> list[Path]:
return [f for f in walls if check_wall(f, filter_size, args.threshold)]
def get_thumb(wall: Path) -> Path:
thumb = (thumbnail_cache_dir / compute_hash(wall)).with_suffix(".jpg")
def get_thumb(wall: Path, cache: Path) -> Path:
thumb = cache / "thumbnail.jpg"
if not thumb.exists():
with Image.open(wall) as img:
@@ -68,28 +68,38 @@ def get_thumb(wall: Path) -> Path:
return thumb
def get_smart_mode(wall: Path) -> str:
with Image.open(get_thumb(wall)) as img:
def get_smart_mode(wall: Path, cache: Path) -> str:
mode_cache = cache / "mode.txt"
if mode_cache.exists():
return mode_cache.read_text()
with Image.open(get_thumb(wall, cache)) as img:
img.thumbnail((1, 1), Image.LANCZOS)
tone = Hct.from_int(argb_from_rgb(*img.getpixel((0, 0)))).tone
return "light" if tone > 60 else "dark"
mode = "light" if Hct.from_int(argb_from_rgb(*img.getpixel((0, 0)))).tone > 60 else "dark"
mode_cache.parent.mkdir(parents=True, exist_ok=True)
mode_cache.write_text(mode)
return mode
def get_colours_for_wall(wall: Path | str, no_smart: bool) -> None:
scheme = get_scheme()
cache = wallpapers_cache_dir / compute_hash(wall)
if not no_smart:
scheme = Scheme(
{
"name": scheme.name,
"flavour": scheme.flavour,
"mode": get_smart_mode(wall),
"mode": get_smart_mode(wall, cache),
"variant": scheme.variant,
"colours": scheme.colours,
}
)
return get_colours_for_image(get_thumb(wall), scheme)
return get_colours_for_image(get_thumb(wall, cache), scheme)
def set_wallpaper(wall: Path | str, no_smart: bool) -> None:
@@ -103,8 +113,10 @@ def set_wallpaper(wall: Path | str, no_smart: bool) -> None:
wallpaper_link_path.unlink(missing_ok=True)
wallpaper_link_path.symlink_to(wall)
cache = wallpapers_cache_dir / compute_hash(wall)
# Generate thumbnail or get from cache
thumb = get_thumb(wall)
thumb = get_thumb(wall, cache)
wallpaper_thumbnail_path.parent.mkdir(parents=True, exist_ok=True)
wallpaper_thumbnail_path.unlink(missing_ok=True)
wallpaper_thumbnail_path.symlink_to(thumb)
@@ -112,7 +124,8 @@ def set_wallpaper(wall: Path | str, no_smart: bool) -> None:
scheme = get_scheme()
# Change mode based on wallpaper colour
scheme.mode = get_smart_mode(wall)
if not no_smart:
scheme.mode = get_smart_mode(wall, cache)
# Update colours
scheme.update_colours()