from django.db import models
from clubs.models import Team
from accounts.models import CustomUser

class Poll(models.Model):
    question = models.CharField(max_length=255)
    team = models.ForeignKey(Team, on_delete=models.CASCADE, related_name='polls')
    creator = models.ForeignKey(CustomUser, on_delete=models.CASCADE, related_name='polls_created')
    multiple_choice = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.question

class Choice(models.Model):
    poll = models.ForeignKey(Poll, on_delete=models.CASCADE, related_name='choices')
    choice_text = models.CharField(max_length=100)
    votes = models.ManyToManyField(CustomUser, related_name='voted_choices', blank=True)

    def __str__(self):
        return self.choice_text