~widelands-dev/widelands-website/django_staticfiles

from django.contrib.contenttypes.models import ContentType

import itertools

 

class VoteQuerySet(QuerySet):

    def delete(self, *args, **kwargs):

        # XXX: circular import

        from fields import RatingField

 

        to_update = []

        for content_type, objects in itertools.groupby(qs, key=lambda x: x[0]):

            if model_class:

        retval = super(VoteQuerySet, self).delete(*args, **kwargs)

        # TODO: this could be improved

        for obj in to_update:

            for field in getattr(obj, '_djangoratings', []):

                getattr(obj, field.name)._update(commit=False)

            obj.save()

        return retval

class VoteManager(Manager):

    def get_query_set(self):

        return VoteQuerySet(self.model)

 

        if len(objects) > 0:

            ctype = ContentType.objects.get_for_model(objects[0])

            votes = list(self.filter(content_type__pk=ctype.id,

                                                    for obj in objects],

                                     user__pk=user.id))

            vote_dict = dict([(vote.object_id, vote) for vote in votes])

            vote_dict = {}

        return vote_dict

 

class SimilarUserManager(Manager):

    def get_recommendations(self, user, model_class, min_score=1):

        from djangoratings.models import Vote, IgnoredObject

        content_type = ContentType.objects.get_for_model(model_class)

        params = dict(

            v=Vote._meta.db_table,

            sm=self.model._meta.db_table,

            m=model_class._meta.db_table,

            io=IgnoredObject._meta.db_table,

        )

        objects = model_class._default_manager.extra(

            tables=[params['v']],

            where=[

        # ).exclude(

        #     object_id__in=Vote.objects.filter(content_type=content_type, user=user).values_list('object_id', flat=True)

        # ).distinct().values_list('object_id', flat=True))

        return objects

    def update_recommendations(self):

        # TODO: this is mysql only atm

        # TODO: this doesnt handle scores that have multiple values (e.g. 10 points, 5 stars)

            t2=Vote._meta.db_table,

        ))

        cursor.execute('commit')