blob: d27089a360ee2c30425ef31b8748208f3d89b917 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
"""Version 1 of API helper for the south orm
usage of this api implementation starts with migration 24
"""
from askbot.migrations_api import BaseAPI
from django.db import models
#from django.contrib.contenttypes.models import ContentType
class API(BaseAPI):
def get_origin_post_from_content_object(self, parent):
"""follows relations from the parent object
until origin post (question) is found
starting point is always ``parent.content_object``
"""
model = parent.content_type.model
id = parent.object_id
#print 'model is ' + model
if model == 'question':
return self.orm.Question.objects.get(id=id)
elif model == 'answer':
return self.orm.Answer.objects.get(id=id).question
elif model == 'favoritequestion':
try:
return self.orm.FavoriteQuestion.objects.get(id=id).question
except self.orm.FavoriteQuestion.DoesNotExist:
#ignore this issue for now
return None
elif model == 'answerrevision':
return self.orm.AnswerRevision.objects.get(id=id).answer.question
elif model == 'questionrevision':
return self.orm.QuestionRevision.objects.get(id=id).question
elif model == 'comment':
comment = self.orm.Comment.objects.get(id=id)
return self.get_origin_post_from_content_object(comment)
else:
#print 'dropped migration of activity in %s' % model
return None
def get_moderators_and_admins(self):
"""return list of forum moderators and django site admins
"""
filter_expression = models.Q(status='m') | models.Q(is_superuser=True)
return self.orm['auth.User'].objects.filter(filter_expression)
def get_activity_items_for_object(self, instance):
"""get all activity items that have content_object set
where ``<Activity instance>.content_object == instance``
"""
return self.orm.Activity.objects.filter(
object_id = instance.id,
content_type = self.get_content_type_for_model(instance)
)
def get_content_type_for_model(self, instance):
ct = self.orm['contenttypes.ContentType'].objects
return ct.get(app_label='askbot', model=instance._meta.object_name.lower())
def add_recipients_to_activity(self, recipients, activity):
for recipient in recipients:
memo = self.orm.ActivityAuditStatus(
user = recipient,
activity = activity,
)
memo.save()
|
