allow secret both in cookies and in headers

This commit is contained in:
norohind 2021-12-08 18:10:09 +03:00
parent 71fa51f69a
commit 17efd41457
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1

7
web.py
View File

@ -11,9 +11,12 @@ logger.propagate = False
def check_secret(req: falcon.request.Request, resp: falcon.response.Response, resource, params) -> None:
cookies_secret = req.headers.get('AUTH')
header_secret = req.headers.get('AUTH') # for legacy reasons
if cookies_secret != config.access_key:
cookies_secret = req.get_cookie_values('key')
if header_secret != config.access_key:
if cookies_secret is None or cookies_secret[0] != config.access_key:
raise falcon.HTTPForbidden