maple-bbs/maple/topic/controls.py

#!/usr/bin/env python
# -*- coding=UTF-8 -*-
# **************************************************************************
# Copyright © 2016 jianglin
# File Name: controls.py
# Author: jianglin
# Email: xiyang0807@gmail.com
# Created: 2016-06-15 10:22:42 (CST)
# Last Update:星期六 2016-7-30 21:19:24 (CST)
#          By:
# Description:
# **************************************************************************
from flask import current_app
from flask_login import current_user
from sqlalchemy.sql import func
from maple import db
from maple.helpers import make_uid
from maple.main.models import RedisData
from maple.forums.controls import reply as notice_reply
from maple.tag.models import Tags
from .models import Topic, Reply, Like
# from .redis import get_detail_cache
from re import split as sp


def vote(count):
    if count > 0:
        html = '''
                <a id="topic-up-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-up">%d</i>
                </a>
                <a id="topic-down-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-down"></i>
                </a>
        ''' % (count)
    elif count == 0:
        html = '''
                <a id="topic-up-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-up"></i>
                </a>
                <a  id="topic-down-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-down"></i>
                </a>
        '''

    else:
        html = '''
                <a id="topic-up-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-up"></i>
                </a>
                <a  id="topic-down-vote" class="vote" href="javascript:void(0)" style="text-decoration:none;">
                    <i class="icon-chevron-down">%d</i>
                </a>
        ''' % (count)
    return html


class TopicModel(object):
    def get_list(page):
        topics = Topic.query.filter_by(is_top=False).paginate(
            page, current_app.config['PER_PAGE'],
            error_out=True)
        top_topics = Topic.query.filter_by(is_top=True).limit(5).all()
        return topics, top_topics

    def get_detail(page, topicId, order='time'):
        topic = Topic.query.filter_by(uid=topicId).first_or_404()
        if order == 'like':
            replies = Reply.query.outerjoin(Like).filter(
                Reply.topic_id == topic.id).group_by(Reply.id).order_by(
                    func.count(Like.id).desc()).paginate(
                        page, current_app.config['PER_PAGE'], True)
            # replies = Reply.query.filter_by(
            #     topic_id=topic.id).order_by(Reply.publish.asc()).paginate(
            #         page, current_app.config['PER_PAGE'], True)
        else:
            replies = Reply.query.filter_by(
                topic_id=topic.id).order_by(Reply.publish.asc()).paginate(
                    page, current_app.config['PER_PAGE'], True)
        RedisData.set_read_count(topic.id)
        return topic, replies

    def post(form):
        topic = Topic()
        topic.title = form.title.data
        topic.content = form.content.data
        topic.is_markdown = True if form.choice.data == 1 else False
        topic.uid = make_uid()
        topic.author = current_user
        tags = sp(',|;|，|；| ', form.tags.data)
        tags = [x for x in list(set(tags)) if x != ''][:4]
        post_tags = []
        for tag in tags:
            if tag != '':
                exsit_tag = Tags.query.filter_by(tagname=tag).first()
                if exsit_tag is not None:
                    post_tags.append(exsit_tag)
                    if exsit_tag not in current_user.following_tags:
                        current_user.following_tags.append(exsit_tag)
                else:
                    t = Tags()
                    t.tagname = tag
                    post_tags.append(t)
                    current_user.following_tags.append(t)
        topic.tags = post_tags
        topic.board_id = form.category.data
        db.session.add(topic)
        db.session.commit()
        current_user.following_topics.append(topic)
        topic.board.count.topics += 1
        topic.board.count.all_topics += 1
        db.session.commit()
        RedisData.set_topics()
        return topic

    def put(form, topicId):
        topic = Topic.query.filter_by(uid=topicId).first_or_404()
        topic.title = form.title.data
        topic.content = form.content.data
        topic.content = form.content.data
        topic.is_markdown = True if form.choice.data == 1 else False
        tags = sp(',|;|，|；| ', form.tags.data)
        tags = [x for x in list(set(tags)) if x != ''][:4]
        post_tags = []
        for tag in tags:
            if tag != '':
                exsit_tag = Tags.query.filter_by(tagname=tag).first()
                if exsit_tag is not None:
                    post_tags.append(exsit_tag)
                    if exsit_tag not in current_user.following_tags:
                        current_user.following_tags.append(exsit_tag)
                else:
                    t = Tags()
                    t.tagname = tag
                    post_tags.append(t)
                    current_user.following_tags.append(t)
        topic.tags = post_tags
        topic.board_id = form.category.data
        db.session.commit()
        return topic


class ReplyModel(object):
    def post_data(form, uid):
        reply = Reply()
        reply.content = form.content.data
        reply.author = current_user
        reply.topic_id = uid
        db.session.add(reply)
        db.session.commit()
        topic = reply.topic
        topic.board.count.all_topics += 1
        if topic.author_id != current_user.id:
            notice_reply(topic, reply)
        db.session.commit()
        RedisData.set_replies(uid)
        return reply