45d81a9578
This seems like a more appropriate place, and now the visualization is done here anyway so we should move the data backing it. Signed-off-by: Dan McGee <dan@archlinux.org>
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import json
|
|
|
|
from django.db.models import Count, Sum
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import render
|
|
from django.views.decorators.cache import cache_page
|
|
|
|
from main.models import Package, Arch, Repo
|
|
|
|
|
|
def index(request):
|
|
return render(request, 'visualize/index.html')
|
|
|
|
|
|
def arch_repo_data():
|
|
qs = Package.objects.select_related().values(
|
|
'arch__name', 'repo__name').annotate(
|
|
count=Count('pk'), csize=Sum('compressed_size'),
|
|
isize=Sum('installed_size'),
|
|
flagged=Count('flag_date')).order_by()
|
|
arches = Arch.objects.values_list('name', flat=True)
|
|
repos = Repo.objects.values_list('name', flat=True)
|
|
|
|
def build_map(name, arch, repo):
|
|
key = '%s:%s' % (repo or '', arch or '')
|
|
return {
|
|
'key': key,
|
|
'name': name,
|
|
'arch': arch,
|
|
'repo': repo,
|
|
'data': [],
|
|
}
|
|
|
|
# now transform these results into two mappings: one ordered (repo, arch),
|
|
# and one ordered (arch, repo).
|
|
arch_groups = dict((a, build_map(a, a, None)) for a in arches)
|
|
repo_groups = dict((r, build_map(r, None, r)) for r in repos)
|
|
for row in qs:
|
|
arch = row['arch__name']
|
|
repo = row['repo__name']
|
|
values = {
|
|
'arch': arch,
|
|
'repo': repo,
|
|
'name': '%s (%s)' % (repo, arch),
|
|
'key': '%s:%s' % (repo, arch),
|
|
'csize': row['csize'],
|
|
'isize': row['isize'],
|
|
'count': row['count'],
|
|
'flagged': row['flagged'],
|
|
}
|
|
arch_groups[arch]['data'].append(values)
|
|
repo_groups[repo]['data'].append(values)
|
|
|
|
data = {
|
|
'by_arch': { 'name': 'Architectures', 'data': arch_groups.values() },
|
|
'by_repo': { 'name': 'Repositories', 'data': repo_groups.values() },
|
|
}
|
|
return data
|
|
|
|
|
|
@cache_page(1800)
|
|
def by_arch(request):
|
|
data = arch_repo_data()
|
|
to_json = json.dumps(data['by_arch'], ensure_ascii=False)
|
|
return HttpResponse(to_json, mimetype='application/json')
|
|
|
|
|
|
@cache_page(1800)
|
|
def by_repo(request):
|
|
data = arch_repo_data()
|
|
to_json = json.dumps(data['by_repo'], ensure_ascii=False)
|
|
return HttpResponse(to_json, mimetype='application/json')
|
|
|
|
# vim: set ts=4 sw=4 et:
|