diff --git a/djangoratings/fields.py b/djangoratings/fields.py index 419e582..a583871 100644 --- a/djangoratings/fields.py +++ b/djangoratings/fields.py @@ -357,11 +357,11 @@ def contribute_to_class(self, cls, name): setattr(cls, name, field) - def get_db_prep_save(self, value): + def get_db_prep_save(self, value, connection=None): # XXX: what happens here? pass - def get_db_prep_lookup(self, lookup_type, value): + def get_db_prep_lookup(self, lookup_type, value, connection=None, prepared=False): # TODO: hack in support for __score and __votes # TODO: order_by on this field should use the weighted algorithm raise NotImplementedError(self.get_db_prep_lookup) @@ -369,9 +369,9 @@ def get_db_prep_lookup(self, lookup_type, value): # lookup_type = # return self.score_field.get_db_prep_lookup() if lookup_type == 'exact': - return [self.get_db_prep_save(value)] + return [self.get_db_prep_save(value, connection)] elif lookup_type == 'in': - return [self.get_db_prep_save(v) for v in value] + return [self.get_db_prep_save(v, connection) for v in value] else: return super(RatingField, self).get_db_prep_lookup(lookup_type, value)