cnbugs
a40a0137cf
功能： - Django + MySQL + 深色主题 - 资产增删改查（含资产编号、BMC地址、设备位置、备注） - Excel导入导出（分类自动创建） - 设备分类管理 - 资产变更记录追踪 - 质保到期提醒 - 用户认证系统 - Docker部署支持
39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.contrib.auth.models import User
|
|
from assetapp.models import Category
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = '初始化系统：创建管理员用户和默认分类'
|
|
|
|
def handle(self, *args, **options):
|
|
# 创建管理员
|
|
if not User.objects.filter(username='admin').exists():
|
|
User.objects.create_superuser('admin', 'admin@example.com', 'admin123')
|
|
self.stdout.write(self.style.SUCCESS('✓ 管理员用户已创建 (admin/admin123)'))
|
|
else:
|
|
self.stdout.write(' 管理员用户已存在，跳过')
|
|
|
|
# 创建默认分类
|
|
default_categories = [
|
|
('服务器', '各类物理服务器、虚拟化主机'),
|
|
('交换机', '网络交换设备'),
|
|
('路由器', '网络路由设备'),
|
|
('存储设备', 'SAN/NAS等存储设备'),
|
|
('防火墙', '网络安全设备'),
|
|
('UPS', '不间断电源设备'),
|
|
('其他', '其他类型设备'),
|
|
]
|
|
|
|
created_count = 0
|
|
for name, desc in default_categories:
|
|
cat, created = Category.objects.get_or_create(
|
|
name=name,
|
|
defaults={'description': desc}
|
|
)
|
|
if created:
|
|
created_count += 1
|
|
|
|
self.stdout.write(self.style.SUCCESS(f'✓ 分类初始化完成 (新增 {created_count} 个)'))
|
|
self.stdout.write(self.style.SUCCESS('\n初始化完成！请使用 admin/admin123 登录'))
|