Nattyboi commited on
Commit
827c42e
·
1 Parent(s): f3b3e61

changed up something for jwt

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. jwtcoding.py +1 -1
app.py CHANGED
@@ -291,7 +291,7 @@ def login(user:UserBody):
291
  result = update_refreshTokenWithPreviouslyUsedAccessToken(db_uri=MONGO_URI,refresh_token=refreshToken,access_token=accessToken)
292
  print(result)
293
  access_token = encode_jwt(user_id=user_id,access_token=accessToken)
294
- return {"userId":user_id,"refreshToken":refreshToken,"accessToken":access_token}
295
  return JSONResponse(status_code=401,content="Invalid login details")
296
 
297
 
@@ -305,7 +305,7 @@ def signUp(user:UserBody):
305
  result = update_refreshTokenWithPreviouslyUsedAccessToken(db_uri=MONGO_URI,refresh_token=refreshToken,access_token=accessToken)
306
  print(result)
307
  access_token = encode_jwt(user_id=user_id,access_token=accessToken)
308
- return {"userId":user_id,"refreshToken":refreshToken,"accessToken":access_token}
309
  return JSONResponse(status_code=401,content="Invalid Sign Up details")
310
 
311
 
@@ -341,6 +341,7 @@ def refresh_access_token(refresh_token:Token, authorization: str = Header(...)):
341
  if is_valid != True: # Example check
342
  raise HTTPException(status_code=401, detail="Invalid token")
343
  new_access_token = create_accessToken(db_uri=MONGO_URI,user_id=decoded_user_id,refresh_token=refresh_token.refreshToken)
 
344
  newly_encoded_access_token = encode_jwt(user_id=decoded_user_id,access_token=new_access_token)
345
  return {"accessToken":newly_encoded_access_token}
346
 
 
291
  result = update_refreshTokenWithPreviouslyUsedAccessToken(db_uri=MONGO_URI,refresh_token=refreshToken,access_token=accessToken)
292
  print(result)
293
  access_token = encode_jwt(user_id=user_id,access_token=accessToken)
294
+ return {"refreshToken":refreshToken,"accessToken":access_token}
295
  return JSONResponse(status_code=401,content="Invalid login details")
296
 
297
 
 
305
  result = update_refreshTokenWithPreviouslyUsedAccessToken(db_uri=MONGO_URI,refresh_token=refreshToken,access_token=accessToken)
306
  print(result)
307
  access_token = encode_jwt(user_id=user_id,access_token=accessToken)
308
+ return {"refreshToken":refreshToken,"accessToken":access_token}
309
  return JSONResponse(status_code=401,content="Invalid Sign Up details")
310
 
311
 
 
341
  if is_valid != True: # Example check
342
  raise HTTPException(status_code=401, detail="Invalid token")
343
  new_access_token = create_accessToken(db_uri=MONGO_URI,user_id=decoded_user_id,refresh_token=refresh_token.refreshToken)
344
+ update_refreshTokenWithPreviouslyUsedAccessToken(db_uri=MONGO_URI,refresh_token=refresh_token.refreshToken,access_token=new_access_token)
345
  newly_encoded_access_token = encode_jwt(user_id=decoded_user_id,access_token=new_access_token)
346
  return {"accessToken":newly_encoded_access_token}
347
 
jwtcoding.py CHANGED
@@ -6,7 +6,7 @@ load_dotenv()
6
  SECRET_KEY = os.getenv("SECRET_KEY")
7
  ALGORITHM = os.getenv("ALGORITHM")
8
 
9
- def encode_jwt(user_id: str, access_token: str, expires_delta: timedelta = timedelta(minutes=130)) -> str:
10
  """Encode user_id and access_token into a JWT."""
11
  payload = {
12
  "user_id": user_id,
 
6
  SECRET_KEY = os.getenv("SECRET_KEY")
7
  ALGORITHM = os.getenv("ALGORITHM")
8
 
9
+ def encode_jwt(user_id: str, access_token: str, expires_delta: timedelta = timedelta(days=130)) -> str:
10
  """Encode user_id and access_token into a JWT."""
11
  payload = {
12
  "user_id": user_id,