diff --git a/ollama_proxy_server/add_user.py b/ollama_proxy_server/add_user.py new file mode 100644 index 0000000..5dcb6bc --- /dev/null +++ b/ollama_proxy_server/add_user.py @@ -0,0 +1,27 @@ +import sys +import random +from getpass import getuser +from pathlib import Path + +def generate_key(length=10): + """Generate a random key of given length""" + chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*()-_=+[]{}|;:,.<>?/~' + return ''.join(random.choice(chars) for _ in range(length)) + +def add_user(users_list=None): + """Add a new user to the users list file""" + user_name = input('Enter your username: ') + key = generate_key() + print(f'Your key is: {key}') + if not users_list or not users_list.exists(): + users_list = Path(users_list) if users_list else Path('authorized_users.txt') + users_list.touch(exist_ok=True) + with open(users_list, 'a') as f: + f.write(f'{user_name}: {key}\n') + print(f'User {user_name} added to the authorized users list') + +if __name__ == '__main__': + if len(sys.argv) > 1 and sys.argv[1] == '--users_list': + add_user(Path(sys.argv[2])) + else: + add_user()