From a6b6725eaffb848cac8ce44babbed5d42c15091f Mon Sep 17 00:00:00 2001
From: Jakub Miazek <the@grillazz.com>
Date: Thu, 9 Nov 2023 21:43:53 +0100
Subject: [PATCH] The Redis database operations are abstracted into separate
 functions

---
 app/services/auth.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/app/services/auth.py b/app/services/auth.py
index 10d22f4..3db0f36 100644
--- a/app/services/auth.py
+++ b/app/services/auth.py
@@ -8,9 +8,17 @@ from fastapi import Request, HTTPException
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 
 
+async def get_from_redis(request: Request, key: str):
+    return await request.app.state.redis.get(key)
+
+
+async def set_to_redis(request: Request, key: str, value: str, ex: int):
+    return await request.app.state.redis.set(key, value, ex=ex)
+
+
 async def verify_jwt(request: Request, token: str) -> bool:
-    _payload = await request.app.state.redis.get(token)
-    return bool(_payload)
+    payload = await get_from_redis(request, token)
+    return bool(payload)
 
 
 class AuthBearer(HTTPBearer):
@@ -29,13 +37,14 @@ class AuthBearer(HTTPBearer):
 
 
 async def create_access_token(user: User, request: Request):
-    _payload = {
+    # sourcery skip: avoid-builtin-shadow
+    payload = {
         "email": user.email,
         "expiry": time.time() + global_settings.jwt_expire,
         "platform": request.headers.get("User-Agent"),
     }
-    _token = jwt.encode(_payload, str(user.password), algorithm=global_settings.jwt_algorithm)
+    token = jwt.encode(payload, str(user.password), algorithm=global_settings.jwt_algorithm)
 
-    _bool = await request.app.state.redis.set(_token, str(_payload), ex=global_settings.jwt_expire)
+    _bool = await set_to_redis(request, token, str(payload), ex=global_settings.jwt_expire)
     if _bool:
-        return _token
+        return token