add verifcation message

api/main.py

@@ -182,7 +182,10 @@ async def login_for_access_token(request: Request, form_data: OAuth2PasswordRequ
     Authorize.set_access_cookies(access_token)
     Authorize.set_refresh_cookies(refresh_token)
     #return templates.TemplateResponse("index.html", {"request": request, "msg": 'Login'})
-    return {"access_token": access_token, "token_type": "bearer"}
+    if util.user.email_veri_pass(user.id):
+        return {"access_token": access_token, "token_type": "bearer",'veri':'ok'}
+    else:
+        return {"access_token": access_token, "token_type": "bearer",'veri':'fail'}
 
 
 @app.post("/token")

api/util/user.py

@@ -76,3 +76,10 @@ def get_user_id( username):
     return str(first(db.query('SELECT COUNT(*) FROM history_input WHERE user_id ='+str(user_obj['id'])))['COUNT(*)'])
     
 
+def email_veri_pass(id):
+    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
+    user_obj = first(db.query('SELECT * FROM register_veri_code where user_id ="'+str(id)+'"'))
+    if user_obj == None:
+        return False
+    else:
+        return True