#!/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")