Beispiel #1
0
def lookup_users(qs):
    user_ids = queryset_to_dict(qs, key='user_id', singular=False)

    users = queryset_to_dict(get_user_model().objects.filter(id__in=user_ids.keys()))

    try:
        Profile = get_profile_model()
        profiles = queryset_to_dict(Profile.objects.filter(user__in=users.keys()), key='user_id')
    except SiteProfileNotAvailable:
        profiles = {}

    for user_id, objs in user_ids.iteritems():
        if user_id in users:
            user = users[user_id]

            for obj in objs:
                obj.user = user

                if user_id in profiles:
                    obj.user._profile_cache = profiles[user_id]
                    obj.user._profile_cache.user = obj.user

        else:
            for obj in objs:
                obj.user_id = None
Beispiel #2
0
def lookup_post_attachments(qs):
    post_ids = queryset_to_dict(qs, key='pk')

    attachments = queryset_to_dict(Attachment.objects.filter(post__in=post_ids.keys()), key='post_id', singular=False)

    for post_id, post in post_ids.iteritems():
        if post_id in attachments:
            post_ids[post_id]._attachments = attachments[post_id]
        else:
            post_ids[post_id]._attachments = []

    return qs
Beispiel #3
0
def lookup_post_topics(qs):
    topic_ids = queryset_to_dict(qs, key='topic_id', singular=False)

    topics = queryset_to_dict(Topic.objects.filter(id__in=topic_ids.keys()).select_related('forum'))

    for topic_id, posts in topic_ids.iteritems():
        topic = None

        if topic_id in topics:
            topic = topics[topic_id]

            for post in posts:
                post.topic = topic
Beispiel #4
0
def lookup_forum_lastposts(qs):
    post_ids = queryset_to_dict(qs, key='last_post_id')

    for post in Post.objects.filter(pk__in=post_ids.keys()).select_related('user', 'topic'):
        post_ids[post.pk].last_post = post
        post_ids[post.pk].last_post.topic.forum = post_ids[post.pk]

    return qs
Beispiel #5
0
def lookup_topic_lastposts(qs, user=True):
    post_ids = queryset_to_dict(qs, key='last_post_id')

    posts = Post.objects.filter(pk__in=post_ids.keys())

    if user:
        posts = posts.select_related('user')

    for post in posts:
        post_ids[post.pk].last_post = post
        post_ids[post.pk].last_post.topic = post_ids[post.pk]

    return qs