Mirror tiering enhancements

Add ability to track tier and upstream mirror in the database.

Signed-off-by: Dan McGee <dan@archlinux.org>

1 files changed, 9 insertions, 0 deletions

diff --git a/main/models.py b/main/models.py
index dd37f885..eff55798 100644
--- a/main/models.py
+++ b/main/models.py
@@ -51,8 +51,17 @@ class PackageManager(models.Manager):
 #############################
 ### General Model Classes ###
 #############################
+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, blank=True)
     country = models.CharField(max_length=255, db_index=True)
     admin_email = models.EmailField(max_length=255, blank=True)
     notes = models.CharField(max_length=255, blank=True)