Blob seriaslization with jsonpickle

axelsrz · axelsrz · commit b1b9a4f3f107 · 2019-10-15T15:38:49.000-07:00
diff --git a/libraries/botbuilder-azure/botbuilder/azure/blob_storage.py b/libraries/botbuilder-azure/botbuilder/azure/blob_storage.py
@@ -1,8 +1,11 @@
 import json
 from typing import Dict, List
 
+from jsonpickle import encode
+from jsonpickle.unpickler import Unpickler
+
 from azure.storage.blob import BlockBlobService, Blob, PublicAccess
-from botbuilder.core import Storage, StoreItem
+from botbuilder.core import Storage
 
 # TODO: sanitize_blob_name
 
@@ -59,7 +62,7 @@ async def read(self, keys: List[str]) -> Dict[str, object]:
 
         return items
 
-    async def write(self, changes: Dict[str, StoreItem]):
+    async def write(self, changes: Dict[str, object]):
         self.client.create_container(self.settings.container_name)
         self.client.set_container_acl(
             self.settings.container_name, public_access=PublicAccess.Container
@@ -71,10 +74,11 @@ async def write(self, changes: Dict[str, StoreItem]):
             )
             if e_tag:
                 item.e_tag = e_tag.replace('"', '\\"')
+            item_str = self._store_item_to_str(item)
             self.client.create_blob_from_text(
                 container_name=self.settings.container_name,
                 blob_name=name,
-                text=str(item),
+                text=item_str,
                 if_match=e_tag,
             )
 
@@ -95,8 +99,12 @@ async def delete(self, keys: List[str]):
                     container_name=self.settings.container_name, blob_name=key
                 )
 
-    def _blob_to_store_item(self, blob: Blob) -> StoreItem:
+    def _blob_to_store_item(self, blob: Blob) -> object:
         item = json.loads(blob.content)
         item["e_tag"] = blob.properties.etag
         item["id"] = blob.name
-        return StoreItem(**item)
+        result = Unpickler().restore(item)
+        return result
+
+    def _store_item_to_str(self, item: object) -> str:
+        return encode(item)
