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")
|