alfred-async 0.1.0__py3-none-any.whl

alfred_async/__init__.py

@@ -0,0 +1,132 @@
+import abc
+import asyncio
+import hashlib
+import json
+import logging
+import pathlib
+import plistlib
+from datetime import UTC, datetime, timedelta
+from functools import wraps
+
+logger = logging.getLogger(__name__)
+
+
+def hash(string):
+    h = hashlib.new("sha256")
+    h.update(string.encode())
+    return h.hexdigest()
+
+
+class Path(pathlib.Path):
+    def mtime(self) -> datetime:
+        return datetime.fromtimestamp(self.stat().st_mtime, tz=UTC)
+
+    def age(self) -> timedelta:
+        return datetime.now(UTC) - self.mtime()
+
+
+def cache_json(f):
+    @wraps(f)
+    async def wrapper(client: "WorkflowClient", *args, **kwds):
+        key = hash(str(kwds))
+        path = Path(client.cache_directory / key).with_suffix(".json")
+        if client.is_cached(path):
+            logger.debug("Reading data for %s", path)
+            with path.open() as fp:
+                return json.load(fp)
+        data = await f(client, *args, **kwds)
+        logger.info("Writing data for %s", path)
+        if not client.cache_directory.exists():
+            client.cache_directory.mkdir(parents=True)
+        with path.open("w") as fp:
+            json.dump(data, fp, indent=True)
+        return data
+
+    return wrapper
+
+
+def cache_path(f):
+    @wraps(f)
+    async def wrapper(client: "WorkflowClient", *args, **kwds):
+        key = hash(str(kwds))
+        path = Path(client.cache_directory / key)
+        if client.is_cached(path):
+            logger.debug("Reading data for %s", path)
+            return path
+        data = await f(client, *args, **kwds)
+        logger.info("Writing data for %s", path)
+        if not client.cache_directory.exists():
+            client.cache_directory.mkdir(parents=True)
+        with path.open("wb+") as fp:
+            fp.write(data)
+        return path
+
+    return wrapper
+
+
+class JsonPathEncoder(json.JSONEncoder):
+    def default(self, o):
+        if isinstance(o, Path):
+            return str(o)
+        return super().default(o)
+
+
+class WorkflowClient:
+    def __init__(self):
+        self.plist_path = Path.cwd() / "info.plist"
+        if self.plist_path.exists():
+            with self.plist_path.open("rb") as fp:
+                self.plist = plistlib.load(fp)
+                self.bundleid = self.plist["bundleid"]
+
+    @property
+    def alfred_preferences(self) -> Path:
+        return (
+            Path.home()
+            / "Library"
+            / "Application Support"
+            / "Alfred"
+            / "Alfred.preferences"
+            / self.bundleid
+        )
+
+    @property
+    def cache_directory(self) -> Path:
+        return (
+            Path.home()
+            / "Caches"
+            / "com.runningwithcrayons.Alfred"
+            / "Workflow Data"
+            / self.bundleid
+        )
+
+    def is_cached(self, path: Path, duration=timedelta(minutes=5)):
+        if path.exists():
+            if path.age() < duration:
+                return True
+        return False
+
+    results = []
+
+    async def add_result(self, icon_path: str = None, **kwargs):
+        if icon_path and icon_path.startswith("http"):
+            logger.debug("need to download %s", icon_path)
+            kwargs["icon_path"] = await self.icon(icon_path)
+        self.results.append(kwargs)
+
+    async def loop(self):
+        try:
+            await self.run()
+        except Exception:
+            logger.exception("Error with workflow")
+        else:
+            print(json.dumps({"items": self.results}, cls=JsonPathEncoder))
+
+    @abc.abstractmethod
+    async def run(self):
+        raise NotImplementedError()
+
+    @classmethod
+    def main(cls):
+        client = cls()
+        return asyncio.run(client.loop())

alfred_async-0.1.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: alfred-async
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown

alfred_async-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+alfred_async/__init__.py,sha256=kuMv1U6cwovX0VaqzHSC9OuLiyVTM8pG3KhnORFBcWY,3639
+alfred_async-0.1.0.dist-info/METADATA,sha256=gXogOzcV_kXkAuUst2_tDPLNW33JYBzr1zAZp7BeALE,155
+alfred_async-0.1.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+alfred_async-0.1.0.dist-info/top_level.txt,sha256=GzxgSiSAemnPvhoEN3fSLMNtdTf3YSgR0PxITPG61cs,13
+alfred_async-0.1.0.dist-info/RECORD,,

alfred_async-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

alfred_async-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+alfred_async