2010-09-06 10:01:46 -07:00
|
|
|
import re
|
2011-02-27 10:31:06 -08:00
|
|
|
from urlparse import urlparse, urlunsplit
|
2010-09-06 10:01:46 -07:00
|
|
|
|
|
|
|
from django import forms
|
|
|
|
from django.contrib import admin
|
|
|
|
|
|
|
|
from .models import Mirror, MirrorProtocol, MirrorUrl, MirrorRsync
|
|
|
|
|
|
|
|
class MirrorUrlForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
|
|
model = MirrorUrl
|
|
|
|
def clean_url(self):
|
2011-02-27 10:31:06 -08:00
|
|
|
# is this a valid-looking URL?
|
|
|
|
url_parts = urlparse(self.cleaned_data["url"])
|
|
|
|
if not url_parts.scheme:
|
|
|
|
raise forms.ValidationError("No URL scheme (protocol) provided.")
|
|
|
|
if not url_parts.netloc:
|
|
|
|
raise forms.ValidationError("No URL host provided.")
|
|
|
|
if url_parts.params or url_parts.query or url_parts.fragment:
|
|
|
|
raise forms.ValidationError(
|
|
|
|
"URL parameters, query, and fragment elements are not supported.")
|
2010-09-06 10:01:46 -07:00
|
|
|
# ensure we always save the URL with a trailing slash
|
2011-02-27 10:31:06 -08:00
|
|
|
path = url_parts.path
|
|
|
|
if not path.endswith('/'):
|
|
|
|
path += '/'
|
|
|
|
url = urlunsplit((url_parts.scheme, url_parts.netloc, path, '', ''))
|
|
|
|
return url
|
2010-09-06 10:01:46 -07:00
|
|
|
|
|
|
|
class MirrorUrlInlineAdmin(admin.TabularInline):
|
|
|
|
model = MirrorUrl
|
|
|
|
form = MirrorUrlForm
|
2011-02-27 10:44:30 -08:00
|
|
|
readonly_fields = ('protocol', 'has_ipv4', 'has_ipv6')
|
2010-09-06 10:01:46 -07:00
|
|
|
extra = 3
|
|
|
|
|
|
|
|
# ripped off from django.forms.fields, adding netmask ability
|
2011-10-11 17:29:15 -07:00
|
|
|
IPV4NM_RE = re.compile(r'^(25[0-5]|2[0-4]\d|[0-1]?\d?\d)(\.(25[0-5]|2[0-4]\d|[0-1]?\d?\d)){3}(/(\d|[1-2]\d|3[0-2])){0,1}$')
|
|
|
|
|
2010-09-06 10:01:46 -07:00
|
|
|
class IPAddressNetmaskField(forms.fields.RegexField):
|
|
|
|
default_error_messages = {
|
|
|
|
'invalid': u'Enter a valid IPv4 address, possibly including netmask.',
|
|
|
|
}
|
|
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
2011-10-11 17:29:15 -07:00
|
|
|
super(IPAddressNetmaskField, self).__init__(IPV4NM_RE, *args, **kwargs)
|
2010-09-06 10:01:46 -07:00
|
|
|
|
|
|
|
class MirrorRsyncForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
|
|
model = MirrorRsync
|
|
|
|
ip = IPAddressNetmaskField(label='IP')
|
|
|
|
|
|
|
|
class MirrorRsyncInlineAdmin(admin.TabularInline):
|
|
|
|
model = MirrorRsync
|
|
|
|
form = MirrorRsyncForm
|
|
|
|
extra = 2
|
|
|
|
|
|
|
|
class MirrorAdminForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
|
|
model = Mirror
|
2012-02-13 10:34:35 -08:00
|
|
|
upstream = forms.ModelChoiceField(
|
|
|
|
queryset=Mirror.objects.filter(tier__gte=0, tier__lte=1),
|
|
|
|
required=False)
|
2010-09-06 10:01:46 -07:00
|
|
|
|
|
|
|
class MirrorAdmin(admin.ModelAdmin):
|
|
|
|
form = MirrorAdminForm
|
2012-04-24 22:09:46 -07:00
|
|
|
list_display = ('name', 'tier', 'country', 'active', 'public',
|
2011-08-22 14:42:06 -07:00
|
|
|
'isos', 'admin_email')
|
2012-04-24 22:09:46 -07:00
|
|
|
list_filter = ('tier', 'active', 'public', 'country')
|
2010-09-06 10:01:46 -07:00
|
|
|
search_fields = ('name',)
|
|
|
|
inlines = [
|
|
|
|
MirrorUrlInlineAdmin,
|
|
|
|
MirrorRsyncInlineAdmin,
|
|
|
|
]
|
|
|
|
|
2010-09-30 12:20:45 -07:00
|
|
|
class MirrorProtocolAdmin(admin.ModelAdmin):
|
2012-02-13 10:34:35 -08:00
|
|
|
list_display = ('protocol', 'is_download', 'default')
|
|
|
|
list_filter = ('is_download', 'default')
|
2010-09-30 12:20:45 -07:00
|
|
|
|
2010-09-06 10:01:46 -07:00
|
|
|
admin.site.register(Mirror, MirrorAdmin)
|
2010-09-30 12:20:45 -07:00
|
|
|
admin.site.register(MirrorProtocol, MirrorProtocolAdmin)
|
2010-09-06 10:01:46 -07:00
|
|
|
|
|
|
|
# vim: set ts=4 sw=4 et:
|