Fix docs re: JSONDisk

grantjenks · grantjenks · commit cffbcec2b198 · 2023-04-17T22:30:26.000-07:00
diff --git a/docs/tutorial.rst b/docs/tutorial.rst
@@ -821,30 +821,28 @@ example below uses compressed JSON, available for convenience as
 
 .. code-block:: python
 
-    import json, zlib
-
     class JSONDisk(diskcache.Disk):
         def __init__(self, directory, compress_level=1, **kwargs):
             self.compress_level = compress_level
-            super(JSONDisk, self).__init__(directory, **kwargs)
+            super().__init__(directory, **kwargs)
 
         def put(self, key):
             json_bytes = json.dumps(key).encode('utf-8')
             data = zlib.compress(json_bytes, self.compress_level)
-            return super(JSONDisk, self).put(data)
+            return super().put(data)
 
         def get(self, key, raw):
-            data = super(JSONDisk, self).get(key, raw)
+            data = super().get(key, raw)
             return json.loads(zlib.decompress(data).decode('utf-8'))
 
-        def store(self, value, read):
+        def store(self, value, read, key=UNKNOWN):
             if not read:
                 json_bytes = json.dumps(value).encode('utf-8')
                 value = zlib.compress(json_bytes, self.compress_level)
-            return super(JSONDisk, self).store(value, read)
+            return super().store(value, read, key=key)
 
         def fetch(self, mode, filename, value, read):
-            data = super(JSONDisk, self).fetch(mode, filename, value, read)
+            data = super().fetch(mode, filename, value, read)
             if not read:
                 data = json.loads(zlib.decompress(data).decode('utf-8'))
             return data
