forked from Shinonome/caelestia-cli
2 * r + 2 * t
d7b65b5946
Also make it a true atomic write via os.rename (create temp in parent dir so guaranteed same fs)
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
import hashlib
|
|
import json
|
|
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
config_dir: Path = Path(os.getenv("XDG_CONFIG_HOME", Path.home() / ".config"))
|
|
data_dir: Path = Path(os.getenv("XDG_DATA_HOME", Path.home() / ".local/share"))
|
|
state_dir: Path = Path(os.getenv("XDG_STATE_HOME", Path.home() / ".local/state"))
|
|
cache_dir: Path = Path(os.getenv("XDG_CACHE_HOME", Path.home() / ".cache"))
|
|
pictures_dir: Path = Path(os.getenv("XDG_PICTURES_DIR", Path.home() / "Pictures"))
|
|
videos_dir: Path = Path(os.getenv("XDG_VIDEOS_DIR", Path.home() / "Videos"))
|
|
|
|
c_config_dir: Path = config_dir / "caelestia"
|
|
c_data_dir: Path = data_dir / "caelestia"
|
|
c_state_dir: Path = state_dir / "caelestia"
|
|
c_cache_dir: Path = cache_dir / "caelestia"
|
|
|
|
user_config_path: Path = c_config_dir / "cli.json"
|
|
cli_data_dir: Path = Path(__file__).parent.parent / "data"
|
|
templates_dir: Path = cli_data_dir / "templates"
|
|
user_templates_dir: Path = c_config_dir / "templates"
|
|
theme_dir: Path = c_state_dir / "theme"
|
|
|
|
scheme_path: Path = c_state_dir / "scheme.json"
|
|
scheme_data_dir: Path = cli_data_dir / "schemes"
|
|
scheme_cache_dir: Path = c_cache_dir / "schemes"
|
|
|
|
wallpapers_dir: Path = Path(os.getenv("CAELESTIA_WALLPAPERS_DIR", pictures_dir / "Wallpapers"))
|
|
wallpaper_path_path: Path = c_state_dir / "wallpaper/path.txt"
|
|
wallpaper_link_path: Path = c_state_dir / "wallpaper/current"
|
|
wallpaper_thumbnail_path: Path = c_state_dir / "wallpaper/thumbnail.jpg"
|
|
wallpapers_cache_dir: Path = c_cache_dir / "wallpapers"
|
|
|
|
screenshots_dir: Path = Path(os.getenv("CAELESTIA_SCREENSHOTS_DIR", pictures_dir / "Screenshots"))
|
|
screenshots_cache_dir: Path = c_cache_dir / "screenshots"
|
|
|
|
recordings_dir: Path = Path(os.getenv("CAELESTIA_RECORDINGS_DIR", videos_dir / "Recordings"))
|
|
recording_path: Path = c_state_dir / "record/recording.mp4"
|
|
recording_notif_path: Path = c_state_dir / "record/notifid.txt"
|
|
|
|
|
|
def compute_hash(path: Path | str) -> str:
|
|
sha = hashlib.sha256()
|
|
|
|
with open(path, "rb") as f:
|
|
while chunk := f.read(8192):
|
|
sha.update(chunk)
|
|
|
|
return sha.hexdigest()
|
|
|
|
|
|
def atomic_write(path: Path, content: str) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
f = tempfile.NamedTemporaryFile("w", dir=path.parent, delete=False)
|
|
try:
|
|
with f:
|
|
f.write(content)
|
|
f.flush()
|
|
os.fsync(f.fileno())
|
|
os.replace(f.name, path)
|
|
except BaseException:
|
|
os.unlink(f.name)
|
|
raise
|
|
|
|
|
|
def atomic_dump(path: Path, content: dict[str, Any]) -> None:
|
|
atomic_write(path, json.dumps(content))
|