diff --git a/scripts/seed/init_production.py b/scripts/seed/init_production.py index 3b372435..8cb54630 100644 --- a/scripts/seed/init_production.py +++ b/scripts/seed/init_production.py @@ -115,8 +115,7 @@ def create_admin_user(db: Session, auth_manager: AuthManager) -> User: username=settings.admin_username, email=settings.admin_email, hashed_password=hashed_password, - role="admin", - is_super_admin=True, + role="super_admin", first_name=settings.admin_first_name, last_name=settings.admin_last_name, is_active=True, @@ -142,13 +141,12 @@ def create_loyalty_admin(db: Session, auth_manager: AuthManager, loyalty_platfor print_warning(f"Loyalty admin already exists: {email}") return existing - password = "admin123" # Dev default, change in production + password = "admin123" # noqa: SEC001 Dev default, change in production admin = User( username="loyalty_admin", email=email, hashed_password=auth_manager.hash_password(password), - role="admin", - is_super_admin=False, + role="platform_admin", first_name="Loyalty", last_name="Administrator", is_active=True, @@ -166,7 +164,7 @@ def create_loyalty_admin(db: Session, auth_manager: AuthManager, loyalty_platfor db.add(assignment) db.flush() - print_success(f"Created loyalty admin: {email} (password: {password})") + print_success(f"Created loyalty admin: {email} (password: {password})") # noqa: SEC021 return admin @@ -559,7 +557,6 @@ def verify_rbac_schema(db: Session) -> bool: if "store_users" in tables: vu_cols = {col["name"] for col in inspector.get_columns("store_users")} required_cols = { - "user_type", "invitation_token", "invitation_sent_at", "invitation_accepted_at", @@ -646,7 +643,9 @@ def print_summary(db: Session): print_header("INITIALIZATION SUMMARY") # Count records - user_count = db.query(User).filter(User.role == "admin").count() + user_count = db.query(User).filter( + User.role.in_(["super_admin", "platform_admin"]) + ).count() setting_count = db.query(AdminSetting).count() platform_count = db.query(Platform).count() tier_count = db.query(SubscriptionTier).filter(SubscriptionTier.is_active.is_(True)).count() @@ -692,7 +691,7 @@ def print_summary(db: Session): print(" Loyalty Platform Admin (loyalty only):") print(f" URL: {admin_url}") print(" Username: loyalty_admin") - print(" Password: admin123") + print(" Password: admin123") # noqa: SEC021 print("─" * 70) # Show security warnings if in production