Add option flags nx, xx, gt, lt for expire and pexpire

Author: Saurav Sharma

changelog.d/730.feature

@@ -0,0 +1 @@
+Add option flags nx, xx, gt, lt for expire and pexpire

django_redis/client/default.py

@@ -298,6 +298,10 @@ def expire(
         timeout: ExpiryT,
         version: Optional[int] = None,
         client: Optional[Redis] = None,
+        nx: bool = False,
+        xx: bool = False,
+        gt: bool = False,
+        lt: bool = False,
     ) -> bool:
         if timeout is DEFAULT_TIMEOUT:
             timeout = self._backend.default_timeout  # type: ignore
@@ -309,14 +313,18 @@ def expire(
 
         # for some strange reason mypy complains,
         # saying that timeout type is float | timedelta
-        return client.expire(key, timeout)  # type: ignore
+        return client.expire(key, timeout, nx, xx, gt, lt)  # type: ignore
 
     def pexpire(
         self,
         key: KeyT,
         timeout: ExpiryT,
         version: Optional[int] = None,
         client: Optional[Redis] = None,
+        nx: bool = False,
+        xx: bool = False,
+        gt: bool = False,
+        lt: bool = False,
     ) -> bool:
         if timeout is DEFAULT_TIMEOUT:
             timeout = self._backend.default_timeout  # type: ignore
@@ -330,7 +338,7 @@ def pexpire(
         # is fixed.
         # for some strange reason mypy complains,
         # saying that timeout type is float | timedelta
-        return bool(client.pexpire(key, timeout))  # type: ignore
+        return bool(client.pexpire(key, timeout, nx, xx, gt, lt))  # type: ignore
 
     def pexpire_at(
         self,

tests/test_backend.py

@@ -606,6 +606,12 @@ def test_expire(self, cache: RedisCache):
         ttl = cache.ttl("foo")
         assert pytest.approx(ttl) == 20
         assert cache.expire("not-existent-key", 20) is False
+        cache.set("key1", "value1", timeout=None)
+        assert cache.expire("key1", 20, nx=True) is True
+        cache.set("key2", "value2", timeout=20)
+        assert cache.expire("key2", 21, xx=True) is True
+        assert cache.expire("key2", 30, gt=True) is True
+        assert cache.expire("key2", 20, lt=True) is True
 
     def test_expire_with_default_timeout(self, cache: RedisCache):
         cache.set("foo", "bar", timeout=None)
@@ -619,6 +625,12 @@ def test_pexpire(self, cache: RedisCache):
         # delta is set to 10 as precision error causes tests to fail
         assert pytest.approx(ttl, 10) == 20500
         assert cache.pexpire("not-existent-key", 20500) is False
+        cache.set("key1", "value1", timeout=None)
+        assert cache.pexpire("key1", 20000, nx=True) is True
+        cache.set("key2", "value2", timeout=20000)
+        assert cache.expire("key2", 20500, xx=True) is True
+        assert cache.expire("key2", 30000, gt=True) is True
+        assert cache.expire("key2", 20000, lt=True) is True
 
     def test_pexpire_with_default_timeout(self, cache: RedisCache):
         cache.set("foo", "bar", timeout=None)