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}"