alex/nova-api
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-api.git synced 2024-11-25 18:23:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update core.py

Browse source
This commit is contained in:
Game_Time 2023-08-13 14:16:23 +05:00 committed by GitHub
parent bd4fc0ad86
commit 60a89146da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
api/core.py
View file

@ -12,7 +12,14 @@ from dotenv import load_dotenv
load_dotenv() load_dotenv()
router = fastapi.APIRouter(tags=['core']) router = fastapi.APIRouter(tags=['core'])
async def check_core_auth(request): async def check_core_auth(request):
"""
### Checks the request's auth
Auth is taken from environment variable `CORE_API_KEY`
"""
received_auth = request.headers.get('Authorization') received_auth = request.headers.get('Authorization')
if received_auth != os.getenv('CORE_API_KEY'): if received_auth != os.getenv('CORE_API_KEY'):
@ -20,14 +27,12 @@ async def check_core_auth(request):
@router.get('/users') @router.get('/users')
async def get_users(discord_id: int, incoming_request: fastapi.Request): async def get_users(discord_id: int, incoming_request: fastapi.Request):
auth_error = await check_core_auth(incoming_request) auth = await check_core_auth(incoming_request)
if auth:
if auth_error:
return auth_error return auth_error
user = await users.by_discord_id(discord_id) # Get user by discord ID
if not await users.by_discord_id(discord_id):
if not user:
return fastapi.Response(status_code=404, content='User not found.') return fastapi.Response(status_code=404, content='User not found.')
return user return user