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

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')