Add missing cache_selfie_json and cache_selfie_binary methods.

Author: nedtwigg

python/selfie-lib/selfie_lib/CacheSelfie.py

@@ -33,6 +33,29 @@ def cache_selfie(
     else:
         raise TypeError("Invalid arguments provided to cache_selfie")
 
+def cache_selfie_json(to_cache: Callable[..., T]) -> "CacheSelfie[T]":
+    return cache_selfie(to_cache, Roundtrip.json())
+
+@overload
+def cache_selfie_binary(to_cache: Callable[..., bytes]) -> "CacheSelfieBinary[bytes]": ...
+
+@overload
+def cache_selfie_binary(
+    to_cache: Callable[..., T], roundtrip: Roundtrip[T, bytes]
+) -> "CacheSelfieBinary[T]": ...
+
+def cache_selfie_binary(
+    to_cache: Union[Callable[..., bytes], Callable[..., T]],
+    roundtrip: Optional[Roundtrip[T, bytes]] = None,
+) -> Union["CacheSelfieBinary[bytes]", "CacheSelfieBinary[T]"]:
+    if roundtrip is None:
+        # the cacheable better be a bytes!
+        return cache_selfie_binary(to_cache, Roundtrip.identity())  # type: ignore
+    elif isinstance(roundtrip, Roundtrip) and to_cache is not None:
+        deferred_disk_storage = _selfieSystem().disk_thread_local()
+        return CacheSelfieBinary(deferred_disk_storage, roundtrip, to_cache)  # type: ignore
+    else:
+        raise TypeError("Invalid arguments provided to cache_selfie")
 
 class CacheSelfie(Generic[T]):
     def __init__(

python/selfie-lib/selfie_lib/Roundtrip.py

@@ -26,3 +26,17 @@ def parse(self, serialized: Any) -> Any:
                 return serialized
 
         return Identity()
+
+    @classmethod
+    def json(cls) -> "Roundtrip[T, str]":
+        """Return a Roundtrip that serializes to/from JSON strings."""
+        import json
+
+        class JsonRoundtrip(Roundtrip[Any, str]):
+            def serialize(self, value: Any) -> str:
+                return json.dumps(value, indent=4)
+
+            def parse(self, serialized: str) -> Any:
+                return json.loads(serialized)
+
+        return JsonRoundtrip()

python/selfie-lib/selfie_lib/__init__.py

@@ -3,6 +3,8 @@
 from .ArrayMap import ArraySet as ArraySet
 from .Atomic import AtomicReference as AtomicReference
 from .CacheSelfie import cache_selfie as cache_selfie
+from .CacheSelfie import cache_selfie_binary as cache_selfie_binary
+from .CacheSelfie import cache_selfie_json as cache_selfie_json
 from .CommentTracker import CommentTracker as CommentTracker
 from .FS import FS as FS
 from .Lens import Camera as Camera