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

web: fix /random_token when no users in DB

Browse Source
This commit is contained in:
norohind 2021-12-08 01:07:14 +03:00
parent 317dff9e3b
commit 2b134d26f4
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web.py
View File

@ -95,8 +95,16 @@ class RandomToken:
# for legacy reasons # for legacy reasons
@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
import random import random
random_state = random.choice(capi_authorizer.list_all_users()) list_users = capi_authorizer.list_all_users()
if len(list_users) == 0:
raise falcon.HTTPNotFound(description='No users in DB')
random_user = random.choice(list_users)
resp.text = json.dumps(capi_authorizer.get_token_by_state(random_user['state']))
application = falcon.App() application = falcon.App()