32 lines
938 B
Python
32 lines
938 B
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.postgres.fields import ArrayField
|
|
from common import to_lb
|
|
|
|
UNITS = (
|
|
('m', 'Metric (kg)'),
|
|
('i', 'Imperial (lb)'),
|
|
)
|
|
|
|
|
|
class Profile(models.Model):
|
|
user = models.OneToOneField(User, related_name='profile')
|
|
|
|
lifting_units = models.CharField(max_length=1, choices=UNITS, default='i')
|
|
plate_pairs = ArrayField(models.DecimalField(max_digits=7, decimal_places=3),
|
|
default=['45','45','25','10','5','5','2.5','1.25'])
|
|
|
|
|
|
class Bar(models.Model):
|
|
user = models.OneToOneField(User, related_name='bar')
|
|
|
|
name = models.CharField(max_length=100)
|
|
weight_kg = models.DecimalField(max_digits=7, decimal_places=3)
|
|
|
|
def __str__(self):
|
|
return '{} ({}lb / {}kg)'.format(self.name, self.weight_kg, self.weight_lb)
|
|
|
|
@property
|
|
def weight_lb(self):
|
|
return to_lb(self.weight_kg)
|