65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Check and create admin user"""
|
|
|
|
import asyncio
|
|
import sys
|
|
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy import select
|
|
import bcrypt
|
|
from datetime import datetime, timezone
|
|
|
|
sys.path.append('/media/pts/Website/PromptTech_Solution_Site/backend')
|
|
from models import User, UserRole
|
|
from database import DATABASE_URL
|
|
|
|
async def check_and_create_admin():
|
|
engine = create_async_engine(DATABASE_URL, echo=False)
|
|
async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
async with async_session() as session:
|
|
# Check for existing admin
|
|
result = await session.execute(select(User))
|
|
users = result.scalars().all()
|
|
|
|
print(f"\n📊 Current users in database:")
|
|
for user in users:
|
|
print(f" - {user.email} (Role: {user.role.value})")
|
|
|
|
# Check if admin@prompttech.com exists
|
|
result = await session.execute(
|
|
select(User).where(User.email == "admin@prompttech.com")
|
|
)
|
|
admin = result.scalar_one_or_none()
|
|
|
|
if admin:
|
|
print(f"\n✅ Admin user already exists: admin@prompttech.com")
|
|
print(f" If you can't login, the password should be: admin123")
|
|
else:
|
|
print(f"\n⚠️ Admin user not found. Creating new admin...")
|
|
|
|
# Create admin user
|
|
email = "admin@prompttech.com"
|
|
password = "admin123"
|
|
name = "Admin User"
|
|
|
|
hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
|
|
|
|
admin_user = User(
|
|
email=email,
|
|
name=name,
|
|
password=hashed_password,
|
|
role=UserRole.ADMIN,
|
|
created_at=datetime.now(timezone.utc)
|
|
)
|
|
|
|
session.add(admin_user)
|
|
await session.commit()
|
|
|
|
print(f"\n✅ Admin user created!")
|
|
print(f" Email: {email}")
|
|
print(f" Password: {password}")
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(check_and_create_admin())
|