1 files changed, 121 insertions, 0 deletions
diff --git a/releng/models.py b/releng/models.py
new file mode 100644
index 0000000..dc297d5
--- /dev/null
+++ b/releng/models.py
@@ -0,0 +1,121 @@
+from datetime import datetime
+
+from django.db import models
+from django.db.models import Max
+
+class IsoOption(models.Model):
+    name = models.CharField(max_length=200)
+
+    def __unicode__(self):
+        return self.name
+
+    def get_test_result(self, success):
+        try:
+            return self.test_set.filter(success=success).select_related(
+                    'iso').latest('iso__id').iso
+        except Test.DoesNotExist:
+            return None
+
+    def get_last_success(self):
+        return self.get_test_result(True)
+
+    def get_last_failure(self):
+        return self.get_test_result(False)
+
+    class Meta:
+        abstract = True
+
+class RollbackOption(IsoOption):
+    def get_rollback_test_result(self, success):
+        try:
+            return self.rollback_test_set.filter(success=success).select_related(
+                    'iso').latest('iso__id').iso
+        except Test.DoesNotExist:
+            return None
+
+    def get_last_rollback_success(self):
+        return self.get_rollback_test_result(True)
+
+    def get_last_rollback_failure(self):
+        return self.get_rollback_test_result(False)
+
+    class Meta:
+        abstract = True
+
+class Iso(models.Model):
+    name = models.CharField(max_length=255)
+    created = models.DateTimeField(editable=False)
+    active = models.BooleanField(default=True)
+
+    def __unicode__(self):
+        return self.name
+
+class Architecture(IsoOption):
+    pass
+
+class IsoType(IsoOption):
+    pass
+
+class BootType(IsoOption):
+    pass
+
+class HardwareType(IsoOption):
+    pass
+
+class InstallType(IsoOption):
+    pass
+
+class Source(IsoOption):
+    pass
+
+class ClockChoice(IsoOption):
+    pass
+
+class Filesystem(RollbackOption):
+    pass
+
+class Module(RollbackOption):
+    pass
+
+class Bootloader(IsoOption):
+    pass
+
+class Test(models.Model):
+    user_name = models.CharField(max_length=500)
+    user_email = models.EmailField()
+    ip_address = models.IPAddressField()
+    created = models.DateTimeField(editable=False)
+
+    iso = models.ForeignKey(Iso)
+    architecture = models.ForeignKey(Architecture)
+    iso_type = models.ForeignKey(IsoType)
+    boot_type = models.ForeignKey(BootType)
+    hardware_type = models.ForeignKey(HardwareType)
+    install_type = models.ForeignKey(InstallType)
+    source = models.ForeignKey(Source)
+    clock_choice = models.ForeignKey(ClockChoice)
+    filesystem = models.ForeignKey(Filesystem)
+    modules = models.ManyToManyField(Module, null=True, blank=True)
+    bootloader = models.ForeignKey(Bootloader)
+    rollback_filesystem = models.ForeignKey(Filesystem,
+            related_name="rollback_test_set", null=True, blank=True)
+    rollback_modules = models.ManyToManyField(Module,
+            related_name="rollback_test_set", null=True, blank=True)
+
+    success = models.BooleanField()
+    comments = models.TextField(null=True, blank=True)
+
+def set_created_field(sender, **kwargs):
+    # We use this same callback for both Isos and Tests
+    obj = kwargs['instance']
+    if not obj.created:
+        obj.created = datetime.utcnow()
+
+from django.db.models.signals import pre_save
+
+pre_save.connect(set_created_field, sender=Iso,
+        dispatch_uid="releng.models")
+pre_save.connect(set_created_field, sender=Test,
+        dispatch_uid="releng.models")
+
+# vim: set ts=4 sw=4 et: