from __future__ import unicode_literals

from django.utils.encoding import python_2_unicode_compatible
from django.db import models
from django.contrib.auth import get_user_model
from django.conf import settings

class DefaultManager(models.Manager):
    pass

class BadgeRecipientManager(models.Manager):

    def get_all_recipients(self):
        User = get_user_model()
        return User.objects.filter(badges__isnull=False).distinct()

    def get_all_nonrecipients(self):
        User = get_user_model()
        return User.objects.exclude(badges__isnull=False)

@python_2_unicode_compatible
class Badge(models.Model):
    """
    Three fields:
        level - integer, default: 100
        name - text, max. 20 chars
        description - text, aim for one line

    receivers -> User.badges
    """
    level = models.PositiveIntegerField(default=100)
    name = models.CharField(max_length=20, unique=True)
    description = models.TextField()
    receivers = models.ManyToManyField(settings.AUTH_USER_MODEL, blank=True, related_name='badges')

    objects = BadgeRecipientManager()

    class Meta:
        db_table = 'nano_badge_badge'

    def __str__(self):
        return self.name