Update main.py
Browse files
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
|
50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|