94 lines
2.4 KiB
Python
94 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Script to create an admin user for the ISP Wiremap application.
|
|
Usage: python scripts/create_admin.py
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Add parent directory to path to import app modules
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
from app.database import SessionLocal
|
|
from app.models.user import User
|
|
from app.utils.password import hash_password
|
|
import getpass
|
|
|
|
|
|
def create_admin_user():
|
|
"""Create an admin user interactively."""
|
|
print("=" * 50)
|
|
print("ISP Wiremap - Create Admin User")
|
|
print("=" * 50)
|
|
print()
|
|
|
|
# Get user input
|
|
username = input("Enter admin username: ").strip()
|
|
if not username:
|
|
print("Error: Username cannot be empty")
|
|
return
|
|
|
|
email = input("Enter admin email: ").strip()
|
|
if not email:
|
|
print("Error: Email cannot be empty")
|
|
return
|
|
|
|
password = getpass.getpass("Enter admin password: ")
|
|
if not password:
|
|
print("Error: Password cannot be empty")
|
|
return
|
|
|
|
password_confirm = getpass.getpass("Confirm password: ")
|
|
if password != password_confirm:
|
|
print("Error: Passwords do not match")
|
|
return
|
|
|
|
# Create database session
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# Check if username already exists
|
|
existing_user = db.query(User).filter(User.username == username).first()
|
|
if existing_user:
|
|
print(f"Error: Username '{username}' already exists")
|
|
return
|
|
|
|
# Check if email already exists
|
|
existing_email = db.query(User).filter(User.email == email).first()
|
|
if existing_email:
|
|
print(f"Error: Email '{email}' already exists")
|
|
return
|
|
|
|
# Create admin user
|
|
hashed_password = hash_password(password)
|
|
admin_user = User(
|
|
username=username,
|
|
email=email,
|
|
password_hash=hashed_password,
|
|
is_admin=True
|
|
)
|
|
|
|
db.add(admin_user)
|
|
db.commit()
|
|
db.refresh(admin_user)
|
|
|
|
print()
|
|
print("=" * 50)
|
|
print("Admin user created successfully!")
|
|
print(f"Username: {admin_user.username}")
|
|
print(f"Email: {admin_user.email}")
|
|
print(f"User ID: {admin_user.id}")
|
|
print(f"Is Admin: {admin_user.is_admin}")
|
|
print("=" * 50)
|
|
|
|
except Exception as e:
|
|
db.rollback()
|
|
print(f"Error creating admin user: {e}")
|
|
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
create_admin_user()
|