Update main.py

This commit is contained in:
Saifeddine ALOUI 2024-01-17 18:21:41 +01:00 committed by GitHub
parent dd1f43a70e
commit 416b6d4a65
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 13 deletions

View File

@ -77,6 +77,7 @@ def main():
self.proxy()
def _validate_user_and_key(self):
try:
# Extract the bearer token from the headers
auth_header = self.headers.get('Authorization')
if not auth_header or not auth_header.startswith('Bearer '):
@ -91,7 +92,8 @@ def main():
else:
self.user = "unknown"
return False
except:
return False
def proxy(self):
if not self._validate_user_and_key():
ASCIIColors.red(f'User is not authorized')