def load_discussion_counter(discussion_id):
try:
counters = DiscussionCounter.get(id=discussion_id)
except DoesNotExist:
counters = default_counter
return counters
def new_discussion(title, text, current_user_id=None, topic=None):
disc_id = idgen_client.get()
post_id = idgen_client.get()
slug = slugify(title)
d = Discussion()
d.id = disc_id
d.post_id = post_id
d.message_count = 1
d.user_id = current_user_id
d.title = title
d.user_count = 1
d.topic_id = topic_slug_to_id(topic) if topic is not None else 0
d.users = {current_user_id}
d.slug = slug
d.save()
disc_counter = DiscussionCounter(id=disc_id)
disc_counter.message_count = 1
disc_counter.user_count = 1
disc_counter.view_count = 1
disc_counter.save()
UserDiscussion.create(user_id=current_user_id, discussion_id=d.id)
new_post(text, code='', current_user_id=current_user_id, post_id=post_id)
return disc_id, slug
def remove_discussion_follower(discussion_id, current_user_id):
try:
follower = DiscussionFollower.objects.filter(
disc_id=discussion_id, user_id=current_user_id).first()
follower.delete()
dc = DiscussionCounter.get(id=discussion_id)
dc.follower_count -= 1
dc.save()
except DoesNotExist:
pass
def load_discussions_by_topic_id(topic_id, topic_name):
discussions = Discussion.objects.filter(
topic_id=topic_id).allow_filtering()
disc_ids = [disc.id for disc in discussions]
counter_data = DiscussionCounter.filter(id__in=disc_ids)
counter_map = {c.id: c for c in counter_data}
for disc in discussions:
disc.topic_name = topic_name
disc.topic_slug = topic_name
disc.counter = counter_map.get(disc.id, default_counter)
return discussions
def new_discussion_message(discussion_id, text, current_user_id, nick=''):
logging.warn("DSCSS:[{}]USER:[{}]".format(discussion_id, current_user_id))
discussion = Discussion.objects.get(id=discussion_id)
p = Post()
p.id = idgen_client.get()
p.discussion_id = discussion_id
p.text = text
p.html = markdown_to_html(text)
p.user_id = current_user_id
p.user_nick = nick
## Create an entry in the timeline to say that this user
# has created a post in the given discussion
# Event.new_post_message
p.save()
Event.message(current_user_id, p.id, None)
# FIXME: Move all this part to the JOB WORKER! Speed my friend.
discussion.last_message = p.id
discussion.users.union({current_user_id})
discussion.save()
UserDiscussion.create(user_id=current_user_id, discussion_id=discussion_id)
dp = DiscussionPost.create(disc_id=discussion_id,
post_id=p.id,
user_id=current_user_id)
dp.save()
disc_counter = DiscussionCounter(id=discussion_id)
disc_counter.message_count += 1
disc_counter.save()
return p.id
def load_discussions():
discussions = Discussion.objects.all().limit(50)
disc_ids = [disc.id for disc in discussions]
logging.warn(disc_ids)
counter_data = DiscussionCounter.filter(id__in=disc_ids)
counter_map = {c.id: c for c in counter_data}
logging.warn("Counter Data: %s", counter_map)
topics = {t.id: t for t in load_topics()}
for disc in discussions:
topic = topics.get(disc.topic_id, None)
if topic is not None:
disc.topic_name = topic.name
disc.topic_slug = topic.slug
disc.counter = counter_map.get(disc.id, default_counter)
return discussions
def discussion_messages(discussion_id,
after_message_id=None,
limit=100,
current_user_id=None):
logging.warn("Requesting: {} After: {}".format(discussion_id,
after_message_id))
discussion = Discussion.objects.get(id=discussion_id)
if after_message_id is not None:
post_filter = DiscussionPost.objects.filter(
disc_id=discussion_id, post_id__gt=after_message_id)
else:
post_filter = DiscussionPost.objects.filter(disc_id=discussion_id)
#FIXME: Here we only get 100 records right now. No Sorting, paging, nothing. Too bad!
disc_post_lists = [(p.post_id, p.user_id)
for p in post_filter.limit(limit)]
post_ids = list(set([x[0] for x in disc_post_lists]))
user_ids = list(set([x[1] for x in disc_post_lists]))
users = load_user_profiles(user_ids)
logging.warn("Looking for posts: {}".format(post_ids))
disc_posts = load_posts(post_ids, current_user_id=current_user_id)
for post in disc_posts:
u = filter(lambda x: x.id == post.user_id, users)
post.user = u[0].to_dict() if u is not None and len(u) > 0 else None
try:
counters = DiscussionCounter.get(id=discussion_id)
except DoesNotExist:
counters = {'message_count': 1, 'user_count': 1, 'view_count': 0}
return discussion, disc_posts, users, counters
def new_discussion_follower(discussion_id, current_user_id, nick=None):
dc = DiscussionCounter.get(id=discussion_id)
dc.follower_count += 1
dc.save()
DiscussionFollower.create(disc_id=discussion_id, user_id=current_user_id)