Make sure 'code' returns None instead of crashing if key is missing.

andersnauman · andersnauman · commit a62711075888 · 2023-11-03T17:07:05.000+01:00
diff --git a/authlib/integrations/flask_client/apps.py b/authlib/integrations/flask_client/apps.py
@@ -85,12 +85,12 @@ def authorize_access_token(self, **kwargs):
                 raise OAuthError(error=error, description=description)
 
             params = {
-                'code': request.args['code'],
+                'code': request.args.get('code'),
                 'state': request.args.get('state'),
             }
         else:
             params = {
-                'code': request.form['code'],
+                'code': request.form.get('code'),
                 'state': request.form.get('state'),
             }
 

Original file line number	Diff line number	Diff line change
`@@ -85,12 +85,12 @@ def authorize_access_token(self, **kwargs):`
`85`	`85`	`raise OAuthError(error=error, description=description)`
`86`	`86`
`87`	`87`	`params = {`
`88`		`- 'code': request.args['code'],`
	`88`	`+ 'code': request.args.get('code'),`
`89`	`89`	`'state': request.args.get('state'),`
`90`	`90`	`}`
`91`	`91`	`else:`
`92`	`92`	`params = {`
`93`		`- 'code': request.form['code'],`
	`93`	`+ 'code': request.form.get('code'),`
`94`	`94`	`'state': request.form.get('state'),`
`95`	`95`	`}`
`96`	`96`