Jelajahi Sumber

add login when registered

ming 3 tahun lalu
induk
melakukan
f0fab2e012
1 mengubah file dengan 12 tambahan dan 0 penghapusan
  1. 12 0
      api/main.py

+ 12 - 0
api/main.py

@@ -202,6 +202,18 @@ async def register_old(request: Request):
     user_obj = first(db.query('SELECT * FROM users where username ="'+str(user.username)+'"'))
     if user_obj == None:
         user_register(user)
+        access_token_expires = timedelta(days=ACCESS_TOKEN_EXPIRE_DAYS)
+        access_token = create_access_token(
+            data={"sub": user.username}, expires_delta=access_token_expires
+        )
+        table = db['users']
+        user.token = access_token
+        table.update(dict(user), ['username'])
+        expires = timedelta(days=ACCESS_TOKEN_EXPIRE_DAYS)
+        access_token = Authorize.create_access_token(subject=user.username, expires_time=expires)
+        refresh_token = Authorize.create_refresh_token(subject=user.username, expires_time =expires)
+        Authorize.set_access_cookies(access_token)
+        Authorize.set_refresh_cookies(refresh_token)
         return templates.TemplateResponse("make_video.html", {"request": request, "success": True},status_code=status.HTTP_302_FOUND)
         #return templates.TemplateResponse("login.html", {'request': request,"success": True}, status_code=status.HTTP_302_FOUND)
     else: