okeowo1014 commited on
Commit
4c6e084
1 Parent(s): b5ff2ac

Update app/routers/user.py

Browse files
Files changed (1) hide show
  1. 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("/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
 
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