#models.py
from django.db import models
from django.contrib.auth.models import User
import datetime
from service.managers import CheckRecords
class AlertDate(models.Model):
name = models.CharField(max_length=30)
days = models.IntegerField()
def __unicode__(self):
return self.name
class Admin:
pass
class Group(models.Model):
name = models.CharField(max_length=30)
user = models.ManyToManyField(User, filter_interface=True)
def __unicode__(self):
return self.name
class Admin:
pass
class Service(models.Model):
name = models.CharField(max_length=50)
def __unicode__(self):
return self.name
class Admin:
pass
class Client(models.Model):
name = models.CharField(max_length=50)
def __unicode__(self):
return self.name
class Admin:
pass
class Meta:
ordering = ['name']
class Record(models.Model):
service = models.ForeignKey(Service)
client = models.ForeignKey(Client)
description = models.TextField(blank=True)
information = models.TextField(blank=True)
start_date = models.DateField()
end_date = models.DateField()
alert_group = models.ManyToManyField(Group, filter_interface=True)
alert = models.ForeignKey(AlertDate)
objects = models.Manager()
check = CheckRecords()
def will_expire(self):
_expirein = self.end_date - datetime.date.today()
return _expirein.days
def __unicode__(self):
return self.description
class Admin:
ordering = ('end_date',)
list_display_links = ('service', 'client')
list_display = ('service', 'client', 'description', 'end_date', 'will_expire')
list_filter = ('service', 'client', 'description', 'end_date')
search_fields = ('service__name', 'description', 'client__name')
#managers.py
from django.db import models
from models import Record
from lib.functions import send_report
import datetime
class CheckRecords(models.Manager):
def expired(self):
records = self.filter(end_date__lt=datetime.date.today())
for record in records:
emails = []
for group in record.alert_group.all():
for user in group.user.all():
emails.append(user.email)
send_report(record, emails)
def in_expire_zone(self):
records = self.all()
for record in records:
if (record.end_date - datetime.timedelta(days=record.alert.days)) == datetime.date.today():
emails = []
for group in record.alert_group.all():
for user in group.user.all():
emails.append(user.email)
send_report(record, emails)
def every_week(self):
records = self.filter(end_date__lt=datetime.date.today())
for record in records:
days_left = record.end_date - datetime.date.today()
if days_left.days < record.alert.days and not days_left.days % 7:
emails = []
for group in record.alert_group.all():
for user in record.user.all():
emails.append(user.email)
send_report(record, emails)