Release v2

nyekiapi/lib/cache.py

@@ -0,0 +1,39 @@
+import orjson
+
+from redis.asyncio import Redis
+
+from nyekiapi.lib import settings
+
+
+redis = Redis.from_url(settings.REDIS_URL)
+
+
+async def get(key: str) -> str | None:
+    """Retrieve a value from Redis by key.
+
+    Args:
+        key (str): The key to retrieve the value for.
+
+    Returns:
+        str | None: The value associated with the key, or None if the key does not exist.
+    """
+
+    raw = await redis.get(key)
+
+    if raw:
+        return orjson.loads(raw)
+    
+    else:
+        return None
+
+
+async def set(key: str, value: str, expires_in: int | None = None) -> None:
+    """Set a value in Redis with an optional expiration time.
+
+    Args:
+        key (str): The key to set the value for.
+        value (str): The value to set.
+        expires_in (int | None, optional): The expiration time in seconds. Defaults to None.
+    """
+
+    await redis.set(key, orjson.dumps(value), ex=expires_in)