maple-bbs/maple/main/orderby.py

#!/usr/bin/env python
# -*- coding=UTF-8 -*-
# **************************************************************************
# Copyright © 2016 jianglin
# File Name: orderby.py
# Author: jianglin
# Email: xiyang0807@gmail.com
# Created: 2016-06-05 10:15:58 (CST)
# Last Update:星期二 2016-6-14 23:20:14 (CST)
#          By:
# Description:
# **************************************************************************
from flask import request
from maple.forums.models import Board
from maple.topic.models import Tags, Topic
from datetime import datetime, timedelta


def form_judge(form):
    '''
    0: all topic
    1: one day
    2: one week
    3: one month

    0: time
    1: author

    0: desc
    1: asc
    '''
    t1 = form.display.data
    t2 = form.sort.data
    t3 = form.st.data
    data = form_sort(t1, t2, t3)
    return data


def form_sort(t1, t2, t3):
    orderby = request.get_json()
    type = orderby['type']
    uid = orderby['uid']
    page = int(orderby['page'])

    if t1 == 0:
        time = datetime.now() - timedelta(days=365)
    elif t1 == 1:
        time = datetime.now() - timedelta(days=1)
    elif t1 == 2:
        time = datetime.now() - timedelta(days=7)
    else:
        time = datetime.now() - timedelta(days=30)

    # 发表时间
    if t2 == 0:
        if t3 == 0:
            if type == 'parent_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time, Board.parent_board ==
                    uid).order_by(Topic.publish.desc()).paginate(page, 20,
                                                                 True)
            elif type == 'child_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time,
                    Board.id == uid).order_by(Topic.publish.desc()).paginate(
                        page, 20, True)
            elif type == 'tags':
                topics = Topic.query.join(Topic.tags).filter(
                    Topic.publish > time, Tags.tagname ==
                    uid).order_by(Topic.publish.desc()).paginate(page, 20,
                                                                 True)
            else:
                topics = None
            return topics
        else:
            if type == 'parent_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time, Board.parent_board ==
                    uid).order_by(Topic.publish.asc()).paginate(page, 20, True)
            elif type == 'child_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time,
                    Board.id == uid).order_by(Topic.publish.asc()).paginate(
                        page, 20, True)
            elif type == 'tags':
                topics = Topic.query.join(Topic.tags).filter(
                    Topic.publish > time, Tags.tagname ==
                    uid).order_by(Topic.publish.asc()).paginate(page, 20, True)
            else:
                topics = None
            return topics
    # 作者
    else:
        if t3 == 0:
            if type == 'parent_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time, Board.parent_board ==
                    uid).order_by(Topic.author_id.desc()).paginate(page, 20,
                                                                   True)
            elif type == 'child_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time,
                    Board.id == uid).order_by(Topic.author_id.desc()).paginate(
                        page, 20, True)
            elif type == 'tags':
                topics = Topic.query.join(Topic.tags).filter(
                    Topic.publish > time, Tags.tagname ==
                    uid).order_by(Topic.author_id.desc()).paginate(page, 20,
                                                                   True)
            else:
                topics = None
            return topics
        else:
            if type == 'parent_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time, Board.parent_board ==
                    uid).order_by(Topic.author_id.asc()).paginate(page, 20,
                                                                  True)
            elif type == 'child_b':
                topics = Topic.query.join(Topic.board).filter(
                    Topic.publish > time,
                    Board.id == uid).order_by(Topic.author_id.asc()).paginate(
                        page, 20, True)
            elif type == 'tags':
                topics = Topic.query.join(Topic.tags).filter(
                    Topic.publish > time, Tags.tagname ==
                    uid).order_by(Topic.author_id.asc()).paginate(page, 20,
                                                                  True)
            else:
                topics = None
            return topics