Add role field to JWTUser schema

Wilcolab · Oct 15, 2024 · 9149a2a · 9149a2a
1 parent d2f1f59
commit 9149a2a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/backend/app/models/schemas/jwt.py b/backend/app/models/schemas/jwt.py
@@ -11,3 +11,4 @@ class JWTMeta(BaseModel):
 
 class JWTUser(BaseModel):
     username: str
+    role: UserRole
diff --git a/backend/app/services/jwt.py b/backend/app/services/jwt.py
@@ -26,7 +26,7 @@ def create_jwt_token(
 
 def create_access_token_for_user(user: User, secret_key: str) -> str:
     return create_jwt_token(
-        jwt_content=JWTUser(username=user.username).dict(),
+        jwt_content=JWTUser(username=user.username, role=user.role).dict(),
         secret_key=secret_key,
         expires_delta=timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES),
     )
Original file line number	Diff line number	Diff line change
Expand Up		@@ -11,3 +11,4 @@ class JWTMeta(BaseModel):

		class JWTUser(BaseModel):
		username: str
		role: UserRole