You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
2 years ago
|
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
|
||
|
|
|