From 3240d5227a8328850d9d49cb56d3c9e161d734b8 Mon Sep 17 00:00:00 2001
From: Zvonimir Rudinski <zvonki12@gmail.com>
Date: Sun, 16 Nov 2025 17:02:10 +0100
Subject: [PATCH] add token response

---
 database.py | 4 ++--
 models.py   | 4 ++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/database.py b/database.py
index bd03e76..3755552 100644
--- a/database.py
+++ b/database.py
@@ -1,4 +1,4 @@
-from models import User
+from models import User, TokenResponse
 from settings import settings
 from fastapi import HTTPException, status, Request
 import sqlite3
@@ -101,4 +101,4 @@ def login(user: User) -> str:
             detail="Invalid credentials"
         )
 
-    return security.sign_jwt(row)
+    return TokenResponse(token=security.sign_jwt(row))
diff --git a/models.py b/models.py
index bc275ee..2d11557 100644
--- a/models.py
+++ b/models.py
@@ -4,3 +4,7 @@ from pydantic import BaseModel
 class User(BaseModel):
     name: str
     password: str
+
+
+class TokenResponse(BaseModel):
+    token: str