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/main.py b/main.py index 695da3a..21b6dcd 100644 --- a/main.py +++ b/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 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