def new_message(message, inbox_rels): from r2.lib.comment_tree import add_message from_user = Account._byID(message.author_id) for inbox_rel in tup(inbox_rels): to = inbox_rel._thing1 # moderator message if isinstance(inbox_rel, ModeratorInbox): add_queries([get_subreddit_messages(to)], insert_items=inbox_rel) # personal message else: add_queries([get_sent(from_user)], insert_items=message) add_queries([get_inbox_messages(to)], insert_items=inbox_rel) set_unread(message, to, True) add_message(message)
def new_message(message, inbox_rels): from r2.lib.comment_tree import add_message from_user = Account._byID(message.author_id) for inbox_rel in tup(inbox_rels): to = inbox_rel._thing1 add_queries([get_sent(from_user)], insert_items=message) with CachedQueryMutator() as m: # moderator message if isinstance(inbox_rel, ModeratorInbox): m.insert(get_subreddit_messages(to), [inbox_rel]) # personal message else: m.insert(get_inbox_messages(to), [inbox_rel]) set_unread(message, to, unread=True, mutator=m) add_message(message)