distribution classes lined up

Author: EugeneSukharev1988

montycat/store_classes/inmemory.py

@@ -17,8 +17,15 @@ async def do_snaphots_for_keyspace(cls):
             True if the snapshot operation was successful. Class 'str' if the snapshot operation failed.
         """
 
+        if cls.persistent:
+            raise ValueError("Snapshots can only be set for in-memory keyspaces.")
+
         query = orjson.dumps({
-            "raw": ["do-snapshots-for-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "n"],
+            "raw": [
+                "do-snapshots-for-keyspace", 
+                "store", cls.store, 
+                "keyspace", cls.keyspace, 
+                ],
             "credentials": [cls.username, cls.password]
         })
 
@@ -31,8 +38,15 @@ async def clean_snapshots_for_keyspace(cls):
             True if the snapshot operation was successful. Class 'str' if the snapshot operation failed.
         """
 
+        if cls.persistent:
+            raise ValueError("Snapshots can only be set for in-memory keyspaces.")
+
         query = orjson.dumps({
-            "raw": ["clean-snapshots-for-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "n"],
+            "raw": [
+                "clean-snapshots-for-keyspace", 
+                "store", cls.store, 
+                "keyspace", cls.keyspace, 
+            ],
             "credentials": [cls.username, cls.password]
         })
 
@@ -45,8 +59,15 @@ async def stop_snapshots_for_keyspace(cls):
             True if the snapshot operation was successful. Class 'str' if the snapshot operation failed.
         """
 
+        if cls.persistent:
+            raise ValueError("Snapshots can only be set for in-memory keyspaces.")
+
         query = orjson.dumps({
-            "raw": ["stop-snapshots-for-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "n"],
+            "raw": [
+                "stop-snapshots-for-keyspace",
+                "store", cls.store,
+                "keyspace", cls.keyspace,
+            ],
             "credentials": [cls.username, cls.password]
         })
 
@@ -72,7 +93,7 @@ async def insert_custom_key(cls, custom_key: str, expire_sec: int = 0):
 
     @classmethod
     async def insert_custom_key_value(cls, custom_key: str, value: dict, expire_sec: int = 0):
-        """       
+        """
         Args:
             custom_key: A custom key to insert into the store. This key can be used to retrieve the value later.
             value: A Python class / dict to insert into the store.
@@ -91,10 +112,10 @@ async def insert_custom_key_value(cls, custom_key: str, value: dict, expire_sec:
 
         query = convert_to_binary_query(cls, key=custom_key_converted, value=value, expire_sec=expire_sec)
         return await cls._run_query(query)
-    
+
     @classmethod
     async def insert_value(cls, value: dict, expire_sec: int = 0):
-        """       
+        """
         Args:
             value: A Python class / dict to insert into the store.
             expire_sec: The number of seconds before the inserted value expires.
@@ -103,23 +124,23 @@ async def insert_value(cls, value: dict, expire_sec: int = 0):
         """
         if not value:
             raise ValueError("No value provided for insertion.")
-        
+
         cls.command = "insert_value"
 
         query = convert_to_binary_query(cls, value=value, expire_sec=expire_sec)
         return await cls._run_query(query)
-    
+
     @classmethod
     async def update_value(cls, key: Union[str, None] = None, custom_key: Union[str, None] = None, expire_sec: int = 0, **filters):
         """
         Update the value associated with a given key in the store. If a custom key is provided,
         it will be converted to the appropriate format before updating.
 
         Args:
-            key (int | str, optional): The key whose associated value needs to be updated. 
+            key (int | str, optional): The key whose associated value needs to be updated.
                                        This can either be an integer or a string. Default is an empty string,
                                        which will be ignored if custom_key is provided.
-            custom_key (str, optional): The custom key whose associated value needs to be updated. 
+            custom_key (str, optional): The custom key whose associated value needs to be updated.
                                         Default is an empty string.
             filters (dict): A dictionary of field-value pairs that need to be updated in the store.
 
@@ -135,36 +156,36 @@ async def update_value(cls, key: Union[str, None] = None, custom_key: Union[str,
             raise ValueError("No filters provided")
         if not key:
             raise ValueError("No key provided")
-        
+
         cls.command = "update_value"
 
         query = convert_to_binary_query(cls, key=key, value=filters, expire_sec=expire_sec)  # Convert the key and filters into a binary query format
         return await cls._run_query(query)  # Run the query and return the result
 
     @classmethod
     async def insert_bulk(cls, bulk_values: list, expire_sec: int = 0):
-        """       
+        """
         Args:
             bulk_values: A list of Python objects to insert into the store.
             expire_sec: The number of seconds before the inserted values expire.
-            
+
         Returns:
             True if the bulk insert operation was successful.
             List of values that were not inserted.
         """
 
         if not bulk_values:
             raise ValueError("No values provided for bulk insertion.")
-        
+
         cls.command = "insert_bulk"
         query = convert_to_binary_query(cls, bulk_values=bulk_values, expire_sec=expire_sec)
         return await cls._run_query(query)
-    
+
     @classmethod
     async def get_keys(cls):
         """
         Get all keys in the store.
-        
+
         Returns:
             A list of keys in the store. Class 'str' if the get operation failed.
         """
@@ -173,11 +194,17 @@ async def get_keys(cls):
         query = convert_to_binary_query(cls)
         return await cls._run_query(query)
 
-    @classmethod  
+    @classmethod
     async def create_keyspace(cls):
 
         query = orjson.dumps({
-            "raw": ["create-keyspace", "store", cls.store, "keyspace", cls.keyspace, "persistent", "y" if cls.persistent else "n"],
+            "raw": [
+                "create-keyspace",
+                "store", cls.store,
+                "keyspace", cls.keyspace,
+                "persistent", "y" if cls.persistent else "n",
+                "distributed", "y" if cls.distributed else "n"
+            ],
             "credentials": [cls.username, cls.password]
         })