add (untested) manager

This commit is contained in:
Fred 2016-08-07 21:50:24 +02:00
parent 4d47113aea
commit 37f389d593
2 changed files with 13 additions and 1 deletions

View File

@ -19,7 +19,15 @@ class Gauge(models.Model):
class Diff(object): class Diff(object):
def __init__(self, mesure_1, mesure_2): def __init__(self, mesure_1, mesure_2):
self.increase = abs(mesure_1.diff(mesure_2)) self.increase = abs(mesure_1.diff(mesure_2))
self.delta = abs(mesure_1.moment - mesure_2.moment) self.timedelta = abs(mesure_1.moment - mesure_2.moment)
class MesureManager(models.Manager):
def create(self, moment, value, gauge_sn, category_label):
category, cat_created = Category.objects.get_or_create(name=category_label)
gauge, gauge_created = Gauge.objects.get_or_create(serial_number=gauge_sn, category=category)
mesure = Mesure.objects.create(moment=moment, value=value, gauge=gauge)
return mesure
class Mesure(models.Model): class Mesure(models.Model):
moment = models.DateTimeField() moment = models.DateTimeField()

View File

@ -43,6 +43,10 @@ class TestGaugeModel(TestCase):
self.assertEqual(1, len(gauge_1.mesures.all())) self.assertEqual(1, len(gauge_1.mesures.all()))
class TestMesureManager(TestCase):
def test_create(self):
raise Exception('not implemented yet')
class TestMesureModel(TestCase): class TestMesureModel(TestCase):
def setUp(self): def setUp(self):