66 строки
1.8 KiB
Python
66 строки
1.8 KiB
Python
from django.conf import settings
|
|
from django.core.files.storage import default_storage as storage
|
|
|
|
import commonware.log
|
|
from celeryutils import task
|
|
from lib.es.utils import index_objects
|
|
|
|
from amo.decorators import set_modified_on
|
|
from amo.utils import resize_image
|
|
|
|
from .models import UserProfile
|
|
from . import search
|
|
|
|
task_log = commonware.log.getLogger('z.task')
|
|
|
|
|
|
@task
|
|
def delete_photo(dst, **kw):
|
|
task_log.debug('[1@None] Deleting photo: %s.' % dst)
|
|
|
|
if not dst.startswith(settings.USERPICS_PATH):
|
|
task_log.error("Someone tried deleting something they shouldn't: %s"
|
|
% dst)
|
|
return
|
|
|
|
try:
|
|
storage.delete(dst)
|
|
except Exception, e:
|
|
task_log.error("Error deleting userpic: %s" % e)
|
|
|
|
|
|
@task
|
|
@set_modified_on
|
|
def resize_photo(src, dst, locally=False, **kw):
|
|
"""Resizes userpics to 200x200"""
|
|
task_log.debug('[1@None] Resizing photo: %s' % dst)
|
|
|
|
try:
|
|
resize_image(src, dst, (200, 200), locally=locally)
|
|
return True
|
|
except Exception, e:
|
|
task_log.error("Error saving userpic: %s" % e)
|
|
|
|
|
|
@task
|
|
def index_users(ids, **kw):
|
|
task_log.debug('Indexing users %s-%s [%s].' % (ids[0], ids[-1], len(ids)))
|
|
index = kw.pop('index', None)
|
|
index_objects(ids, UserProfile, search, index)
|
|
|
|
|
|
@task
|
|
def unindex_users(ids, **kw):
|
|
for id in ids:
|
|
task_log.debug('Removing user [%s] from search index.' % id)
|
|
UserProfile.unindex(id)
|
|
|
|
|
|
@task(rate_limit='15/m')
|
|
def update_user_ratings_task(data, **kw):
|
|
task_log.info("[%s@%s] Updating add-on author's ratings." %
|
|
(len(data), update_user_ratings_task.rate_limit))
|
|
for pk, rating in data:
|
|
rating = "%.2f" % round(rating, 2)
|
|
UserProfile.objects.filter(pk=pk).update(averagerating=rating)
|