Support username and password in url (#5)

* Support username and password in url * Update readme
b-scheduler · Jun 6, 2024 · 09983a2 · 09983a2
1 parent eadb1fa
commit 09983a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -48,6 +48,8 @@ async def main():
         db=int(os.getenv("REDIS_DB", 0)),
         cluster=bool(os.getenv("REDIS_CLUSTER", "false") in ["True", "true", "1"]),
         tls=bool(os.getenv("REDIS_TLS", "false") in ["True", "true", "1"]),
+        username=os.getenv("REDIS_USERNAME", ""),
+        password=os.getenv("REDIS_PASSWORD", ""),
     )
     async with get_redis_client(redis_url) as async_redis_client:
         await Producer(async_redis_client).run_job("echo", ["hello"])
@@ -80,6 +82,8 @@ async def main():
         db=int(os.getenv("REDIS_DB", 0)),
         cluster=bool(os.getenv("REDIS_CLUSTER", "false") in ["True", "true", "1"]),
         tls=bool(os.getenv("REDIS_TLS", "false") in ["True", "true", "1"]),
+        username=os.getenv("REDIS_USERNAME", ""),
+        password=os.getenv("REDIS_PASSWORD", ""),
     )
     async with get_redis_client(redis_url) as async_redis_client:
         daemon = Daemon(Consumer(async_redis_client, echo))

diff --git a/brq/tools.py b/brq/tools.py
@@ -42,13 +42,18 @@ def get_redis_url(
     db: int = 0,
     cluster: bool = False,
     tls: bool = False,
+    username: str = "",
+    password: str = "",
 ):
     if not cluster:
         url = f"{host}:{port}/{db}"
     else:
         logger.debug("Cluster mode, no need to specify db")
         url = f"{host}:{port}"
 
+    if username or password:
+        url = f"{username}:{password}@{url}"
+
     if tls:
         return f"rediss://{url}"
     else: