code.kerkeslager.com Git - climbing.kerkeslager.com/blob - src/nutrition/models.py

   1 from django.contrib.auth.models import User
   2 from django.db import models
   3
   4 class Food(models.Model):
   5     name = models.CharField(max_length=64)
   6     brand = models.CharField(max_length=64)
   7     calories = models.PositiveIntegerField()
   8
   9     def __str__(self):
  10         return '{} ({})'.format(self.name, self.brand)
  11
  12 WEIGHT_UNITS_CHOICES = (
  13     ('kgs', 'kgs'),
  14     ('lbs', 'lbs'),
  15 )
  16
  17 class WeightRecord(models.Model):
  18     user = models.ForeignKey(User, on_delete=models.CASCADE)
  19     weight = models.PositiveIntegerField()
  20     units = models.CharField(max_length=3, choices=WEIGHT_UNITS_CHOICES)
  21
  22     def __str__(self):
  23         return '{}: {}{}'.format(
  24             self.user.username,
  25             self.weight,
  26             self.units,
  27         )