models py from django db import models from django contrib auth models

#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)