Gregniuki commited on
Commit
8338c06
·
1 Parent(s): ac01fc9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +12 -4
main.py CHANGED
@@ -5,7 +5,7 @@ from fastapi.requests import Request
5
  from fastapi.responses import HTMLResponse
6
  from fastapi.templating import Jinja2Templates
7
  from sqlalchemy.orm import Session
8
- from auth import verify_token, oauth2_scheme, auth_views, register, UserCreate
9
  from database import get_db, get_user_by_email
10
  #import auth
11
  #import tts
@@ -45,9 +45,17 @@ async def registration_successful(request: Request):
45
 
46
 
47
  @app.get("/verify/{verification_token}", response_class=HTMLResponse)
48
- async def verify_email(verification_token: str, request: Request):
49
- # Perform verification and return an appropriate template
50
- return templates.TemplateResponse("verify.html", {"request": request})
 
 
 
 
 
 
 
 
51
 
52
  # User authentication (protected route)
53
  @app.get("/protected", response_model=str)
 
5
  from fastapi.responses import HTMLResponse
6
  from fastapi.templating import Jinja2Templates
7
  from sqlalchemy.orm import Session
8
+ from auth import verify_token, oauth2_scheme, auth_views, register, UserCreate, verify_email
9
  from database import get_db, get_user_by_email
10
  #import auth
11
  #import tts
 
45
 
46
 
47
  @app.get("/verify/{verification_token}", response_class=HTMLResponse)
48
+ async def verify_email(verification_token: str, request: Request, db: Session = Depends(get_db)):
49
+ # Perform email verification
50
+ verification_result = verify_email(verification_token, db)
51
+
52
+ # Handle a successful verification
53
+ if "message" in verification_result:
54
+ # Redirect the user to the protected area
55
+ return RedirectResponse("/protected")
56
+
57
+ # Handle any other cases, such as errors
58
+ return HTTPException(status_code=400, detail="Verification failed")
59
 
60
  # User authentication (protected route)
61
  @app.get("/protected", response_model=str)