blob: b96849faf627a6efd171609313c0be889e948baa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
import json
from pathlib import Path
from materialyoucolor.hct import Hct
from caelestia.utils.material.generator import gen_scheme
from caelestia.utils.material.score import score
from caelestia.utils.paths import compute_hash, scheme_cache_dir, wallpaper_thumbnail_path
def get_score_for_image(image: Path | str, cache_base: Path) -> Hct:
cache = cache_base / "score.json"
try:
return Hct.from_int(cache.read_text())
except (IOError, TypeError):
pass
s = score(str(image))
cache.parent.mkdir(parents=True, exist_ok=True)
cache.write_text(str(s.to_int()))
return s
def get_colours_for_image(image: Path | str = wallpaper_thumbnail_path, scheme=None) -> dict[str, str]:
if scheme is None:
from caelestia.utils.scheme import get_scheme
scheme = get_scheme()
cache_base = scheme_cache_dir / compute_hash(image)
cache = (cache_base / scheme.variant / scheme.mode).with_suffix(".json")
try:
with cache.open("r") as f:
return json.load(f)
except (IOError, json.JSONDecodeError):
pass
primary = get_score_for_image(image, cache_base)
scheme = gen_scheme(scheme, primary)
cache.parent.mkdir(parents=True, exist_ok=True)
with cache.open("w") as f:
json.dump(scheme, f)
return scheme
|
