coding utf-8 from django db import models from service managers import

# -*- coding: utf-8

from django.db import models
from service.managers import CheckRecords
from django.contrib.auth.models import User
import datetime


class AlertDate(models.Model):
    name = models.CharField(max_length=30)
    days = models.IntegerField()

    def __unicode__(self):
        return self.name


class Group(models.Model):
    name = models.CharField(max_length=30)
    user = models.ManyToManyField(User)

    def __unicode__(self):
        return self.name


class Service(models.Model):
    name = models.CharField(max_length=50)

    def __unicode__(self):
        return self.name


class Client(models.Model):
    name = models.CharField(max_length=50)

    def __unicode__(self):
        return self.name

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