Adding support for GENPASS bits

redis/commands.py

@@ -52,9 +52,22 @@ def acl_deluser(self, username):
         "Delete the ACL for the specified ``username``"
         return self.execute_command('ACL DELUSER', username)
 
-    def acl_genpass(self):
-        "Generate a random password value"
-        return self.execute_command('ACL GENPASS')
+    def acl_genpass(self, bits=None):
+        """Generate a random password value.
+        If ``bits`` is supplied then use this number of bits, rounded to
+        the next multiple of 4.
+        See: https://redis.io/commands/acl-genpass
+        """
+        pieces = []
+        if bits is not None:
+            try:
+                b = int(bits)
+                if b < 0 or b > 4096:
+                    raise ValueError
+            except ValueError:
+                raise DataError('genpass optionally accepts a bits argument, '
+                                'between 0 and 4096.')
+        return self.execute_command('ACL GENPASS', *pieces)
 
     def acl_getuser(self, username):
         """

tests/test_commands.py

@@ -98,6 +98,14 @@ def test_acl_genpass(self, r):
         password = r.acl_genpass()
         assert isinstance(password, str)
 
+        with pytest.raises(exceptions.DataError):
+            r.acl_genpass('value')
+            r.acl_genpass(-5)
+            r.acl_genpass(5555)
+
+        r.acl_genpass(555)
+        assert isinstance(password, str)
+
     @skip_if_server_version_lt(REDIS_6_VERSION)
     def test_acl_getuser_setuser(self, r, request):
         username = 'redis-py-user'