42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import django
|
||
|
|
from getpass import getpass
|
||
|
|
|
||
|
|
def main():
|
||
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "pytv.settings")
|
||
|
|
django.setup()
|
||
|
|
|
||
|
|
from core.models import AppUser
|
||
|
|
|
||
|
|
print("--- Create PYTV User ---")
|
||
|
|
try:
|
||
|
|
username = input("Username: ").strip()
|
||
|
|
if not username:
|
||
|
|
print("Username is required!")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
email = input("Email: ").strip()
|
||
|
|
password = getpass("Password: ")
|
||
|
|
|
||
|
|
is_admin_str = input("Is Admin? (y/N): ").strip().lower()
|
||
|
|
is_admin = is_admin_str in ['y', 'yes']
|
||
|
|
|
||
|
|
if AppUser.objects.filter(username=username).exists():
|
||
|
|
print(f"User '{username}' already exists!")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if is_admin:
|
||
|
|
AppUser.objects.create_superuser(username=username, email=email, password=password)
|
||
|
|
print(f"Superuser '{username}' created successfully!")
|
||
|
|
else:
|
||
|
|
AppUser.objects.create_user(username=username, email=email, password=password)
|
||
|
|
print(f"User '{username}' created successfully!")
|
||
|
|
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
print("\nOperation cancelled.")
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|