c2e84a787a
* Don't capitalize things in verbose_name, Django does this automatically and uses title case * Add overrides for IP, URL, ISO, etc. Signed-off-by: Dan McGee <dan@archlinux.org>
137 lines
4.4 KiB
Python
137 lines
4.4 KiB
Python
import socket
|
|
from urlparse import urlparse
|
|
|
|
from django.db import models
|
|
from django.core.exceptions import ValidationError
|
|
|
|
class NullCharField(models.CharField):
|
|
description = "String (up to %(max_length)s), NULL if value is empty"
|
|
_south_introspects = True
|
|
|
|
def get_prep_value(self, value):
|
|
if value == '':
|
|
return None
|
|
return self.to_python(value)
|
|
|
|
TIER_CHOICES = (
|
|
(0, 'Tier 0'),
|
|
(1, 'Tier 1'),
|
|
(2, 'Tier 2'),
|
|
(-1, 'Untiered'),
|
|
)
|
|
|
|
class Mirror(models.Model):
|
|
name = models.CharField(max_length=255, unique=True)
|
|
tier = models.SmallIntegerField(default=2, choices=TIER_CHOICES)
|
|
upstream = models.ForeignKey('self', null=True, on_delete=models.SET_NULL)
|
|
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("ISOs", 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).order_by('protocol').distinct()
|
|
return sorted(protocols)
|
|
|
|
def downstream(self):
|
|
return Mirror.objects.filter(upstream=self).order_by('name')
|
|
|
|
def get_absolute_url(self):
|
|
return '/mirrors/%s/' % self.name
|
|
|
|
class MirrorProtocol(models.Model):
|
|
protocol = models.CharField(max_length=10, unique=True)
|
|
is_download = models.BooleanField(default=True,
|
|
help_text="Is protocol useful for end-users, e.g. FTP/HTTP")
|
|
|
|
def __unicode__(self):
|
|
return self.protocol
|
|
|
|
class Meta:
|
|
ordering = ('protocol',)
|
|
|
|
class MirrorUrl(models.Model):
|
|
url = models.CharField("URL", max_length=255, unique=True)
|
|
protocol = models.ForeignKey(MirrorProtocol, related_name="urls",
|
|
editable=False, on_delete=models.PROTECT)
|
|
mirror = models.ForeignKey(Mirror, related_name="urls")
|
|
country = NullCharField(max_length=255, null=True, blank=True,
|
|
db_index=True)
|
|
has_ipv4 = models.BooleanField("IPv4 capable", default=True,
|
|
editable=False)
|
|
has_ipv6 = models.BooleanField("IPv6 capable", default=False,
|
|
editable=False)
|
|
|
|
def address_families(self):
|
|
hostname = urlparse(self.url).hostname
|
|
info = socket.getaddrinfo(hostname, None, 0, socket.SOCK_STREAM)
|
|
families = [x[0] for x in info]
|
|
return families
|
|
|
|
@property
|
|
def hostname(self):
|
|
return urlparse(self.url).hostname
|
|
|
|
@property
|
|
def real_country(self):
|
|
return self.country or self.mirror.country
|
|
|
|
def clean(self):
|
|
try:
|
|
# Auto-map the protocol field by looking at the URL
|
|
protocol = urlparse(self.url).scheme
|
|
self.protocol = MirrorProtocol.objects.get(protocol=protocol)
|
|
except Exception as e:
|
|
raise ValidationError(e)
|
|
try:
|
|
families = self.address_families()
|
|
self.has_ipv4 = socket.AF_INET in families
|
|
self.has_ipv6 = socket.AF_INET6 in families
|
|
except socket.error as e:
|
|
# We don't fail in this case; we'll just set both to False
|
|
self.has_ipv4 = False
|
|
self.has_ipv6 = False
|
|
|
|
def __unicode__(self):
|
|
return self.url
|
|
|
|
class Meta:
|
|
verbose_name = 'mirror URL'
|
|
|
|
class MirrorRsync(models.Model):
|
|
ip = models.CharField("IP", 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'
|
|
|
|
class MirrorLog(models.Model):
|
|
url = models.ForeignKey(MirrorUrl, related_name="logs")
|
|
check_time = models.DateTimeField(db_index=True)
|
|
last_sync = models.DateTimeField(null=True)
|
|
duration = models.FloatField(null=True)
|
|
is_success = models.BooleanField(default=True)
|
|
error = models.CharField(max_length=255, blank=True, default='')
|
|
|
|
def __unicode__(self):
|
|
return "Check of %s at %s" % (self.url.url, self.check_time)
|
|
|
|
class Meta:
|
|
verbose_name = 'mirror check log'
|
|
|
|
# vim: set ts=4 sw=4 et:
|