mirror of
https://github.com/Hopiu/django-tos.git
synced 2026-04-24 21:04:43 +00:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from django.core.exceptions import ValidationError
|
|
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
|
|
class NoActiveTermsOfService(ValidationError):
|
|
pass
|
|
|
|
|
|
class BaseModel(models.Model):
|
|
created = models.DateTimeField(auto_now_add=True, editable=False)
|
|
modified = models.DateTimeField(auto_now=True, editable=False)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class TermsOfServiceManager(models.Manager):
|
|
def get_current_tos(self):
|
|
try:
|
|
return self.get(active=True)
|
|
except self.model.DoesNotExist:
|
|
raise NoActiveTermsOfService(u'Please create an active Terms-of-Service')
|
|
|
|
|
|
class TermsOfService(BaseModel):
|
|
active = models.BooleanField(verbose_name=_('active'),
|
|
help_text=_(u'Only one terms of service is allowed to be active'))
|
|
content = models.TextField(verbose_name=_('content'), blank=True)
|
|
objects = TermsOfServiceManager()
|
|
|
|
class Meta:
|
|
get_latest_by = 'created'
|
|
ordering = ('-created',)
|
|
verbose_name = _('Terms of Service')
|
|
verbose_name_plural = _('Terms of Service')
|
|
|
|
def __unicode__(self):
|
|
active = 'inactive'
|
|
if self.active:
|
|
active = 'active'
|
|
return '%s: %s' % (self.created, active)
|
|
|
|
def save(self, *args, **kwargs):
|
|
""" Ensure we're being saved properly """
|
|
|
|
if self.active:
|
|
TermsOfService.objects.exclude(id=self.id).update(active=False)
|
|
|
|
else:
|
|
if not TermsOfService.objects\
|
|
.exclude(id=self.id)\
|
|
.filter(active=True):
|
|
raise NoActiveTermsOfService(u'One of the terms of service must be marked active')
|
|
|
|
super(TermsOfService, self).save(*args, **kwargs)
|
|
|
|
|
|
class UserAgreement(BaseModel):
|
|
terms_of_service = models.ForeignKey(TermsOfService, related_name='terms')
|
|
user = models.ForeignKey(User, related_name='user_agreement')
|
|
|
|
def __unicode__(self):
|
|
return u'%s agreed to TOS: %s' % (self.user.username,
|
|
unicode(self.terms_of_service))
|
|
|
|
|
|
def has_user_agreed_latest_tos(user):
|
|
if UserAgreement.objects.filter(terms_of_service=TermsOfService.objects.get_current_tos(), user=user):
|
|
return True
|
|
return False
|