2009-10-31 08:22:09 -07:00
|
|
|
from django import forms
|
2010-09-12 08:14:34 -07:00
|
|
|
from django.db.models import Avg, Count, Max, Min, StdDev
|
|
|
|
from django.db.models import Q
|
2010-07-02 17:24:54 -07:00
|
|
|
from django.views.decorators.csrf import csrf_exempt
|
2010-09-03 11:44:27 -07:00
|
|
|
from django.views.generic.simple import direct_to_template
|
|
|
|
|
2010-02-09 22:29:27 -08:00
|
|
|
from main.utils import make_choice
|
2010-09-06 10:01:46 -07:00
|
|
|
from .models import Mirror, MirrorUrl, MirrorProtocol
|
2010-09-21 16:31:26 -07:00
|
|
|
from .utils import get_mirror_statuses, get_mirror_errors
|
2010-09-12 08:14:34 -07:00
|
|
|
|
|
|
|
import datetime
|
2010-09-21 16:58:34 -07:00
|
|
|
from operator import attrgetter
|
2009-10-31 08:22:09 -07:00
|
|
|
|
|
|
|
class MirrorlistForm(forms.Form):
|
2010-07-02 17:24:54 -07:00
|
|
|
country = forms.MultipleChoiceField(required=False)
|
2010-07-02 18:02:26 -07:00
|
|
|
protocol = forms.MultipleChoiceField(required=False)
|
2010-09-21 16:58:34 -07:00
|
|
|
use_mirror_status = forms.BooleanField(required=False)
|
2009-10-31 08:22:09 -07:00
|
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
super(MirrorlistForm, self).__init__(*args, **kwargs)
|
2010-07-02 17:24:54 -07:00
|
|
|
mirrors = Mirror.objects.filter(active=True).values_list(
|
2009-10-31 08:22:09 -07:00
|
|
|
'country', flat=True).distinct().order_by('country')
|
2010-07-02 18:02:26 -07:00
|
|
|
self.fields['country'].choices = make_choice(mirrors)
|
2010-07-02 18:11:48 -07:00
|
|
|
self.fields['country'].initial = ['Any']
|
2010-07-02 18:02:26 -07:00
|
|
|
protos = make_choice(
|
|
|
|
MirrorProtocol.objects.exclude(protocol__iexact='rsync'))
|
|
|
|
self.fields['protocol'].choices = protos
|
|
|
|
self.fields['protocol'].initial = [t[0] for t in protos]
|
2009-10-31 08:22:09 -07:00
|
|
|
|
2010-07-02 17:24:54 -07:00
|
|
|
@csrf_exempt
|
2010-09-10 16:42:59 -07:00
|
|
|
def generate_mirrorlist(request):
|
2010-07-02 17:24:54 -07:00
|
|
|
if request.REQUEST.get('country', ''):
|
|
|
|
form = MirrorlistForm(data=request.REQUEST)
|
2009-10-31 08:22:09 -07:00
|
|
|
if form.is_valid():
|
2010-07-02 17:24:54 -07:00
|
|
|
countries = form.cleaned_data['country']
|
2010-07-02 18:02:26 -07:00
|
|
|
protocols = form.cleaned_data['protocol']
|
2010-09-21 16:58:34 -07:00
|
|
|
use_status = form.cleaned_data['use_mirror_status']
|
|
|
|
return find_mirrors(request, countries, protocols, use_status)
|
2009-10-31 08:22:09 -07:00
|
|
|
else:
|
|
|
|
form = MirrorlistForm()
|
2009-10-31 08:29:12 -07:00
|
|
|
|
2010-09-03 11:44:27 -07:00
|
|
|
return direct_to_template(request, 'mirrors/index.html', {'mirrorlist_form': form})
|
2009-10-31 08:22:09 -07:00
|
|
|
|
2010-09-21 16:58:34 -07:00
|
|
|
def find_mirrors(request, countries=None, protocols=None, use_status=False):
|
2010-07-02 18:02:26 -07:00
|
|
|
if not protocols:
|
|
|
|
protocols = MirrorProtocol.objects.exclude(
|
|
|
|
protocol__iexact='rsync').values_list('protocol', flat=True)
|
2010-01-30 11:50:16 -08:00
|
|
|
qset = MirrorUrl.objects.select_related().filter(
|
2010-07-02 18:02:26 -07:00
|
|
|
protocol__protocol__in=protocols,
|
2009-10-31 08:22:09 -07:00
|
|
|
mirror__public=True, mirror__active=True, mirror__isos=True
|
|
|
|
)
|
2010-07-02 17:24:54 -07:00
|
|
|
if countries and 'all' not in countries:
|
|
|
|
qset = qset.filter(mirror__country__in=countries)
|
2010-09-21 16:58:34 -07:00
|
|
|
if not use_status:
|
|
|
|
urls = qset.order_by('mirror__country', 'mirror__name', 'url')
|
|
|
|
template = 'mirrors/mirrorlist.txt'
|
|
|
|
else:
|
2010-09-23 08:33:32 -07:00
|
|
|
status_info = get_mirror_statuses()
|
|
|
|
scores = dict([(u.id, u.score) for u in status_info['urls']])
|
2010-09-21 16:58:34 -07:00
|
|
|
urls = []
|
|
|
|
for u in qset:
|
|
|
|
u.score = scores[u.id]
|
|
|
|
if u.score and u.score < 100.0:
|
|
|
|
urls.append(u)
|
|
|
|
urls = sorted(urls, key=attrgetter('score'))
|
|
|
|
template = 'mirrors/mirrorlist_status.txt'
|
|
|
|
|
|
|
|
return direct_to_template(request, template, {
|
|
|
|
'mirror_urls': urls,
|
2009-10-31 08:22:09 -07:00
|
|
|
},
|
2010-09-03 11:44:27 -07:00
|
|
|
mimetype='text/plain')
|
2009-10-31 08:22:09 -07:00
|
|
|
|
2010-09-23 16:15:00 -07:00
|
|
|
def mirrors(request):
|
|
|
|
mirrors = Mirror.objects.select_related().order_by('tier', 'country')
|
|
|
|
if not request.user.is_authenticated():
|
|
|
|
mirrors = mirrors.filter(public=True, active=True)
|
|
|
|
return direct_to_template(request, 'mirrors/mirrors.html',
|
|
|
|
{'mirror_list': mirrors})
|
2010-09-21 16:58:34 -07:00
|
|
|
|
2010-09-12 08:14:34 -07:00
|
|
|
def status(request):
|
|
|
|
bad_timedelta = datetime.timedelta(days=3)
|
2010-09-23 08:33:32 -07:00
|
|
|
status_info = get_mirror_statuses()
|
2010-09-12 08:14:34 -07:00
|
|
|
|
2010-09-23 08:33:32 -07:00
|
|
|
urls = status_info['urls']
|
2010-09-12 08:14:34 -07:00
|
|
|
good_urls = []
|
|
|
|
bad_urls = []
|
|
|
|
for url in urls:
|
|
|
|
# split them into good and bad lists based on delay
|
|
|
|
if not url.delay or url.delay > bad_timedelta:
|
|
|
|
bad_urls.append(url)
|
|
|
|
else:
|
|
|
|
good_urls.append(url)
|
|
|
|
|
2010-09-23 08:33:32 -07:00
|
|
|
context = status_info.copy()
|
|
|
|
context.update({
|
2010-09-12 08:14:34 -07:00
|
|
|
'good_urls': good_urls,
|
|
|
|
'bad_urls': bad_urls,
|
2010-09-21 16:31:26 -07:00
|
|
|
'error_logs': get_mirror_errors(),
|
2010-09-23 08:33:32 -07:00
|
|
|
})
|
2010-09-12 08:14:34 -07:00
|
|
|
return direct_to_template(request, 'mirrors/status.html', context)
|
|
|
|
|
2009-10-31 08:22:09 -07:00
|
|
|
# vim: set ts=4 sw=4 et:
|