from django.db import models
from django.contrib.auth.models import User
from library.models import QivipQuiz
import random, string

# Create your models here.
class QuizGame(models.Model):
    host_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='host_user')
    join_code = models.CharField(max_length=6, unique=True, blank=True)
    quiz_id = models.ForeignKey(QivipQuiz, on_delete=models.CASCADE)

    def save(self, *args, **kwargs):
        if not self.join_code:
            self.join_code = self.generate_unique_code()
        super().save(*args, **kwargs)

    def generate_unique_code(self):
        for i in range(10):
            new_code = ''.join(random.choices(string.digits, k=6))
            if not QuizGame.objects.filter(join_code=new_code).exists():
                return new_code

class QuizGameParticipant(models.Model):
    display_name = models.CharField(verbose_name="Anzeigename", max_length=15)
    quiz_game = models.ForeignKey(QuizGame, on_delete=models.CASCADE, related_name="participant")
    score = models.IntegerField(verbose_name="Punkte", max_length=10)
    avatar = models.CharField(max_length=200)