chore: update store

iunary · Apr 30, 2023 · 9824f8a · 9824f8a
1 parent 6001a7b
commit 9824f8a
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/burrowkv/burrowkv.py b/burrowkv/burrowkv.py
@@ -28,7 +28,7 @@ def set(self, key: str, value: str) -> None:
             value (str): The value to associate with the key.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
         """
         with self.__lock:
@@ -46,7 +46,7 @@ def get(self, key: str) -> Optional[str]:
             if the key does not exist.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
             >>> store.get('name')
             'John'
@@ -76,7 +76,7 @@ def contains(self, key: str) -> bool:
             bool: True if the key exists, False otherwise.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
             >>> store.contains('name')
             True
@@ -93,7 +93,7 @@ def keys(self) -> List[str]:
             list: A list of keys.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
             >>> store.keys()
             ['name']
@@ -108,7 +108,7 @@ def values(self) -> List[str]:
             list: A list of values.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
             >>> store.values()
             ['John']
@@ -123,7 +123,7 @@ def items(self) -> List[Tuple[str, str]]:
             list: A list of (key, value) tuples.
 
         Example:
-            >>> store = KeyValueStore()
+            >>> store = Burrowkv()
             >>> store.set('name', 'John')
             >>> store.items()
             [('name', 'John')]
@@ -160,17 +160,14 @@ def transaction(self):
         performing multiple operations on the store.
 
         Example:
-        >>> store = KeyValueStore()
+        >>> store = Burrowkv()
         >>> with store.transaction():
         ...     store.set('name', 'John')
         ...     store.set('age', 30)
         >>> store.get('name')
         >>> store.get('age')
         """
-        try:
-            yield
-        finally:
-            self.__store.clear()
+        yield
 
     def clear(self):
         """
@@ -190,3 +187,11 @@ def __len__(self) -> int:
             int: The number of key-value pairs.
         """
         return len(self.__store)
+
+    def __repr__(self) -> str:
+        """Return the store representation
+
+        Returns:
+            str: JSON respresentation of key-value store
+        """
+        return self.to_json()