from functools import reduce
from django.db import models
from django.contrib.auth.models import User
from django.contrib import admin

class Movie(models.Model):

    class Meta:
        permissions = [
            ("moderate_movies", "Can edit other's movies and mark them as watched"),
        ]

    name = models.CharField(max_length=100)
    suggested_by = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
    watched = models.BooleanField(default=False)

    @property
    def score(self):
        return reduce(lambda result,v: result+v.vote, self.movievote_set.all(), 0)

    @property
    def seen_score(self):
        return reduce(lambda result,v: result+int(v.seen), self.movievote_set.all(), 0)

    def __str__(self):
        return self.name

class MovieVote(models.Model):

    class Vote(models.IntegerChoices):
        UPVOTE = 1
        NOVOTE = 0
        DOWNVOTE = -1
    
    movie = models.ForeignKey(Movie, on_delete=models.CASCADE)
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    vote = models.IntegerField(choices=Vote.choices, default=Vote.NOVOTE)
    seen = models.BooleanField(default=False, null=True)

    def __str__(self):
        return f"{self.user.username}'s vote for {self.movie.name}"