218 lines
7.4 KiB
Python
218 lines
7.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Seed initial admin user, roles, and menus
|
|
|
|
Creates:
|
|
1. Two roles: admin and user
|
|
2. Admin user: cosmo / cosmo
|
|
3. Admin menu structure
|
|
|
|
Usage:
|
|
python scripts/seed_admin.py
|
|
"""
|
|
import asyncio
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
|
|
from sqlalchemy import select
|
|
from app.database import AsyncSessionLocal
|
|
from app.models.db import User, Role, Menu, RoleMenu
|
|
import bcrypt
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def hash_password(password: str) -> str:
|
|
"""Hash password using bcrypt"""
|
|
return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
|
|
|
|
|
|
async def main():
|
|
"""Seed admin data"""
|
|
async with AsyncSessionLocal() as session:
|
|
try:
|
|
# 1. Create roles
|
|
logger.info("Creating roles...")
|
|
|
|
# Check if roles already exist
|
|
result = await session.execute(select(Role))
|
|
existing_roles = result.scalars().all()
|
|
|
|
if existing_roles:
|
|
logger.info(f"Roles already exist: {[r.name for r in existing_roles]}")
|
|
admin_role = next((r for r in existing_roles if r.name == 'admin'), None)
|
|
user_role = next((r for r in existing_roles if r.name == 'user'), None)
|
|
else:
|
|
admin_role = Role(
|
|
name='admin',
|
|
display_name='管理员',
|
|
description='系统管理员，拥有所有权限'
|
|
)
|
|
user_role = Role(
|
|
name='user',
|
|
display_name='普通用户',
|
|
description='普通用户，仅有基本访问权限'
|
|
)
|
|
session.add(admin_role)
|
|
session.add(user_role)
|
|
await session.flush()
|
|
logger.info(f"✓ Created roles: admin, user")
|
|
|
|
# 2. Create admin user
|
|
logger.info("Creating admin user...")
|
|
|
|
# Check if admin user already exists
|
|
result = await session.execute(
|
|
select(User).where(User.username == 'cosmo')
|
|
)
|
|
existing_user = result.scalar_one_or_none()
|
|
|
|
if existing_user:
|
|
logger.info(f"Admin user 'cosmo' already exists (id={existing_user.id})")
|
|
admin_user = existing_user
|
|
else:
|
|
admin_user = User(
|
|
username='cosmo',
|
|
password_hash=hash_password('cosmo'),
|
|
email='admin@cosmo.com',
|
|
full_name='Cosmo Administrator',
|
|
is_active=True
|
|
)
|
|
session.add(admin_user)
|
|
await session.flush()
|
|
|
|
# Assign admin role to user using direct insert to avoid lazy loading
|
|
from app.models.db.user import user_roles
|
|
await session.execute(
|
|
user_roles.insert().values(
|
|
user_id=admin_user.id,
|
|
role_id=admin_role.id
|
|
)
|
|
)
|
|
await session.flush()
|
|
|
|
logger.info(f"✓ Created admin user: cosmo / cosmo")
|
|
|
|
# 3. Create admin menus
|
|
logger.info("Creating admin menus...")
|
|
|
|
# Check if menus already exist
|
|
result = await session.execute(select(Menu))
|
|
existing_menus = result.scalars().all()
|
|
|
|
if existing_menus:
|
|
logger.info(f"Menus already exist ({len(existing_menus)} menus)")
|
|
else:
|
|
# Root menu items
|
|
dashboard_menu = Menu(
|
|
name='dashboard',
|
|
title='控制台',
|
|
icon='dashboard',
|
|
path='/admin/dashboard',
|
|
component='admin/Dashboard',
|
|
sort_order=1,
|
|
is_active=True,
|
|
description='系统控制台'
|
|
)
|
|
|
|
data_management_menu = Menu(
|
|
name='data_management',
|
|
title='数据管理',
|
|
icon='database',
|
|
path=None, # Parent menu, no direct path
|
|
component=None,
|
|
sort_order=2,
|
|
is_active=True,
|
|
description='数据管理模块'
|
|
)
|
|
|
|
session.add(dashboard_menu)
|
|
session.add(data_management_menu)
|
|
await session.flush()
|
|
|
|
# Sub-menu items under data_management
|
|
celestial_bodies_menu = Menu(
|
|
parent_id=data_management_menu.id,
|
|
name='celestial_bodies',
|
|
title='天体数据列表',
|
|
icon='planet',
|
|
path='/admin/celestial-bodies',
|
|
component='admin/CelestialBodies',
|
|
sort_order=1,
|
|
is_active=True,
|
|
description='查看和管理天体数据'
|
|
)
|
|
|
|
static_data_menu = Menu(
|
|
parent_id=data_management_menu.id,
|
|
name='static_data',
|
|
title='静态数据列表',
|
|
icon='data',
|
|
path='/admin/static-data',
|
|
component='admin/StaticData',
|
|
sort_order=2,
|
|
is_active=True,
|
|
description='查看和管理静态数据（星座、星系等）'
|
|
)
|
|
|
|
nasa_data_menu = Menu(
|
|
parent_id=data_management_menu.id,
|
|
name='nasa_data',
|
|
title='NASA数据下载管理',
|
|
icon='download',
|
|
path='/admin/nasa-data',
|
|
component='admin/NasaData',
|
|
sort_order=3,
|
|
is_active=True,
|
|
description='管理NASA Horizons数据下载'
|
|
)
|
|
|
|
session.add(celestial_bodies_menu)
|
|
session.add(static_data_menu)
|
|
session.add(nasa_data_menu)
|
|
await session.flush()
|
|
|
|
logger.info(f"✓ Created {5} menu items")
|
|
|
|
# 4. Assign all menus to admin role
|
|
logger.info("Assigning menus to admin role...")
|
|
all_menus = [
|
|
dashboard_menu,
|
|
data_management_menu,
|
|
celestial_bodies_menu,
|
|
static_data_menu,
|
|
nasa_data_menu
|
|
]
|
|
|
|
for menu in all_menus:
|
|
role_menu = RoleMenu(role_id=admin_role.id, menu_id=menu.id)
|
|
session.add(role_menu)
|
|
|
|
await session.flush()
|
|
logger.info(f"✓ Assigned {len(all_menus)} menus to admin role")
|
|
|
|
await session.commit()
|
|
|
|
logger.info("\n" + "=" * 60)
|
|
logger.info("Admin data seeded successfully!")
|
|
logger.info("=" * 60)
|
|
logger.info("Admin credentials:")
|
|
logger.info(" Username: cosmo")
|
|
logger.info(" Password: cosmo")
|
|
logger.info("=" * 60)
|
|
|
|
except Exception as e:
|
|
await session.rollback()
|
|
logger.error(f"Error seeding admin data: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|