Spaces:
Sleeping
Sleeping
Update app/routers/user.py
Browse files- app/routers/user.py +30 -30
app/routers/user.py
CHANGED
@@ -1,30 +1,30 @@
|
|
1 |
-
# fastapi_crud/app/routers/user.py
|
2 |
-
from fastapi import APIRouter, Depends, HTTPException
|
3 |
-
from sqlalchemy.orm import Session
|
4 |
-
from fastapi.security import OAuth2PasswordRequestForm
|
5 |
-
from app import crud, schemas, database, auth
|
6 |
-
router = APIRouter()
|
7 |
-
|
8 |
-
@router.post("/users/", response_model=schemas.User)
|
9 |
-
def create_user(user: schemas.UserCreate, db: Session = Depends(database.get_db)):
|
10 |
-
db_user = crud.get_user_by_username(db, username=user.username)
|
11 |
-
if db_user:
|
12 |
-
raise HTTPException(status_code=400, detail="Username already registered")
|
13 |
-
return crud.create_user(db=db, user=user)
|
14 |
-
|
15 |
-
# status_code=status.HTTP_401_UNAUTHORIZED
|
16 |
-
@router.post("/
|
17 |
-
async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends(), db: Session = Depends(database.get_db)):
|
18 |
-
user = auth.authenticate_user(db, form_data.username, form_data.password)
|
19 |
-
if not user:
|
20 |
-
raise HTTPException(
|
21 |
-
status_code=401,
|
22 |
-
detail="Incorrect username or password",
|
23 |
-
headers={"WWW-Authenticate": "Bearer"},
|
24 |
-
)
|
25 |
-
access_token = auth.create_access_token(data={"sub": user.username})
|
26 |
-
return {"access_token": access_token, "token_type": "bearer"}
|
27 |
-
|
28 |
-
@router.get("/users/me/", response_model=schemas.User)
|
29 |
-
async def read_users_me(current_user: schemas.User = Depends(auth.get_current_user)):
|
30 |
-
return current_user
|
|
|
1 |
+
# fastapi_crud/app/routers/user.py
|
2 |
+
from fastapi import APIRouter, Depends, HTTPException
|
3 |
+
from sqlalchemy.orm import Session
|
4 |
+
from fastapi.security import OAuth2PasswordRequestForm
|
5 |
+
from app import crud, schemas, database, auth
|
6 |
+
router = APIRouter()
|
7 |
+
|
8 |
+
@router.post("/users/", response_model=schemas.User)
|
9 |
+
def create_user(user: schemas.UserCreate, db: Session = Depends(database.get_db)):
|
10 |
+
db_user = crud.get_user_by_username(db, username=user.username)
|
11 |
+
if db_user:
|
12 |
+
raise HTTPException(status_code=400, detail="Username already registered")
|
13 |
+
return crud.create_user(db=db, user=user)
|
14 |
+
|
15 |
+
# status_code=status.HTTP_401_UNAUTHORIZED
|
16 |
+
@router.post("/get_token", response_model=dict)
|
17 |
+
async def login_for_access_token(form_data: OAuth2PasswordRequestForm = Depends(), db: Session = Depends(database.get_db)):
|
18 |
+
user = auth.authenticate_user(db, form_data.username, form_data.password)
|
19 |
+
if not user:
|
20 |
+
raise HTTPException(
|
21 |
+
status_code=401,
|
22 |
+
detail="Incorrect username or password",
|
23 |
+
headers={"WWW-Authenticate": "Bearer"},
|
24 |
+
)
|
25 |
+
access_token = auth.create_access_token(data={"sub": user.username})
|
26 |
+
return {"access_token": access_token, "token_type": "bearer"}
|
27 |
+
|
28 |
+
@router.get("/users/me/", response_model=schemas.User)
|
29 |
+
async def read_users_me(current_user: schemas.User = Depends(auth.get_current_user)):
|
30 |
+
return current_user
|