Use a generic JWTClaim class for respones

1 files changed, 8 insertions, 5 deletions

diff --git a/thallium-backend/src/dto/login.py b/thallium-backend/src/dto/login.py
index 706299c..bd1e719 100644
--- a/thallium-backend/src/dto/login.py
+++ b/thallium-backend/src/dto/login.py
@@ -1,17 +1,21 @@
 from pydantic import BaseModel
 
 
+class JWTClaim(BaseModel):
+    """A response with a JWT present."""
+
+    jwt: str
+
+
 class VoucherLogin(BaseModel):
     """The data needed to login with a voucher."""
 
     voucher_code: str
 
 
-class VoucherClaim(VoucherLogin):
+class VoucherClaim(VoucherLogin, JWTClaim):
     """A JWT for a verified voucher."""
 
-    jwt: str
-
 
 class UserLogin(BaseModel):
     """The data needed to login as a user."""
@@ -20,13 +24,12 @@ class UserLogin(BaseModel):
     password: str
 
 
-class UserClaim(BaseModel):
+class UserClaim(JWTClaim, BaseModel):
     """The response given to a successfully logged in user."""
 
     username: str
     permissions: int
     require_password_change: bool
-    jwt: str
 
 
 class PasswordReset(BaseModel):