baby_tracker/main/models.py

35 lines
1.1 KiB
Python

from django.db import models
# Create your models here.
from django.db import models
class Feeding(models.Model):
FEEDING_TYPES = [
('bottle', 'Bottle'),
('breast', 'Breast'),
]
timestamp = models.DateTimeField(auto_now_add=True)
feeding_type = models.CharField(max_length=10, choices=FEEDING_TYPES)
amount = models.FloatField(null=True, blank=True) # Store ounces for bottle feed
duration = models.IntegerField(null=True, blank=True) # Store duration in minutes for breastfeed
notes = models.TextField(blank=True)
def __str__(self):
return f"{self.get_feeding_type_display()} - {self.timestamp}"
class Potty(models.Model):
timestamp = models.DateTimeField(auto_now_add=True)
consistency = models.CharField(max_length=50, choices=[
('solid', 'Solid'),
('soft', 'Soft'),
('watery', 'Watery'),
('pee', 'Pee'),
('other', 'Other'),
])
notes = models.TextField(blank=True)
def __str__(self):
return f"{self.get_potty_type_display()} - {self.timestamp}"