forum/models/user.py

   1 from base import *\r
   2 from forum import const\r
   3 from django.contrib.contenttypes.models import ContentType\r
   4 from django.contrib.auth.models import User as DjangoUser, AnonymousUser as DjangoAnonymousUser\r
   5 from django.db.models import Q\r
   6 try:\r
   7     from hashlib import md5\r
   8 except:\r
   9     from md5 import new as md5\r
  10 \r
  11 import string\r
  12 from random import Random\r
  13 \r
  14 from django.utils.translation import ugettext as _\r
  15 import django.dispatch\r
  16 \r
  17 \r
  18 QUESTIONS_PER_PAGE_CHOICES = (\r
  19    (10, u'10'),\r
  20    (30, u'30'),\r
  21    (50, u'50'),\r
  22 )\r
  23 \r
  24 class UserManager(CachedManager):\r
  25     def get_site_owner(self):\r
  26         return self.all().order_by('date_joined')[0]\r
  27 \r
  28 class AnonymousUser(DjangoAnonymousUser):\r
  29     def get_visible_answers(self, question):\r
  30         return question.answers.filter(deleted=None)\r
  31 \r
  32     def can_view_deleted_post(self, post):\r
  33         return False\r
  34 \r
  35     def can_vote_up(self):\r
  36         return False\r
  37 \r
  38     def can_vote_down(self):\r
  39         return False\r
  40 \r
  41     def can_flag_offensive(self, post=None):\r
  42         return False\r
  43 \r
  44     def can_view_offensive_flags(self, post=None):\r
  45         return False\r
  46 \r
  47     def can_comment(self, post):\r
  48         return False\r
  49 \r
  50     def can_like_comment(self, comment):\r
  51         return False\r
  52 \r
  53     def can_edit_comment(self, comment):\r
  54         return False\r
  55 \r
  56     def can_delete_comment(self, comment):\r
  57         return False\r
  58 \r
  59     def can_accept_answer(self, answer):\r
  60         return False\r
  61 \r
  62     def can_edit_post(self, post):\r
  63         return False\r
  64 \r
  65     def can_retag_questions(self):\r
  66         return False\r
  67 \r
  68     def can_close_question(self, question):\r
  69         return False\r
  70 \r
  71     def can_reopen_question(self, question):\r
  72         return False\r
  73 \r
  74     def can_delete_post(self, post):\r
  75         return False\r
  76 \r
  77     def can_upload_files(self):\r
  78         return False\r
  79 \r
  80 def true_if_is_super_or_staff(fn):\r
  81     def decorated(self, *args, **kwargs):\r
  82         return self.is_superuser or self.is_staff or fn(self, *args, **kwargs)\r
  83     return decorated\r
  84 \r
  85 class User(BaseModel, DjangoUser):\r
  86     is_approved = models.BooleanField(default=False)\r
  87     email_isvalid = models.BooleanField(default=False)\r
  88 \r
  89     reputation = models.PositiveIntegerField(default=0)\r
  90     gold = models.PositiveIntegerField(default=0)\r
  91     silver = models.PositiveIntegerField(default=0)\r
  92     bronze = models.PositiveIntegerField(default=0)\r
  93     \r
  94     last_seen = models.DateTimeField(default=datetime.datetime.now)\r
  95     real_name = models.CharField(max_length=100, blank=True)\r
  96     website = models.URLField(max_length=200, blank=True)\r
  97     location = models.CharField(max_length=100, blank=True)\r
  98     date_of_birth = models.DateField(null=True, blank=True)\r
  99     about = models.TextField(blank=True)\r
 100 \r
 101     subscriptions = models.ManyToManyField('Node', related_name='subscribers', through='QuestionSubscription')\r
 102 \r
 103     vote_up_count = DenormalizedField("actions", canceled=False, action_type="voteup")\r
 104     vote_down_count = DenormalizedField("actions", canceled=False, action_type="votedown")\r
 105    \r
 106     objects = UserManager()\r
 107 \r
 108     @property\r
 109     def gravatar(self):\r
 110         return md5(self.email).hexdigest()\r
 111 \r
 112     def save(self, *args, **kwargs):\r
 113         if self.reputation < 0:\r
 114             self.reputation = 0\r
 115 \r
 116         new = not bool(self.id)\r
 117 \r
 118         super(User, self).save(*args, **kwargs)\r
 119 \r
 120         if new:\r
 121             sub_settings = SubscriptionSettings(user=self)\r
 122             sub_settings.save()\r
 123 \r
 124     def get_absolute_url(self):\r
 125         return self.get_profile_url()\r
 126 \r
 127     def get_messages(self):\r
 128         messages = []\r
 129         for m in self.message_set.all():\r
 130             messages.append(m.message)\r
 131         return messages\r
 132 \r
 133     def delete_messages(self):\r
 134         self.message_set.all().delete()\r
 135 \r
 136     def get_profile_url(self):\r
 137         return "/%s%d/%s" % (_('users/'), self.id, slugify(self.username))\r
 138 \r
 139     def get_profile_link(self):\r
 140         profile_link = u'<a href="%s">%s</a>' % (self.get_profile_url(),self.username)\r
 141         return mark_safe(profile_link)\r
 142 \r
 143     def get_visible_answers(self, question):\r
 144         return question.answers.filter(deleted=None, in_moderation=None)\r
 145 \r
 146     def get_vote_count_today(self):\r
 147         today = datetime.date.today()\r
 148         return self.actions.filter(canceled=False, action_type__in=("voteup", "votedown"),\r
 149                 action_date__range=(today - datetime.timedelta(days=1), today)).count()\r
 150 \r
 151     def get_reputation_by_upvoted_today(self):\r
 152         today = datetime.datetime.now()\r
 153         sum = self.reputes.filter(\r
 154                 models.Q(reputation_type=TYPE_REPUTATION_GAIN_BY_UPVOTED) |\r
 155                 models.Q(reputation_type=TYPE_REPUTATION_LOST_BY_UPVOTE_CANCELED),\r
 156                 reputed_at__range=(today - datetime.timedelta(days=1), today)).aggregate(models.Sum('value'))\r
 157 \r
 158         if sum.get('value__sum', None) is not None: return sum['value__sum']\r
 159         return 0\r
 160 \r
 161     def get_flagged_items_count_today(self):\r
 162         today = datetime.date.today()\r
 163         return self.actions.filter(canceled=False, action_type="flag",\r
 164                 action_date__range=(today - datetime.timedelta(days=1), today)).count()\r
 165 \r
 166     @true_if_is_super_or_staff\r
 167     def can_view_deleted_post(self, post):\r
 168         return post.author == self\r
 169 \r
 170     @true_if_is_super_or_staff\r
 171     def can_vote_up(self):\r
 172         return self.reputation >= int(settings.REP_TO_VOTE_UP)\r
 173 \r
 174     @true_if_is_super_or_staff\r
 175     def can_vote_down(self):\r
 176         return self.reputation >= int(settings.REP_TO_VOTE_DOWN)\r
 177 \r
 178     def can_flag_offensive(self, post=None):\r
 179         if post is not None and post.author == self:\r
 180             return False\r
 181         return self.is_superuser or self.is_staff or self.reputation >= int(settings.REP_TO_FLAG)\r
 182 \r
 183     @true_if_is_super_or_staff\r
 184     def can_view_offensive_flags(self, post=None):\r
 185         if post is not None and post.author == self:\r
 186             return True\r
 187         return self.reputation >= int(settings.REP_TO_VIEW_FLAGS)\r
 188 \r
 189     @true_if_is_super_or_staff\r
 190     def can_comment(self, post):\r
 191         return self == post.author or self.reputation >= int(settings.REP_TO_COMMENT\r
 192         ) or (post.__class__.__name__ == "Answer" and self == post.question.author)\r
 193 \r
 194     @true_if_is_super_or_staff\r
 195     def can_like_comment(self, comment):\r
 196         return self != comment.author and (self.reputation >= int(settings.REP_TO_LIKE_COMMENT))\r
 197 \r
 198     @true_if_is_super_or_staff\r
 199     def can_edit_comment(self, comment):\r
 200         return (comment.author == self and comment.added_at >= datetime.datetime.now() - datetime.timedelta(minutes=60)\r
 201         ) or self.is_superuser\r
 202 \r
 203     @true_if_is_super_or_staff\r
 204     def can_delete_comment(self, comment):\r
 205         return self == comment.author or self.reputation >= int(settings.REP_TO_DELETE_COMMENTS)\r
 206 \r
 207     @true_if_is_super_or_staff\r
 208     def can_accept_answer(self, answer):\r
 209         return self == answer.question.author\r
 210 \r
 211     @true_if_is_super_or_staff\r
 212     def can_edit_post(self, post):\r
 213         return self == post.author or self.reputation >= int(settings.REP_TO_EDIT_OTHERS\r
 214         ) or (post.wiki and self.reputation >= int(settings.REP_TO_EDIT_WIKI))\r
 215 \r
 216     @true_if_is_super_or_staff\r
 217     def can_retag_questions(self):\r
 218         return self.reputation >= int(settings.REP_TO_RETAG)\r
 219 \r
 220     @true_if_is_super_or_staff\r
 221     def can_close_question(self, question):\r
 222         return (self == question.author and self.reputation >= int(settings.REP_TO_CLOSE_OWN)\r
 223         ) or self.reputation >= int(settings.REP_TO_CLOSE_OTHERS)\r
 224 \r
 225     @true_if_is_super_or_staff\r
 226     def can_reopen_question(self, question):\r
 227         return self == question.author and self.reputation >= settings.REP_TO_REOPEN_OWN\r
 228 \r
 229     @true_if_is_super_or_staff\r
 230     def can_delete_post(self, post):\r
 231         if post.node_type == "comment":\r
 232             return self.can_delete_comment(post)\r
 233             \r
 234         return (self == post.author and (post.__class__.__name__ == "Answer" or\r
 235             not post.answers.exclude(author=self).count()))\r
 236 \r
 237     @true_if_is_super_or_staff\r
 238     def can_upload_files(self):\r
 239         return self.reputation >= int(settings.REP_TO_UPLOAD)\r
 240 \r
 241     class Meta:\r
 242         app_label = 'forum'\r
 243 \r
 244 class SubscriptionSettings(models.Model):\r
 245     user = models.OneToOneField(User, related_name='subscription_settings')\r
 246 \r
 247     enable_notifications = models.BooleanField(default=True)\r
 248 \r
 249     #notify if\r
 250     member_joins = models.CharField(max_length=1, default='n', choices=const.NOTIFICATION_CHOICES)\r
 251     new_question = models.CharField(max_length=1, default='d', choices=const.NOTIFICATION_CHOICES)\r
 252     new_question_watched_tags = models.CharField(max_length=1, default='i', choices=const.NOTIFICATION_CHOICES)\r
 253     subscribed_questions = models.CharField(max_length=1, default='i', choices=const.NOTIFICATION_CHOICES)\r
 254     \r
 255     #auto_subscribe_to\r
 256     all_questions = models.BooleanField(default=False)\r
 257     all_questions_watched_tags = models.BooleanField(default=False)\r
 258     questions_asked = models.BooleanField(default=True)\r
 259     questions_answered = models.BooleanField(default=True)\r
 260     questions_commented = models.BooleanField(default=False)\r
 261     questions_viewed = models.BooleanField(default=False)\r
 262 \r
 263     #notify activity on subscribed\r
 264     notify_answers = models.BooleanField(default=True)\r
 265     notify_reply_to_comments = models.BooleanField(default=True)\r
 266     notify_comments_own_post = models.BooleanField(default=True)\r
 267     notify_comments = models.BooleanField(default=False)\r
 268     notify_accepted = models.BooleanField(default=False)\r
 269 \r
 270     class Meta:\r
 271         app_label = 'forum'\r
 272 \r
 273 from forum.utils.time import one_day_from_now\r
 274 \r
 275 class ValidationHashManager(models.Manager):\r
 276     def _generate_md5_hash(self, user, type, hash_data, seed):\r
 277         return md5("%s%s%s%s" % (seed, "".join(map(str, hash_data)), user.id, type)).hexdigest()\r
 278 \r
 279     def create_new(self, user, type, hash_data=[], expiration=None):\r
 280         seed = ''.join(Random().sample(string.letters+string.digits, 12))\r
 281         hash = self._generate_md5_hash(user, type, hash_data, seed)\r
 282 \r
 283         obj = ValidationHash(hash_code=hash, seed=seed, user=user, type=type)\r
 284 \r
 285         if expiration is not None:\r
 286             obj.expiration = expiration\r
 287 \r
 288         try:\r
 289             obj.save()\r
 290         except:\r
 291             return None\r
 292             \r
 293         return obj\r
 294 \r
 295     def validate(self, hash, user, type, hash_data=[]):\r
 296         try:\r
 297             obj = self.get(hash_code=hash)\r
 298         except:\r
 299             return False\r
 300 \r
 301         if obj.type != type:\r
 302             return False\r
 303 \r
 304         if obj.user != user:\r
 305             return False\r
 306 \r
 307         valid = (obj.hash_code == self._generate_md5_hash(obj.user, type, hash_data, obj.seed))\r
 308 \r
 309         if valid:\r
 310             if obj.expiration < datetime.datetime.now():\r
 311                 obj.delete()\r
 312                 return False\r
 313             else:\r
 314                 obj.delete()\r
 315                 return True\r
 316 \r
 317         return False\r
 318 \r
 319 class ValidationHash(models.Model):\r
 320     hash_code = models.CharField(max_length=255,unique=True)\r
 321     seed = models.CharField(max_length=12)\r
 322     expiration = models.DateTimeField(default=one_day_from_now)\r
 323     type = models.CharField(max_length=12)\r
 324     user = models.ForeignKey(User)\r
 325 \r
 326     objects = ValidationHashManager()\r
 327 \r
 328     class Meta:\r
 329         unique_together = ('user', 'type')\r
 330         app_label = 'forum'\r
 331 \r
 332     def __str__(self):\r
 333         return self.hash_code\r
 334 \r
 335 class AuthKeyUserAssociation(models.Model):\r
 336     key = models.CharField(max_length=255,null=False,unique=True)\r
 337     provider = models.CharField(max_length=64)\r
 338     user = models.ForeignKey(User, related_name="auth_keys")\r
 339     added_at = models.DateTimeField(default=datetime.datetime.now)\r
 340 \r
 341     class Meta:\r
 342         app_label = 'forum'\r