from django.db import models

# Create your models here.
class Question (models.Model):
    ID = models.AutoField(primary_key=True)
    q = models.CharField(max_length=128)
    a1 = models.CharField(max_length=128)
    a2 = models.CharField(max_length=128)
    a3 = models.CharField(max_length=128)
    a4 = models.CharField(max_length=128)
    answer = models.IntegerField()

    def get_random_question():
       return Question.objects.order_by('?')[:1]

    def search_id(self):
        return Question.objects.filter(ID=self.ID)

class AnswerRecord (models.Model):
    user = models.CharField(max_length=128)
    mobile = models.IntegerField()
    question = models.CharField(max_length=128)
    result = models.CharField(max_length=8)
    your_answer = models.CharField(max_length=128)
    correct_answer = models.CharField(max_length=128, null=True)

    def add_record(self):
        AnswerRecord.objects.create(user=self.user, mobile=self.mobile, question=self.question, result=self.result,
                                    your_answer=self.your_answer, correct_answer=self.correct_answer)

    def get_record(self):
        return AnswerRecord.objects.filter(user=self.user, mobile=self.mobile)