post.py

import datetime
from anysearch.search_dsl import (
    analyzer,
    Boolean,
    connections,
    Completion,
    Date,
    Document,
    InnerDoc,
    Keyword,
    Nested,
    Text,
    Integer,
    Float,
)
from .read_only import ReadOnlyDocument
from .settings import BLOG_POST_DOCUMENT_NAME, ELASTICSEARCH_CONNECTION

try:
    from elasticsearch import logger
except ImportError:
    import logging
    logger = logging.getLogger(__name__)

__all__ = (
    'Comment',
    'Post',
    'ReadOnlyPost',
)

connections.create_connection(**ELASTICSEARCH_CONNECTION)


html_strip = analyzer(
    'html_strip',
    tokenizer="standard",
    filter=["lowercase", "stop", "snowball"],
    char_filter=["html_strip"]
)


class CommentAuthor(InnerDoc):
    name = Text(fields={'raw': Keyword()})
    age = Integer()


class Comment(InnerDoc):
    author = Nested(CommentAuthor)
    # author = Text(fields={'raw': Keyword()})
    tag = Text(fields={'raw': Keyword()})
    content = Text(analyzer='snowball')
    created_at = Date()

    def age(self):
        return datetime.datetime.now() - self.created_at


class Post(Document):
    title = Text(
        analyzer=html_strip,
        fields={'raw': Keyword()}
    )
    # title_suggest = Completion()
    content = Text()
    created_at = Date()
    published = Boolean()
    category = Text(
        analyzer=html_strip,
        fields={'raw': Keyword()}
    )
    comments = Nested(Comment)
    tags = Text(
        analyzer=html_strip,
        fields={'raw': Keyword(multi=True)},
        multi=True
    )
    num_views = Integer()
    user_id = Integer()

    class Index:
        name = BLOG_POST_DOCUMENT_NAME
        settings = {
            'number_of_shards': 1,
            'number_of_replicas': 1,
            'blocks': {'read_only_allow_delete': None},
        }

    def add_comment(self, author, tag, content):
        self.comments.append(
            Comment(
                author=author,
                content=content,
                tag=tag,
                created_at=datetime.datetime.now()
            )
        )

    def add_tag(self, name):
        self.tags.append(name)

    def save(self, ** kwargs):
        if not self.created_at:
            self.created_at = datetime.datetime.now()
        return super().save(** kwargs)


class ReadOnlyPost(ReadOnlyDocument):
    title = Text(
        analyzer=html_strip,
        fields={'raw': Keyword()}
    )
    # title_suggest = Completion()
    content = Text()
    created_at = Date()
    published = Boolean()
    category = Text(
        analyzer=html_strip,
        fields={'raw': Keyword()}
    )
    comments = Nested(Comment)
    tags = Text(
        analyzer=html_strip,
        fields={'raw': Keyword(multi=True)},
        multi=True
    )
    num_views = Integer()

    class Index:
        name = BLOG_POST_DOCUMENT_NAME
        settings = {
            'number_of_shards': 1,
            'number_of_replicas': 1,
            'blocks': {'read_only_allow_delete': None},
        }


try:
    # Create the mappings in Elasticsearch
    Post.init()
except Exception as err:
    logger.error(err)