skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed oauth error (#223)

Browse source
This commit is contained in:
SocietyMoment 2022-03-17 03:14:46 -04:00 • committed by GitHub
parent dc0b86f371
commit b5ce29bbbd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
files/routes/oauth.py
View file

@ -5,6 +5,7 @@ from files.helpers.const import *
from files.classes import * from files.classes import *
from flask import * from flask import *
from files.__main__ import app, limiter from files.__main__ import app, limiter
import sqlalchemy.exc
@app.get("/authorize") @app.get("/authorize")
@auth_required @auth_required
@ -24,11 +25,15 @@ def authorize(v):
application = g.db.query(OauthApp).filter_by(client_id=client_id).one_or_none() application = g.db.query(OauthApp).filter_by(client_id=client_id).one_or_none()
if not application: return {"oauth_error": "Invalid `client_id`"}, 401 if not application: return {"oauth_error": "Invalid `client_id`"}, 401
access_token = secrets.token_urlsafe(128)[:128] access_token = secrets.token_urlsafe(128)[:128]
new_auth = ClientAuth(oauth_client = application.id, user_id = v.id, access_token=access_token)
g.db.add(new_auth) try:
new_auth = ClientAuth(oauth_client = application.id, user_id = v.id, access_token=access_token)
g.db.commit() g.db.add(new_auth)
g.db.commit()
except sqlalchemy.exc.IntegrityError:
g.db.rollback()
old_auth = g.db.query(ClientAuth).filter_by(oauth_client = application.id, user_id = v.id).one()
access_token = old_auth.access_token
return redirect(f"{application.redirect_uri}?token={access_token}") return redirect(f"{application.redirect_uri}?token={access_token}")