54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
|
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
创建默认用户并迁移现有数据
|
|||
|
|
"""
|
|||
|
|
import os, sys
|
|||
|
|
sys.path.insert(0, '/root/.openclaw/workspace/diary-system/backend')
|
|||
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'diary_system.settings'
|
|||
|
|
|
|||
|
|
import django
|
|||
|
|
django.setup()
|
|||
|
|
|
|||
|
|
from django.contrib.auth.models import User
|
|||
|
|
from diary.models import DiaryEntry, Experience, Task, Comment
|
|||
|
|
|
|||
|
|
# 1. 创建默认用户(北极星)
|
|||
|
|
print("📝 创建默认用户...")
|
|||
|
|
default_user, created = User.objects.get_or_create(
|
|||
|
|
username='beijixing',
|
|||
|
|
defaults={'email': 'beijixing@example.com'}
|
|||
|
|
)
|
|||
|
|
if created:
|
|||
|
|
default_user.set_password('beijixing123')
|
|||
|
|
default_user.save()
|
|||
|
|
print(f"✅ 创建默认用户:beijixing / beijixing123")
|
|||
|
|
else:
|
|||
|
|
print(f"✅ 默认用户已存在:beijixing")
|
|||
|
|
|
|||
|
|
# 2. 迁移现有数据到默认用户
|
|||
|
|
print("\n📦 迁移现有数据...")
|
|||
|
|
|
|||
|
|
# 迁移日记
|
|||
|
|
diary_count = DiaryEntry.objects.filter(user__isnull=True).count()
|
|||
|
|
DiaryEntry.objects.filter(user__isnull=True).update(user=default_user)
|
|||
|
|
print(f" 日记:{diary_count} 条")
|
|||
|
|
|
|||
|
|
# 迁移经验
|
|||
|
|
exp_count = Experience.objects.filter(user__isnull=True).count()
|
|||
|
|
Experience.objects.filter(user__isnull=True).update(user=default_user)
|
|||
|
|
print(f" 经验:{exp_count} 条")
|
|||
|
|
|
|||
|
|
# 迁移任务
|
|||
|
|
task_count = Task.objects.filter(user__isnull=True).count()
|
|||
|
|
Task.objects.filter(user__isnull=True).update(user=default_user)
|
|||
|
|
print(f" 任务:{task_count} 条")
|
|||
|
|
|
|||
|
|
# 迁移评论
|
|||
|
|
comment_count = Comment.objects.filter(user__isnull=True).count()
|
|||
|
|
Comment.objects.filter(user__isnull=True).update(user=default_user)
|
|||
|
|
print(f" 评论:{comment_count} 条")
|
|||
|
|
|
|||
|
|
print("\n✅ 数据迁移完成!")
|
|||
|
|
print(f"\n📝 默认用户:beijixing")
|
|||
|
|
print(f"🔑 密码:beijixing123")
|