ZK-AI
/
ollama_proxy_server
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update main.py

This commit is contained in:
Saifeddine ALOUI 2024-01-16 10:24:30 +01:00 committed by GitHub
parent e3d5b52068
commit 18261b625b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ollama_proxy_server/main.py
View File

@ -44,19 +44,19 @@ def main():
authorized_users = get_authorized_users(args.users_list) authorized_users = get_authorized_users(args.users_list)
class RequestHandler(BaseHTTPRequestHandler): class RequestHandler(BaseHTTPRequestHandler):
def add_access_log_entry(user, ip_address): def add_access_log_entry(self, user, ip_address, access):
log_file_path = Path(sys.argv[1]) log_file_path = Path(sys.argv[1])
if not log_file_path.exists(): if not log_file_path.exists():
with open(log_file_path, mode='w', newline='') as csvfile: with open(log_file_path, mode='w', newline='') as csvfile:
fieldnames = ['time_stamp', 'user_name', 'ip_address'] fieldnames = ['time_stamp', 'user_name', 'ip_address','access']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames) writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader() writer.writeheader()
with open(log_file_path, mode='a', newline='') as csvfile: with open(log_file_path, mode='a', newline='') as csvfile:
fieldnames = ['time_stamp', 'user_name', 'ip_address'] fieldnames = ['time_stamp', 'user_name', 'ip_address']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames) writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
row = {'time_stamp': str(datetime.datetime.now()), 'user_name': user, 'ip_address': ip_address} row = {'time_stamp': str(datetime.datetime.now()), 'user_name': user, 'ip_address': ip_address, 'access': access}
writer.writerow(row) writer.writerow(row)
def _send_response(self, response): def _send_response(self, response):
@ -87,6 +87,7 @@ def main():
def proxy(self): def proxy(self):
if not self._validate_user_and_key(): if not self._validate_user_and_key():
ASCIIColors.red(f'User is not authorized') ASCIIColors.red(f'User is not authorized')
# self.add_access_log_entry("unknown")
self.send_response(403) self.send_response(403)
self.end_headers() self.end_headers()
return return