X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/970b17e67841daa480973c8bdad6d21168b0525d..c6c2f8819fc57f538764f3bccc7580bc6d94e25b:/forum/models/user.py diff --git a/forum/models/user.py b/forum/models/user.py index 34c6b36..fb13c27 100644 --- a/forum/models/user.py +++ b/forum/models/user.py @@ -315,7 +315,7 @@ class User(BaseModel, DjangoUser): return self.can_delete_comment(post) return (self == post.author and (post.__class__.__name__ == "Answer" or - not post.answers.exclude(author=self).count())) + not post.answers.exclude(author__id=self.id).count())) @true_if_is_super_or_staff def can_upload_files(self): @@ -364,7 +364,7 @@ class User(BaseModel, DjangoUser): class Meta: app_label = 'forum' -class UserProperty(models.Model): +class UserProperty(BaseModel): user = models.ForeignKey(User, related_name='properties') key = models.CharField(max_length=16) value = PickledObjectField() @@ -373,6 +373,16 @@ class UserProperty(models.Model): app_label = 'forum' unique_together = ('user', 'key') + def cache_key(self): + return self._generate_cache_key("%s:%s" % (self.user.id, self.key)) + + @classmethod + def infer_cache_key(cls, querydict): + if 'user' in querydict and 'key' in querydict: + return cls._generate_cache_key("%s:%s" % (querydict['user'].id, querydict['key'])) + + return None + class UserPropertyDict(object): def __init__(self, user): self.__dict__['_user'] = user @@ -408,7 +418,7 @@ class UserPropertyDict(object): if current: current.value = value self.__dict__[name] = value - current.save() + current.save(full_save=True) else: user = self.__dict__['_user'] prop = UserProperty(user=user, value=value, key=name)