mirror of
https://github.com/Hopiu/django-tos.git
synced 2026-03-16 20:10:24 +00:00
PEP8
This commit is contained in:
parent
8980a51888
commit
abb2043759
1 changed files with 32 additions and 31 deletions
|
|
@ -1,68 +1,69 @@
|
|||
from django.core.exceptions import ValidationError
|
||||
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 _
|
||||
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)
|
||||
|
||||
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)
|
||||
return self.get(active=True)
|
||||
except self.model.DoesNotExist:
|
||||
raise NoActiveTermsOfService('Please create an active Terms-of-Service')
|
||||
|
||||
|
||||
class TermsOfService(BaseModel):
|
||||
active = models.BooleanField(_('active'), _('Only one terms of service is allowed to be active'))
|
||||
content = models.TextField(_('content'), blank=True)
|
||||
objects = TermsOfServiceManager()
|
||||
|
||||
active = models.BooleanField(_('active'), _('Only one terms of service is allowed to be active'))
|
||||
content = models.TextField(_('content'), blank=True)
|
||||
objects = TermsOfServiceManager()
|
||||
|
||||
class Meta:
|
||||
class Meta:
|
||||
get_latest_by = 'created'
|
||||
ordering = ('-created',)
|
||||
verbose_name=_('Terms of Service')
|
||||
verbose_name_plural=_('Terms of Service')
|
||||
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 """
|
||||
active = 'active'
|
||||
return '{0}: {1}'.format(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('One of the terms of service must be marked active')
|
||||
|
||||
super(TermsOfService,self).save(*args, **kwargs)
|
||||
|
||||
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')
|
||||
|
||||
user = models.ForeignKey(User, related_name='user_agreement')
|
||||
|
||||
def __unicode__(self):
|
||||
return '%s agreed to TOS: %s ' % (self.user.username, self.terms_of_service.__unicode__())
|
||||
|
||||
|
||||
|
||||
|
||||
def has_user_agreed_latest_tos(user):
|
||||
if UserAgreement.objects.filter(terms_of_service=TermsOfService.objects.get_current_tos(),user=user):
|
||||
if UserAgreement.objects.filter(terms_of_service=TermsOfService.objects.get_current_tos(), user=user):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue