from django.core.management.base import BaseCommand
from university.models import UserOnlineExam, OnlineExamRegistrate


class Command(BaseCommand):

    def handle(self, *args, **options):
        for u_online_exam in UserOnlineExam.objects.all():
            OnlineExamRegistrate.objects.create(
                user=u_online_exam.user,
                online_exam=u_online_exam.online_exam,
                name=u_online_exam.user.exam_real_name or '',
                chinese_id=u_online_exam.user.exam_chinese_id_number or '',
                phone=u_online_exam.user.exam_contact_phone or '',
                email=u_online_exam.user.exam_contact_email or '',
                payment_method=OnlineExamRegistrate.PAYMENT_ONLINE,
                online_payment_success=True
            )
            u_online_exam.examinee_name = u_online_exam.user.exam_real_name or ''
            u_online_exam.examinee_chinese_id = u_online_exam.user.exam_chinese_id_number or ''
            u_online_exam.save(update_fields=['examinee_name', 'examinee_chinese_id'])