82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
|
from django.db.models import Q
|
||
|
|
|
||
|
|
|
||
|
|
def get_profile(user):
|
||
|
|
from euscan_accounts.models import UserProfile
|
||
|
|
try:
|
||
|
|
return user.get_profile()
|
||
|
|
except UserProfile.DoesNotExist:
|
||
|
|
UserProfile.objects.create(user=user)
|
||
|
|
return user.get_profile()
|
||
|
|
|
||
|
|
|
||
|
|
def get_account_categories(user):
|
||
|
|
from djeuscan.models import Package
|
||
|
|
# TODO: This is quite ugly
|
||
|
|
category_names = [obj.name for obj in get_profile(user).categories.all()]
|
||
|
|
return [c for c in Package.objects.categories()
|
||
|
|
if c["category"] in category_names]
|
||
|
|
|
||
|
|
|
||
|
|
def get_account_herds(user):
|
||
|
|
from djeuscan.models import Package
|
||
|
|
|
||
|
|
ids = [herd.pk for herd in get_profile(user).herds.all()]
|
||
|
|
return Package.objects.herds(ids=ids)
|
||
|
|
|
||
|
|
|
||
|
|
def get_account_maintainers(user):
|
||
|
|
from djeuscan.models import Package
|
||
|
|
|
||
|
|
ids = [obj.pk for obj in get_profile(user).maintainers.all()]
|
||
|
|
return Package.objects.maintainers(ids=ids)
|
||
|
|
|
||
|
|
|
||
|
|
def get_account_versionlogs(profile):
|
||
|
|
"""
|
||
|
|
Returns all watched packages
|
||
|
|
"""
|
||
|
|
from djeuscan.models import Package, VersionLog
|
||
|
|
|
||
|
|
q_categories = Q(category__in=[
|
||
|
|
category.name for category in profile.categories.all()])
|
||
|
|
q_herds = Q(herds__in=profile.herds.all())
|
||
|
|
q_maintainers = Q(maintainers__in=profile.maintainers.all())
|
||
|
|
packages = list(profile.packages.all()) + list(Package.objects.filter(
|
||
|
|
q_categories | q_herds | q_maintainers))
|
||
|
|
|
||
|
|
overlays = [o.name for o in profile.overlays.all()]
|
||
|
|
|
||
|
|
return VersionLog.objects.filter(
|
||
|
|
Q(package__in=packages) | Q(overlay__in=overlays)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def get_user_fav_infos(user):
|
||
|
|
upstream_k = lambda c: c["n_versions"] - c["n_packaged"] - c["n_overlay"]
|
||
|
|
|
||
|
|
categories = sorted(get_account_categories(user),
|
||
|
|
key=upstream_k, reverse=True)
|
||
|
|
c_upstream = sum([upstream_k(c) for c in categories])
|
||
|
|
herds = sorted(get_account_herds(user),
|
||
|
|
key=upstream_k, reverse=True)
|
||
|
|
h_upstream = sum([upstream_k(c) for c in herds])
|
||
|
|
maintainers = sorted(get_account_maintainers(user),
|
||
|
|
key=upstream_k, reverse=True)
|
||
|
|
m_upstream = sum([upstream_k(c) for c in maintainers])
|
||
|
|
packages = sorted(
|
||
|
|
get_profile(user).packages.all(),
|
||
|
|
key=lambda p: p.n_versions - p.n_packaged - p.n_overlay,
|
||
|
|
reverse=True
|
||
|
|
)
|
||
|
|
p_upstream = sum(
|
||
|
|
[c.n_versions - c.n_packaged - c.n_overlay for c in packages]
|
||
|
|
)
|
||
|
|
|
||
|
|
return {
|
||
|
|
"categories": categories, "categories_upstream": c_upstream,
|
||
|
|
"herds": herds, "herds_upstream": h_upstream,
|
||
|
|
"maintainers": maintainers, "maintainers_upstream": m_upstream,
|
||
|
|
"packages": packages, "packages_upstream": p_upstream,
|
||
|
|
}
|