HBlog/app/docx.py

from flask import Blueprint, render_template, abort, redirect, url_for, flash, make_response, g, request
from flask_wtf import FlaskForm
from flask_login import login_required, current_user
from wtforms import TextAreaField, StringField, SelectMultipleField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length

import app
from sql.base import DBBit
from object.blog import BlogArticle, load_blog_by_id
from object.comment import Comment
from object.archive import load_archive_by_id, Archive

docx = Blueprint("docx", __name__)


class WriteBlogForm(FlaskForm):
    title = StringField("标题", description="博文主标题",
                        validators=[
                            DataRequired(message="必须填写标题"),
                            Length(1, 20, message="标题长度1-20个字符")])
    subtitle = StringField("副标题", description="博文副标题",
                           validators=[Length(-1, 20, message="副标题长度20个字符以内")])
    archive = SelectMultipleField("归档", coerce=int)
    context = TextAreaField("博客内容", validators=[DataRequired()])
    submit = SubmitField("提交博客")

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.context.data = "# Blog Title\n## Blog subtitle\nHello, World"
        archive = Archive.get_archive_list()
        self.archive_res = []
        self.archive_choices = [(-1, "None")]
        for i in archive:
            self.archive_res.append(i[0])
            self.archive_choices.append((i[0], f"{i[1]} ({i[3]})"))
        self.archive.choices = self.archive_choices

    def validate_archive(self, field):
        if -1 in field.data:
            if len(field.data) != 1:
                raise ValidationError("归档指定错误(none归档不能和其他归档同时被指定)")
        else:
            for i in field.data:
                if i not in self.archive_res:
                    raise ValidationError("错误的归档被指定")


class WriteCommentForm(FlaskForm):
    context = TextAreaField("", description="评论正文",
                            validators=[DataRequired(message="请输入评论的内容"),
                                        Length(1, 100, message="请输入1-100个字的评论")])
    submit = SubmitField("评论")


def __load_docx_page(page: int, form: WriteBlogForm):
    if page < 1:
        app.HBlogFlask.print_user_opt_fail_log(f"Load docx list with error page({page})")
        abort(404)
        return

    blog_list = BlogArticle.get_blog_list(limit=20, offset=(page - 1) * 20)
    max_page = app.HBlogFlask.get_max_page(BlogArticle.get_blog_count(), 20)
    page_list = app.HBlogFlask.get_page("docx.docx_page", page, max_page)
    app.HBlogFlask.print_load_page_log(f"docx list (page: {page})")
    return render_template("docx/docx.html",
                           page=page,
                           blog_list=blog_list,
                           is_top=DBBit.BIT_1,
                           page_list=page_list,
                           form=form,
                           show_delete=current_user.check_role("DeleteBlog"))


@docx.route('/<int:page>')
def docx_page(page: int = 1):
    return __load_docx_page(page, WriteBlogForm())


@docx.route('/<int:archive>/<int:page>')
def archive_page(archive: int, page: int = 1):
    if page < 1:
        app.HBlogFlask.print_user_opt_fail_log(f"Load archive-docx list with error page({page}) archive: {archive}")
        abort(404)
        return

    blog_list = BlogArticle.get_blog_list(archive_id=archive, limit=20, offset=(page - 1) * 20)
    max_page = app.HBlogFlask.get_max_page(BlogArticle.get_blog_count(archive_id=archive), 20)
    page_list = app.HBlogFlask.get_page("docx.archive_page", page, max_page)
    app.HBlogFlask.print_load_page_log(f"archive-docx list (archive-id: {archive} page: {page})")
    return render_template("docx/docx.html",
                           blog_list=blog_list,
                           is_top=DBBit.BIT_1,
                           page_list=page_list,
                           form=None)


def __load_article_page(blog_id: int, form: WriteCommentForm):
    article = load_blog_by_id(blog_id)
    if article is None:
        app.HBlogFlask.print_user_opt_fail_log(f"Load article with error id({blog_id})")
        abort(404)
        return
    app.HBlogFlask.print_load_page_log(f"article (id: {blog_id})")
    return render_template("docx/article.html",
                           article=article,
                           archive_list=article.archive,
                           form=form,
                           show_delete=current_user.check_role("DeleteComment"),
                           show_email=current_user.check_role("ReadUserInfo"))


@docx.route('/article/<int:blog_id>')
def article_page(blog_id: int):
    return __load_article_page(blog_id, WriteCommentForm())


@docx.route('/down/<int:blog_id>')
def article_down_page(blog_id: int):
    article = load_blog_by_id(blog_id)
    if article is None:
        app.HBlogFlask.print_user_opt_fail_log(f"Download article with error id({blog_id})")
        abort(404)
        return

    response = make_response(article.context)
    response.headers["Content-Disposition"] = f"attachment;filename={article.title.encode().decode('latin-1')}.md"
    app.HBlogFlask.print_load_page_log(f"download article (id: {blog_id})")
    return response


@docx.route('/comment/<int:blog_id>', methods=["POST"])
@login_required
@app.form_required(WriteCommentForm, "write comment", __load_article_page)
@app.role_required("WriteComment", "write comment")
def comment_page(blog_id: int):
    form: WriteCommentForm = g.form
    context = form.context.data
    if Comment(None, blog_id, current_user, context).create():
        app.HBlogFlask.print_user_opt_success_log("comment")
        flash("评论成功")
    else:
        app.HBlogFlask.print_user_opt_error_log("comment")
        flash("评论失败")
    return redirect(url_for("docx.article_page", blog_id=blog_id))


@docx.route('/create-docx', methods=["POST"])
@login_required
@app.form_required(WriteBlogForm, "write blog", lambda form: __load_docx_page(int(request.args.get("page", 1)), form))
@app.role_required("WriteBlog", "write blog")
def create_docx_page():
    form: WriteBlogForm = g.form
    title = form.title.data
    subtitle = form.subtitle.data
    archive = []
    if -1 not in form.archive.data:
        for i in form.archive.data:
            i = load_archive_by_id(i)
            if i is not None:
                archive.append(i)

    if BlogArticle(None, current_user, title, subtitle, form.context.data, archive=archive).create():
        app.HBlogFlask.print_sys_opt_success_log("write blog")
        flash(f"博客 {title} 发表成功")
    else:
        app.HBlogFlask.print_sys_opt_fail_log("write blog")
        flash(f"博客 {title} 发表失败")
    return redirect(url_for("docx.docx_page", page=1))


@docx.route("delete/<int:blog_id>")
@login_required
@app.role_required("DeleteBlog", "delete blog")
def delete_blog_page(blog_id: int):
    if BlogArticle(blog_id, None, None, None, None).delete():
        app.HBlogFlask.print_sys_opt_success_log("delete blog")
        flash("博文删除成功")
    else:
        app.HBlogFlask.print_sys_opt_fail_log("delete blog")
        flash("博文删除失败")
    return redirect(url_for("docx.docx_page", page=1))


@docx.route("delete_comment/<int:comment_id>")
@login_required
@app.role_required("DeleteComment", "delete comment")
def delete_comment_page(comment_id: int):
    if Comment(comment_id, None, None, None).delete():
        app.HBlogFlask.print_sys_opt_success_log("delete comment")
        flash("博文评论成功")
    else:
        app.HBlogFlask.print_sys_opt_fail_log("delete comment")
        flash("博文评论失败")
    return redirect(url_for("docx.docx_page", page=1))


@docx.context_processor
def inject_base():
    return {"top_nav": ["", "", "active", "", "", ""]}