From 93e0552e3dcaa3b2feb82a5dddc0ae43d82eeee0 Mon Sep 17 00:00:00 2001
From: Dan McGee <dan@archlinux.org>
Date: Sat, 6 Mar 2010 09:42:08 -0600
Subject: Mirror tiering enhancements

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

Signed-off-by: Dan McGee <dan@archlinux.org>
---
 main/models.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'main/models.py')

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 @@ def flagged(self):
 #############################
 ### 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)
-- 
cgit v1.2.3-54-g00ecf