k-space/doorboy-proxy
11
0
Fork 0
Code Issues 3 Pull Requests 1 Projects Activity

Fix key check
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Details

Browse Source
This commit is contained in:
Madis Mägi
2023-08-01 23:02:58 +03:00
parent 37027baf3e
commit 0fd17b7cbb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/doorboy-proxy.py
View File

@@ -31,7 +31,7 @@ async def setup_db(app, loop):
@app.route("/allowed")
async def view_doorboy_uids(request):
key = request.headers.get("KEY")
if key not in [DOORBOY_SECRET_FLOOR, DOORBOY_SECRET_WORKSHOP]:
if not key or key not in [DOORBOY_SECRET_FLOOR, DOORBOY_SECRET_WORKSHOP]:
return text("how about no")
if key == DOORBOY_SECRET_FLOOR:
@@ -56,7 +56,7 @@ async def view_doorboy_uids(request):
@app.route("/longpoll", stream=True)
async def view_longpoll(request):
response = await request.respond(content_type="text/event-stream")
if key not in [DOORBOY_SECRET_FLOOR, DOORBOY_SECRET_WORKSHOP]:
if not key or key not in [DOORBOY_SECRET_FLOOR, DOORBOY_SECRET_WORKSHOP]:
return text("Invalid token")
await response.send("data: response-generator-started\n\n")