ED/FDEV-CAPI-Handler
1
0
Fork 0
mirror of https://github.com/norohind/FDEV-CAPI-Handler.git synced 2025-06-07 10:53:23 +03:00
Code Issues Projects Releases Wiki Activity

web: improve logic for random token

Browse Source
This commit is contained in:
norohind 2022-09-13 13:40:05 +03:00
parent b6392f8d20
commit d42991fcd7
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
web.py
View File

@ -1,6 +1,7 @@
import falcon import falcon
import waitress import waitress
import json import json
import random
import capi import capi
from capi import capi_authorizer from capi import capi_authorizer
@ -123,15 +124,24 @@ class RandomToken:
@falcon.before(check_secret) @falcon.before(check_secret)
def on_get(self, req: falcon.request.Request, resp: falcon.response.Response): def on_get(self, req: falcon.request.Request, resp: falcon.response.Response):
resp.content_type = falcon.MEDIA_JSON resp.content_type = falcon.MEDIA_JSON
import random
list_users = capi_authorizer.list_all_valid_users() list_users = capi_authorizer.list_all_valid_users()
if len(list_users) == 0: if len(list_users) == 0:
raise falcon.HTTPNotFound(description='No users in DB') raise falcon.HTTPNotFound(description='No users in DB')
random_user = random.choice(list_users) random_user_tokens = None
resp.text = json.dumps(capi_authorizer.get_token_by_state(random_user['state'])) for attempt in range(0, 3):
random_user = random.choice(list_users)
random_user_tokens = capi_authorizer.get_token_by_state(random_user['state'])
if random_user_tokens is None or 'access_token' not in random_user_tokens: # To be sure
continue
if random_user_tokens is None or 'access_token' not in random_user_tokens:
raise falcon.HTTPInternalServerError
resp.text = json.dumps(random_user_tokens)
application = falcon.App() application = falcon.App()