mirror of https://github.com/snachodog/mybuddy.git
157 lines
4.7 KiB
Python
157 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
from .utils import duration_string
|
|
|
|
|
|
class Child(models.Model):
|
|
first_name = models.CharField(max_length=255)
|
|
last_name = models.CharField(max_length=255)
|
|
birth_date = models.DateField(blank=False, null=False)
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['last_name', 'first_name']
|
|
verbose_name_plural = 'Children'
|
|
|
|
def __str__(self):
|
|
return '{} {}'.format(self.first_name, self.last_name)
|
|
|
|
|
|
class DiaperChange(models.Model):
|
|
child = models.ForeignKey('Child', related_name='diaper_change')
|
|
time = models.DateTimeField(blank=False, null=False)
|
|
wet = models.BooleanField()
|
|
solid = models.BooleanField()
|
|
color = models.CharField(max_length=255, blank=True, choices=[
|
|
('black', 'Black'),
|
|
('brown', 'Brown'),
|
|
('green', 'Green'),
|
|
('yellow', 'Yellow'),
|
|
])
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-time']
|
|
|
|
def __str__(self):
|
|
return 'Diaper change for {} on {}'.format(
|
|
self.child, self.time.date())
|
|
|
|
|
|
class Feeding(models.Model):
|
|
child = models.ForeignKey('Child', related_name='feeding')
|
|
start = models.DateTimeField(blank=False, null=False)
|
|
end = models.DateTimeField(blank=False, null=False)
|
|
type = models.CharField(max_length=255, choices=[
|
|
('breast milk', 'Breast milk'),
|
|
('formula', 'Formula'),
|
|
])
|
|
method = models.CharField(max_length=255, choices=[
|
|
('bottle', 'Bottle'),
|
|
('left breast', 'Left breast'),
|
|
('right breast', 'Right breast'),
|
|
])
|
|
amount = models.FloatField(blank=True, null=True)
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-start']
|
|
|
|
def __str__(self):
|
|
return 'Feeding for {} on {} ({})'.format(
|
|
self.child, self.end.date(), self.duration())
|
|
|
|
def duration(self):
|
|
return duration_string(self.start, self.end)
|
|
|
|
|
|
class Note(models.Model):
|
|
child = models.ForeignKey('Child', related_name='note')
|
|
note = models.TextField()
|
|
time = models.DateTimeField(auto_now=True)
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-time']
|
|
|
|
def __str__(self):
|
|
return 'Note about {} on {}'.format(self.child, self.time.date())
|
|
|
|
|
|
class Sleep(models.Model):
|
|
child = models.ForeignKey('Child', related_name='sleep')
|
|
start = models.DateTimeField(blank=False, null=False)
|
|
end = models.DateTimeField(blank=False, null=False)
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-start']
|
|
verbose_name_plural = 'Sleep'
|
|
|
|
def __str__(self):
|
|
return 'Sleep for {} on {} ({})'.format(
|
|
self.child, self.end.date(), self.duration())
|
|
|
|
def duration(self):
|
|
return duration_string(self.start, self.end)
|
|
|
|
|
|
class Timer(models.Model):
|
|
name = models.CharField(max_length=255, null=True, blank=True)
|
|
start = models.DateTimeField(auto_now_add=True)
|
|
end = models.DateTimeField(blank=True, null=True, editable=False)
|
|
active = models.BooleanField(default=True, editable=False)
|
|
user = models.ForeignKey('auth.User', related_name='timers')
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-active', '-start', '-end']
|
|
|
|
def __str__(self):
|
|
return self.name or 'Timer #{}'.format(self.id)
|
|
|
|
def duration(self):
|
|
return duration_string(self.start, self.end or timezone.now(),
|
|
short=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.active = self.end is None
|
|
self.name = self.name or None
|
|
super(Timer, self).save(*args, **kwargs)
|
|
|
|
|
|
class TummyTime(models.Model):
|
|
child = models.ForeignKey('Child', related_name='tummy_time')
|
|
start = models.DateTimeField(blank=False, null=False)
|
|
end = models.DateTimeField(blank=False, null=False)
|
|
milestone = models.CharField(max_length=255, blank=True)
|
|
|
|
objects = models.Manager()
|
|
|
|
class Meta:
|
|
default_permissions = ('view', 'add', 'change', 'delete')
|
|
ordering = ['-start']
|
|
|
|
def __str__(self):
|
|
return 'Tummy time for {} on {} ({})'.format(
|
|
self.child, self.end.date(), self.duration())
|
|
|
|
def duration(self):
|
|
return duration_string(self.start, self.end)
|