import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'chunyu_project.settings')
django.setup()

from django.contrib.auth import get_user_model

User = get_user_model()

# 为现有的test@example.com用户设置密码
username = 'test_admin'
email = 'admin@test.com'
password = 'admin123456'

# 检查用户是否存在
existing_user = User.objects.filter(username=username).first()
if existing_user:
    existing_user.set_password(password)
    existing_user.save()
    print(f'更新已有用户: {existing_user.username}, Email: {existing_user.email}')
else:
    print(f'用户 {username} 不存在，使用现有的 test@example.com 用户')

# 设置test@example.com的密码
test_user = User.objects.filter(username='test@example.com').first()
if test_user:
    test_user.set_password(password)
    test_user.save()
    print(f'设置 test@example.com 密码成功')
else:
    print('test@example.com 用户不存在')

print('\n========================================')
print('测试账号信息：')
print('========================================')
print(f'用户名/邮箱: test@example.com')
print(f'密码: {password}')
print('========================================')
print('\n或者使用：')
print(f'用户名: {username}')
print(f'邮箱: {email}')
print(f'密码: {password}')
print('========================================')
print('\n所有用户列表：')
for u in User.objects.all():
    print(f'  - {u.username} (Email: {u.email}, ID: {u.id})')