addons-server/apps/bandwagon/tasks.py

76 строки
2.5 KiB
Python
Исходник Обычный вид История

2010-07-22 01:58:57 +04:00
import logging
import math
2010-07-27 22:35:26 +04:00
import os
2010-07-22 01:58:57 +04:00
from django.db.models import Count
from celeryutils import task
2010-07-31 04:18:25 +04:00
from easy_thumbnails import processors
from PIL import Image
2010-07-22 01:58:57 +04:00
2010-07-27 00:14:45 +04:00
import amo
from tags.models import Tag
2010-07-22 01:58:57 +04:00
from . import cron # Pull in tasks run through cron.
2010-07-27 00:14:45 +04:00
from .models import Collection, CollectionAddon, CollectionVote
2010-07-22 01:58:57 +04:00
log = logging.getLogger('z.task')
@task
2010-08-05 22:55:08 +04:00
def collection_votes(*ids, **kw):
2010-07-22 01:58:57 +04:00
log.info('[%s@%s] Updating collection votes.' %
(len(ids), collection_votes.rate_limit))
2010-08-05 22:55:08 +04:00
using = kw.get('using')
2010-07-22 01:58:57 +04:00
for collection in ids:
2010-08-05 22:55:08 +04:00
v = CollectionVote.objects.filter(collection=collection).using(using)
2010-07-22 01:58:57 +04:00
votes = dict(v.values_list('vote').annotate(Count('vote')))
2010-08-05 22:55:08 +04:00
c = Collection.objects.get(id=collection)
c.upvotes = up = votes.get(1, 0)
c.downvotes = down = votes.get(-1, 0)
try:
# Use log to limit the effect of the multiplier.
c.rating = (up - down) * math.log(up + down)
except ValueError:
c.rating = 0
2010-08-05 22:55:08 +04:00
c.save()
2010-07-27 22:35:26 +04:00
2010-07-31 04:18:25 +04:00
@task
2010-08-21 00:40:49 +04:00
def resize_icon(src, dst):
2010-07-31 04:18:25 +04:00
"""Resizes collection icons to 32x32"""
2010-08-21 01:10:46 +04:00
log.info('[1@None] Resizing icon: %s' % dst)
2010-08-18 00:37:03 +04:00
2010-07-31 04:18:25 +04:00
try:
im = Image.open(src)
im = processors.scale_and_crop(im, (32, 32))
2010-08-21 00:40:49 +04:00
im.save(dst)
2010-07-31 04:18:25 +04:00
os.remove(src)
except Exception, e:
log.error("Error saving collection icon: %s" % e)
2010-07-27 00:14:45 +04:00
@task
2010-08-11 07:33:08 +04:00
def collection_meta(*ids, **kw):
2010-07-27 00:14:45 +04:00
log.info('[%s@%s] Updating collection metadata.' %
(len(ids), collection_meta.rate_limit))
2010-08-11 07:33:08 +04:00
using = kw.get('using')
2010-07-27 00:14:45 +04:00
qs = (CollectionAddon.objects.filter(collection__in=ids)
2010-08-11 07:33:08 +04:00
.using(using).values_list('collection'))
2010-07-27 00:14:45 +04:00
counts = dict(qs.annotate(Count('id')))
persona_counts = dict(qs.filter(addon__type=amo.ADDON_PERSONA)
.annotate(Count('id')))
tags = (Tag.objects.not_blacklisted().values_list('id')
.annotate(cnt=Count('id')).filter(cnt__gt=1).order_by('-cnt'))
2010-08-11 07:33:08 +04:00
for c in Collection.objects.no_cache().filter(id__in=ids):
addon_count = counts.get(c.id, 0)
all_personas = addon_count == persona_counts.get(c.id, None)
addons = list(c.addons.values_list('id', flat=True))
c.top_tags = [t for t, _ in tags.filter(addons__in=addons)[:5]]
2010-08-11 07:33:08 +04:00
Collection.objects.filter(id=c.id).update(addon_count=addon_count,
all_personas=all_personas)
@task(rate_limit='10/m')
def cron_collection_meta(*addons):
collection_meta(*addons)