Add support for PEXPIRE command's option

Author: dogukanteber

redis/commands/core.py

@@ -1755,17 +1755,26 @@ def persist(self, name: KeyT) -> ResponseT:
         """
         return self.execute_command("PERSIST", name)
 
-    def pexpire(self, name: KeyT, time: ExpiryT) -> ResponseT:
+    def pexpire(self, name: KeyT, time: ExpiryT, option: str = None) -> ResponseT:
         """
-        Set an expire flag on key ``name`` for ``time`` milliseconds.
-        ``time`` can be represented by an integer or a Python timedelta
-        object.
+        Set an expire flag on key ``name`` for ``time`` milliseconds
+        with given ``option``. ``time`` can be represented by an
+        integer or a Python timedelta object.
+
+        Valid options are:
+            NX -> Set expiry only when the key has no expiry
+            XX -> Set expiry only when the key has an existing expiry
+            GT -> Set expiry only when the new expiry is greater than current one
+            LT -> Set expiry only when the new expiry is less than current one
 
         For more information check https://redis.io/commands/pexpire
         """
         if isinstance(time, datetime.timedelta):
             time = int(time.total_seconds() * 1000)
-        return self.execute_command("PEXPIRE", name, time)
+        exp_option = list()
+        if option is not None:
+            exp_option.append(option)
+        return self.execute_command("PEXPIRE", name, time, *exp_option)
 
     def pexpireat(self, name: KeyT, when: AbsExpiryT) -> ResponseT:
         """

tests/test_commands.py

@@ -1212,6 +1212,33 @@ def test_pexpire(self, r):
         assert r.persist("a")
         assert r.pttl("a") == -1
 
+    @skip_if_server_version_lt("7.0.0")
+    def test_pexpire_option_nx(self, r):
+        assert r.set("key", "val") is True
+        assert r.pexpire("key", 60000, "NX") is True
+        assert r.pexpire("key", 60000, "NX") is False
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_pexpire_option_xx(self, r):
+        assert r.set("key", "val") is True
+        assert r.pexpire("key", 60000, "XX") is False
+        assert r.pexpire("key", 60000) is True
+        assert r.pexpire("key", 70000, "XX") is True
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_pexpire_option_gt(self, r):
+        assert r.set("key", "val") is True
+        assert r.pexpire("key", 60000) is True
+        assert r.pexpire("key", 70000, "GT") is True
+        assert r.pexpire("key", 50000, "GT") is False
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_pexpire_option_lt(self, r):
+        assert r.set("key", "val") is True
+        assert r.pexpire("key", 60000) is True
+        assert r.pexpire("key", 50000, "LT") is True
+        assert r.pexpire("key", 70000, "LT") is False
+
     @skip_if_server_version_lt("2.6.0")
     def test_pexpireat_datetime(self, r):
         expire_at = redis_server_time(r) + datetime.timedelta(minutes=1)