Ran Linter and Black, fixed export error

valkey-io · Jan 20, 2025 · 63f6f7d · 63f6f7d
1 parent 75384a1
commit 63f6f7d
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 deletions.
diff --git a/python/python/glide/__init__.py b/python/python/glide/__init__.py
@@ -20,7 +20,6 @@
 from glide.async_commands.command_args import Limit, ListDirection, ObjectType, OrderBy
 from glide.async_commands.core import (
     ConditionalChange,
-    OnlyIfEqual,
     CoreCommands,
     ExpireOptions,
     ExpiryGetEx,
@@ -31,6 +30,7 @@
     FunctionRestorePolicy,
     InfoSection,
     InsertPosition,
+    OnlyIfEqual,
     UpdateOptions,
 )
 from glide.async_commands.server_modules import ft, glide_json, json_transaction
@@ -226,6 +226,7 @@
     "Script",
     "ScoreBoundary",
     "ConditionalChange",
+    "OnlyIfEqual",
     "ExpireOptions",
     "ExpiryGetEx",
     "ExpirySet",

diff --git a/python/python/glide/async_commands/core.py b/python/python/glide/async_commands/core.py
@@ -78,15 +78,17 @@ class ConditionalChange(Enum):
     ONLY_IF_EXISTS = "XX"
     ONLY_IF_DOES_NOT_EXIST = "NX"
 
+
 @dataclass
-class OnlyIfEqual():
+class OnlyIfEqual:
     """
     Change condition to the `SET` command,
     For additional conditonal options see ConditionalChange
-    - comparison_value - value to compare to the current value of a key. 
+    - comparison_value - value to compare to the current value of a key.
     If comparison_value is equal to the key, it will overwrite the value of key to the new provided value
-    Equivalent to the IFEQ comparison-value in the Valkey API 
+    Equivalent to the IFEQ comparison-value in the Valkey API
     """
+
     comparison_value: TEncodable
 
 

diff --git a/python/python/tests/test_async_client.py b/python/python/tests/test_async_client.py
@@ -29,7 +29,6 @@
 from glide.async_commands.command_args import Limit, ListDirection, OrderBy
 from glide.async_commands.core import (
     ConditionalChange,
-    OnlyIfEqual,
     ExpireOptions,
     ExpiryGetEx,
     ExpirySet,
@@ -40,6 +39,7 @@
     InfBound,
     InfoSection,
     InsertPosition,
+    OnlyIfEqual,
     UpdateOptions,
 )
 from glide.async_commands.sorted_set import (
@@ -484,12 +484,16 @@ async def test_set_only_if_equal(self, glide_client: TGlideClient):
         await glide_client.set(key, value)
 
         res = await glide_client.set(
-            key, "foobar", conditional_set=OnlyIfEqual(wrong_comparison_value),
+            key,
+            "foobar",
+            conditional_set=OnlyIfEqual(wrong_comparison_value),
         )
         assert res is None
         assert await glide_client.get(key) == value.encode()
         res = await glide_client.set(
-            key, value2, conditional_set=OnlyIfEqual(value),
+            key,
+            value2,
+            conditional_set=OnlyIfEqual(value),
         )
         assert res == OK
         assert await glide_client.get(key) == value2.encode()