171 lines
4.6 KiB
Python
171 lines
4.6 KiB
Python
"""
|
|
djeuscan.managers
|
|
"""
|
|
|
|
from django.db import models
|
|
from djeuscan.helpers import xint, rename_fields, select_related_last_versions
|
|
|
|
|
|
def _gen_n_function(field_name):
|
|
def n_method(self):
|
|
res = self.aggregate(models.Sum(field_name))[field_name + '__sum']
|
|
return xint(res)
|
|
n_method.func_name = field_name
|
|
return n_method
|
|
|
|
|
|
def _gen_for_function(field):
|
|
def for_method(self, val, last_versions=False):
|
|
"""
|
|
Returns packages that belong to the given parametrs
|
|
"""
|
|
res = self.filter(**{field: val})
|
|
|
|
if last_versions:
|
|
select_related_last_versions(res)
|
|
|
|
return res
|
|
|
|
for_method.func_name = 'for_' + field
|
|
return for_method
|
|
|
|
|
|
ANNOTATE_DICT = {name: models.Sum(name)
|
|
for name in ['n_packaged', 'n_overlay', 'n_versions']}
|
|
|
|
|
|
class PackageMixin(object):
|
|
|
|
for_maintainer = _gen_for_function('maintainers')
|
|
for_herd = _gen_for_function('herds')
|
|
for_category = _gen_for_function('category')
|
|
|
|
n_packaged = _gen_n_function("n_packaged")
|
|
n_overlay = _gen_n_function("n_overlay")
|
|
n_versions = _gen_n_function("n_versions")
|
|
|
|
def n_upstream(self):
|
|
return self.n_versions() - self.n_packaged() - self.n_overlay()
|
|
|
|
def categories(self):
|
|
"""
|
|
Returns all the available categories
|
|
"""
|
|
return self.values('category').annotate(**ANNOTATE_DICT)
|
|
|
|
def herds(self, rename=False):
|
|
"""
|
|
Returns all the available herds
|
|
"""
|
|
# FIXME: optimize the query, it uses 'LEFT OUTER JOIN' instead of
|
|
# 'INNER JOIN'
|
|
res = self.filter(herds__isnull=False)
|
|
res = res.values('herds__herd').annotate(**ANNOTATE_DICT)
|
|
|
|
if rename:
|
|
res = rename_fields(res, [('herds__herd', 'herd')])
|
|
|
|
return res
|
|
|
|
def maintainers(self, rename=False):
|
|
"""
|
|
Returns all the available maintainers
|
|
"""
|
|
res = self.filter(maintainers__isnull=False).values(
|
|
'maintainers__id', 'maintainers__name', 'maintainers__email'
|
|
)
|
|
res = res.annotate(**ANNOTATE_DICT)
|
|
|
|
if rename:
|
|
res = rename_fields(
|
|
res,
|
|
[('maintainers__id', 'id'),
|
|
('maintainers__name', 'name'),
|
|
('maintainers__email', 'email')]
|
|
)
|
|
|
|
return res
|
|
|
|
def overlays(self):
|
|
"""
|
|
Returns the all available overlays
|
|
"""
|
|
res = self.values('version__overlay').exclude(version__overlay='')
|
|
return res.distinct()
|
|
|
|
def for_overlay(self, overlay):
|
|
"""
|
|
Returns packages that belong to the given overlay
|
|
"""
|
|
packages = self.values(
|
|
'id', 'name', 'category', 'n_versions', 'n_packaged', 'n_overlay'
|
|
)
|
|
return packages.filter(version__overlay=overlay).distinct()
|
|
|
|
|
|
class PackageQuerySet(models.query.QuerySet, PackageMixin):
|
|
pass
|
|
|
|
|
|
class PackageManager(models.Manager, PackageMixin):
|
|
def get_query_set(self):
|
|
return PackageQuerySet(self.model, using=self._db)
|
|
|
|
|
|
class VersionLogMixin(object):
|
|
def for_package(self, package, order=False):
|
|
res = self.filter(package=package)
|
|
if order:
|
|
res = res.order_by('-id')
|
|
return res
|
|
|
|
def for_maintainer(self, maintainer, order=False):
|
|
res = self.filter(package__maintainers__id=maintainer.id)
|
|
if order:
|
|
res = res.order_by('-id')
|
|
return res
|
|
|
|
def for_category(self, category, order=False):
|
|
res = self.filter(package__category=category)
|
|
if order:
|
|
res = res.order_by('-id')
|
|
return res
|
|
|
|
def for_herd(self, herd, order=False):
|
|
res = self.filter(package__herds__id=herd.id)
|
|
if order:
|
|
res = res.order_by('-id')
|
|
return res
|
|
|
|
|
|
class VersionLogQuerySet(models.query.QuerySet, VersionLogMixin):
|
|
pass
|
|
|
|
|
|
class VersionLogManager(models.Manager, VersionLogMixin):
|
|
def get_query_set(self):
|
|
return VersionLogQuerySet(self.model, using=self._db)
|
|
|
|
|
|
class EuscanResultMixin(object):
|
|
def for_package(self, package):
|
|
return self.filter(package=package)
|
|
|
|
def for_maintainer(self, maintainer):
|
|
return self.filter(package__maintainers__id=maintainer.id)
|
|
|
|
def for_category(self, category):
|
|
return self.filter(package__category=category)
|
|
|
|
def for_herd(self, herd):
|
|
return self.filter(package__herds__id=herd.id)
|
|
|
|
|
|
class EuscanResultQuerySet(models.query.QuerySet, EuscanResultMixin):
|
|
pass
|
|
|
|
|
|
class EuscanResultManager(models.Manager, EuscanResultMixin):
|
|
def get_query_set(self):
|
|
return EuscanResultQuerySet(self.model, using=self._db)
|