add token response

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))

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Depends
+from fastapi import FastAPI, Depends, status
 from contextlib import asynccontextmanager
 import database
 import models
@@ -25,7 +25,7 @@ async def me(user: models.User = Depends(database.get_user_by_token)):
     return user
 
 
-@app.post("/users")
+@app.post("/register", status_code=status.HTTP_201_CREATED)
 async def register(user: models.User):
     database.register(user)
     return user

models.py

@@ -4,3 +4,7 @@ from pydantic import BaseModel
 class User(BaseModel):
     name: str
     password: str
+
+
+class TokenResponse(BaseModel):
+    token: str