88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
import os
|
|
import subprocess
|
|
from datetime import datetime
|
|
|
|
users_file_path = '/etc/hysteria/users/users.json'
|
|
|
|
def load_users():
|
|
with open(users_file_path, 'r') as file:
|
|
return json.load(file)
|
|
|
|
def save_users(users):
|
|
with open(users_file_path, 'w') as file:
|
|
json.dump(users, file, indent=4)
|
|
|
|
def generate_password():
|
|
result = subprocess.run(['pwgen', '-s', '32', '1'], capture_output=True, text=True)
|
|
return result.stdout.strip()
|
|
|
|
def list_users(users):
|
|
for i, user in enumerate(users.keys(), start=1):
|
|
print(f"{i}. {user}")
|
|
|
|
def edit_user(users):
|
|
list_users(users)
|
|
try:
|
|
choice = int(input("Enter the number of the user to edit: "))
|
|
username = list(users.keys())[choice - 1]
|
|
except (ValueError, IndexError):
|
|
print("Invalid choice.")
|
|
return
|
|
|
|
print(f"Editing user: {username}")
|
|
|
|
change_password = input(f"Change password? (current: {users[username]['password']}) [y/N]: ").lower() == 'y'
|
|
if change_password:
|
|
new_password = generate_password()
|
|
users[username]['password'] = new_password
|
|
print(f"New password: {new_password}")
|
|
|
|
while True:
|
|
max_download_gb = input(f"Enter new max download bytes in GB (current: {users[username]['max_download_bytes'] // (1024 ** 3)} GB): ")
|
|
if max_download_gb.isdigit():
|
|
users[username]['max_download_bytes'] = int(max_download_gb) * (1024 ** 3)
|
|
break
|
|
else:
|
|
print("Invalid input. Please enter a valid number.")
|
|
|
|
while True:
|
|
expiration_days = input(f"Enter new expiration days (current: {users[username]['expiration_days']}): ")
|
|
if expiration_days.isdigit():
|
|
users[username]['expiration_days'] = int(expiration_days)
|
|
break
|
|
else:
|
|
print("Invalid input. Please enter a valid number.")
|
|
|
|
blocked = input(f"Blocked? (current: {users[username]['blocked']}) [true/false]: ").lower()
|
|
if blocked:
|
|
users[username]['blocked'] = blocked == 'true'
|
|
|
|
change_date = input(f"Change account creation date to today? (current: {users[username]['account_creation_date']}) [y/N]: ").lower() == 'y'
|
|
if change_date:
|
|
users[username]['account_creation_date'] = datetime.today().strftime('%Y-%m-%d')
|
|
|
|
def main():
|
|
if not os.path.exists(users_file_path):
|
|
print(f"File {users_file_path} does not exist.")
|
|
return
|
|
|
|
users = load_users()
|
|
|
|
while True:
|
|
print("1. Edit user")
|
|
print("2. Exit")
|
|
choice = input("Enter your choice: ")
|
|
|
|
if choice == "1":
|
|
edit_user(users)
|
|
save_users(users)
|
|
elif choice == "2":
|
|
return
|
|
else:
|
|
print("Invalid choice. Please try again.")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|