boards/abstracts/sync_filters.py

import xml.etree.ElementTree as et

from boards.models import Post, Tag

TAG_THREAD = 'thread'
TAG_TAGS = 'tags'
TAG_TAG = 'tag'
TAG_TIME_FROM = 'timestamp_from'


class PostFilter:
    def __init__(self, content=None):
        self.content = content

    def filter(self, posts):
        return posts

    def add_filter(self, model_tag, value):
        return model_tag


class ThreadFilter(PostFilter):
    def filter(self, posts):
        op_id = self.content.text

        op = Post.objects.filter(opening=True, id=op_id).first()
        if op:
            return posts.filter(thread=op.get_thread())
        else:
            return posts.none()

    def add_filter(self, model_tag, value):
        thread_tag = et.SubElement(model_tag, TAG_THREAD)
        thread_tag.text = str(value)


class TagsFilter(PostFilter):
    def filter(self, posts):
        tags = []
        for tag_tag in self.content:
            try:
                tags.append(Tag.objects.get(name=tag_tag.text))
            except Tag.DoesNotExist:
                pass

        if tags:
            return posts.filter(thread__tags__in=tags)
        else:
            return posts.none()

    def add_filter(self, model_tag, value):
        tags_tag = et.SubElement(model_tag, TAG_TAGS)
        for tag_name in value:
            tag_tag = et.SubElement(tags_tag, TAG_TAG)
            tag_tag.text = tag_name


class TimestampFromFilter(PostFilter):
    def filter(self, posts):
        from_time = self.content.text
        return posts.filter(pub_time__gt=from_time)

    def add_filter(self, model_tag, value):
        tags_from_time = et.SubElement(model_tag, TAG_TIME_FROM)
        tags_from_time.text = value