initial commit

macos_wallpaper.py

@@ -0,0 +1,72 @@
+import contextlib
+import dataclasses
+import plistlib
+import sqlite3
+import subprocess
+from pathlib import Path
+from typing import Optional
+
+DB_PATH = Path("~/Library/Application Support/Dock/desktoppicture.db").expanduser()
+
+
+@dataclasses.dataclass
+class SpaceInfo:
+    uuid: str
+    image_path: Optional[Path]
+    data_id: Optional[Path]
+
+
+@contextlib.contextmanager
+def get_connection() -> sqlite3.Connection:
+    con = sqlite3.connect(str(DB_PATH))
+    con.row_factory = sqlite3.Row
+    yield con
+    con.close()
+
+
+def get_current_wallpapers() -> list[SpaceInfo]:
+    sql = """
+    select distinct s.space_uuid, pr.key, d.rowid, d.value from preferences pr
+    left join payload d on d.rowid = pr.data_id
+    left join pictures p on p.rowid = pr.picture_id
+    inner join spaces s on s.rowid = p.space_id
+    where key = 1
+    
+    order by space_uuid, key
+    """
+    spaces = list_spaces()
+
+    with get_connection() as con:
+        cur = con.execute(sql)
+
+        space_props = {}
+        for i, row in enumerate(cur):
+            space_uuid = row["space_uuid"]
+            data_id = row["rowid"]
+            image_path = row["value"]
+
+            if image_path:
+                image_path = Path(image_path).expanduser()
+            space_props[space_uuid] = SpaceInfo(uuid=space_uuid, image_path=image_path, data_id=data_id)
+    return [space_props.get(id, SpaceInfo(uuid="", image_path=None, data_id=None)) for id in spaces]
+
+
+def read_plist(plist_path):
+    with plist_path.open("rb") as f:
+        return plistlib.load(f)
+
+
+def list_spaces():
+    spaces_plist_path = Path("~/Library/Preferences/com.apple.spaces.plist").expanduser()
+    data = read_plist(spaces_plist_path)
+    spaces = data["SpacesDisplayConfiguration"]["Management Data"]["Monitors"][0]["Spaces"]
+    return [s["uuid"] for s in spaces]
+
+
+def set_wallpaper(image_path: Path):
+    script = f'tell application "Finder" to set desktop picture to "{image_path.expanduser().resolve()}" as POSIX file'
+    subprocess.run(["osascript", "-e", script], check=True)
+
+
+if __name__ == "__main__":
+    pass