#!/usr/bin/env python
# -*- coding: utf-8 -*-
# **************************************************************************
# Copyright © 2016 jianglin
# File Name: views.py
# Author: jianglin
# Email: xiyang0807@gmail.com
# Created: 2016-12-15 22:07:39 (CST)
# Last Update:星期六 2017-3-25 22:47:59 (CST)
# By:
# Description:
# **************************************************************************
from flask import redirect, render_template, request, url_for
from flask_babelex import gettext as _
from flask_login import current_user
from flask_maple.auth.forms import form_validate
from flask_maple.response import HTTPResponse
from forums.api.forums.models import Board
from forums.api.tag.models import Tags
from forums.common.serializer import Serializer
from forums.common.views import BaseMethodView as MethodView
from forums.common.utils import (gen_filter_dict, gen_order_by)
from .forms import (CollectForm, ReplyForm, TopicForm, collect_error_callback,
error_callback, form_board)
from .models import Collect, Topic, Reply
class TopicAskView(MethodView):
def get(self):
boardId = request.args.get('boardId', type=int)
form = form_board()
if boardId is not None:
form.category.data = boardId
data = {'title': _('Ask - '), 'form': form}
return render_template('topic/ask.html', **data)
class TopicEditView(MethodView):
def get(self, topicId):
topic = Topic.query.filter_by(id=topicId).first_or_404()
form = form_board()
form.title.data = topic.title
form.category.data = topic.board_id
form.tags.data = ','.join([tag.name for tag in topic.tags])
form.content.data = topic.content
data = {'title': _('Edit -'), 'form': form, 'topic': topic}
return render_template('topic/edit.html', **data)
class TopicPreviewView(MethodView):
def post(self):
choice = request.values.get('choice')
content = request.values.get('content')
return
class TopicListView(MethodView):
def get(self):
query_dict = request.data
page, number = self.page_info
keys = ['title']
order_by = gen_order_by(query_dict, keys)
filter_dict = gen_filter_dict(query_dict, keys)
if request.path.endswith('good'):
filter_dict.update(is_good=True)
elif request.path.endswith('top'):
filter_dict.update(is_top=True)
topics = Topic.query.filter_by(
**filter_dict).order_by(*order_by).paginate(page, number, True)
return render_template('topic/topic_list.html', topics=topics)
@form_validate(form_board, error=error_callback, f='')
def post(self):
form = form_board()
post_data = form.data
title = post_data.pop('title', None)
content = post_data.pop('content', None)
tags = post_data.pop('tags', None)
content_type = post_data.pop('content_type', None)
board = post_data.pop('category', None)
topic = Topic(
title=title,
content=content,
content_type=content_type,
board_id=int(board))
tags = tags.split(',')
topic_tags = []
for tag in tags:
topic_tag = Tags.query.filter_by(name=tag).first()
if topic_tag is None:
topic_tag = Tags()
topic_tag.name = tag
topic_tag.description = tag
topic_tag.save()
topic_tags.append(topic_tag)
topic.tags = topic_tags
topic.author = current_user
topic.save()
return redirect(url_for('topic.topic', topicId=topic.id))
class TopicView(MethodView):
def get(self, topicId):
form = ReplyForm()
topic = Topic.query.filter_by(id=topicId).first_or_404()
data = {'title': topic.title, 'form': form, 'topic': topic}
return render_template('topic/topic.html', **data)
def put(self, topicId):
post_data = request.data
topic = Topic.query.filter_by(id=topicId).first()
title = post_data.pop('title', None)
content = post_data.pop('content', None)
content_type = post_data.pop('content_type', None)
board = post_data.pop('board', None)
if title is not None:
topic.title = title
if content is not None:
topic.content = content
if content_type is not None:
topic.content_type = content_type
if board is not None:
topic.board = int(board)
topic.save()
return redirect(url_for('topic.topic', topicId=topic.id))
# def delete(self, topicId):
# topic = Topic.query.filter_by(id=topicId).first()
# topic.delete()
# serializer = Serializer(topic, many=False)
# return HTTPResponse(HTTPResponse.NORMAL_STATUS,
# **serializer.data).to_response()
class CollectListView(MethodView):
def get(self):
form = CollectForm()
page, number = self.page_info
collects = Collect.get_list(page, number)
data = {'collects': collects, 'form': form}
return render_template('collect/collect_list.html', **data)
@form_validate(CollectForm, error=collect_error_callback, f='')
def post(self):
form = CollectForm()
post_data = form.data
name = post_data.pop('name', None)
description = post_data.pop('description', None)
privacy = post_data.pop('private', None)
privacy = True if privacy == '0' else False
collect = Collect(name=name, description=description, privacy=privacy)
collect.author = current_user
collect.save()
return collect_error_callback()
class CollectView(MethodView):
def get(self, collectId):
form = CollectForm()
collect = Collect.get(id=collectId)
topics = collect.topics.paginate(1, 23, True)
data = {'collect': collect, 'topics': topics, 'form': form}
return render_template('collect/collect.html', **data)
def put(self, collectId):
post_data = request.data
collect = Collect.query.filter_by(id=collectId).first()
name = post_data.pop('name', None)
description = post_data.pop('description', None)
privacy = post_data.pop('privacy', None)
if name is not None:
collect.name = name
if description is not None:
collect.description = description
if privacy is not None:
collect.privacy = privacy
collect.save()
serializer = Serializer(collect, many=False)
return HTTPResponse(HTTPResponse.NORMAL_STATUS,
**serializer.data).to_response()
def delete(self, collectId):
collect = Collect.query.filter_by(id=collectId).first()
collect.delete()
serializer = Serializer(collect, many=False)
return HTTPResponse(HTTPResponse.NORMAL_STATUS,
**serializer.data).to_response()
class ReplyListView(MethodView):
@form_validate(ReplyForm, error=error_callback, f='')
def post(self, topicId):
topic = Topic.query.filter_by(id=topicId).first_or_404()
post_data = request.data
content = post_data.pop('content', None)
reply = Reply(content=content, topic_id=topic.id)
reply.author = current_user
reply.save()
return redirect(url_for('topic.topic', topicId=topic.id))
class ReplyView(MethodView):
def put(self, replyId):
post_data = request.data
reply = Reply.query.filter_by(id=replyId).first()
content = post_data.pop('content', None)
if content is not None:
reply.content = content
reply.save()
serializer = Serializer(reply, many=False)
return HTTPResponse(HTTPResponse.NORMAL_STATUS,
**serializer.data).to_response()
def delete(self, replyId):
reply = Reply.query.filter_by(id=replyId).first()
reply.delete()
serializer = Serializer(reply, many=False)
return HTTPResponse(HTTPResponse.NORMAL_STATUS,
**serializer.data).to_response()
class LikeView(MethodView):
def post(self, replyId):
user = request.user
reply = Reply.query.filter_by(id=replyId).first_or_404()
reply.likers.append(user)
reply.save()
serializer = Serializer(reply, many=False)
return HTTPResponse(
HTTPResponse.NORMAL_STATUS, data=serializer.data).to_response()
def delete(self, replyId):
user = request.user
reply = Reply.query.filter_by(id=replyId).first_or_404()
reply.likers.remove(user)
reply.save()
serializer = Serializer(reply, many=False)
return HTTPResponse(
HTTPResponse.NORMAL_STATUS, data=serializer.data).to_response()