1 files changed, 57 insertions, 0 deletions
diff --git a/mirrors/models.py b/mirrors/models.py
index e69de29..94256a9 100644
--- a/mirrors/models.py
+++ b/mirrors/models.py
@@ -0,0 +1,57 @@
+from django.db import models
+
+TIER_CHOICES = (
+    (0, 'Tier 0'),
+    (1, 'Tier 1'),
+    (2, 'Tier 2'),
+    (-1, 'Untiered'),
+)
+
+class Mirror(models.Model):
+    name = models.CharField(max_length=255)
+    tier = models.SmallIntegerField(default=2, choices=TIER_CHOICES)
+    upstream = models.ForeignKey('self', null=True)
+    country = models.CharField(max_length=255, db_index=True)
+    admin_email = models.EmailField(max_length=255, blank=True)
+    public = models.BooleanField(default=True)
+    active = models.BooleanField(default=True)
+    isos = models.BooleanField(default=True)
+    rsync_user = models.CharField(max_length=50, blank=True, default='')
+    rsync_password = models.CharField(max_length=50, blank=True, default='')
+    notes = models.TextField(blank=True)
+
+    class Meta:
+        ordering = ('country', 'name')
+
+    def __unicode__(self):
+        return self.name
+
+    def supported_protocols(self):
+        protocols = MirrorProtocol.objects.filter(urls__mirror=self).distinct()
+        return ", ".join([p.protocol for p in protocols])
+
+class MirrorProtocol(models.Model):
+    protocol = models.CharField(max_length=10, unique=True)
+    def __unicode__(self):
+        return self.protocol
+    class Meta:
+        verbose_name = 'Mirror Protocol'
+
+class MirrorUrl(models.Model):
+    url = models.CharField(max_length=255)
+    protocol = models.ForeignKey(MirrorProtocol, related_name="urls")
+    mirror = models.ForeignKey(Mirror, related_name="urls")
+    def __unicode__(self):
+        return self.url
+    class Meta:
+        verbose_name = 'Mirror URL'
+
+class MirrorRsync(models.Model):
+    ip = models.CharField(max_length=24)
+    mirror = models.ForeignKey(Mirror, related_name="rsync_ips")
+    def __unicode__(self):
+        return "%s" % (self.ip)
+    class Meta:
+        verbose_name = 'Mirror Rsync IP'
+
+# vim: set ts=4 sw=4 et: