发布帖子

from exts import db
from datetime import datetime

# 板块模型
class BoardModel(db.Model):
    __tablename__ = 'board'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(20), nullable=False)
    create_time = db.Column(db.DateTime, default=datetime.now)
    is_active = db.Column(db.Boolean, default=True)

# 帖子模型
class PostModel(db.Model):
    __tablename__ = 'post'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    title = db.Column(db.String(200), nullable=False)
    content = db.Column(db.Text, nullable=False)
    create_time = db.Column(db.DateTime, default=datetime.now)
    read_count = db.Column(db.Integer, default=0)
    is_active = db.Column(db.Boolean, default=True)
    board_id = db.Column(db.Integer, db.ForeignKey("board.id"))
    author_id = db.Column(db.String(100), db.ForeignKey("user.id"), nullable=False)

    board = db.relationship("BoardModel", backref="posts")
    author = db.relationship("UserModel", backref='posts')

# 评论模型
class CommentModel(db.Model):
    __tablename__ = 'comment'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    content = db.Column(db.Text, nullable=False)
    create_time = db.Column(db.DateTime, default=datetime.now)
    is_active = db.Column(db.Boolean, default=True)
    post_id = db.Column(db.Integer, db.ForeignKey("post.id"))
    author_id = db.Column(db.String(100), db.ForeignKey("user.id"), nullable=False)

    post = db.relationship("PostModel", backref='comments')
    author = db.relationship("UserModel", backref='comments')

$ flask db migrate -m "add post comment board model"
$ flask db upgrade

def create_board():
    board_names = ['Python语法', 'web开发', '数据分析', '测试开发', '运维开发']
    for board_name in board_names:
        board = BoardModel(name=board_name)
        db.session.add(board)
    db.session.commit()
    click.echo("板块添加成功！")

...
app.cli.command("create-board")(commands.create_board)

@bp.route("/post/public", methods=['GET', 'POST'])
def public_post():
    if request.method == 'GET':
        boards = BoardModel.query.all()
        return render_template("front/public_post.html", boards=boards)
    else:
        pass

...
<div class="form-group">
    <label>板块</label>
    <select name="board_id" class="form-control">
        {% for board in boards %}
            <option value="{{ board.id }}">{{ board.name }}</option>
        {% endfor %}
    </select>
</div>
...

{
    "errno": 0,
    "data": [
        {
            "url": "url",
            "alt": "",
            "href": ""
        }
    ]
}

from flask import Blueprint, current_app
import os

bp = Blueprint("media", __name__, url_prefix="/media")

@bp.get("/<path:filename>")
def media_file(filename):
    return os.path.join(current_app.config.get("UPLOAD_IMAGE_PATH"), filename)

from blueprints.media import bp as media_bp
...
app.register_blueprint(media_bp)
...

var editor = new window.wangEditor("#editor");
editor.config.uploadImgServer = "/upload/image";
editor.config.uploadFileName = "image";
editor.create();

from flask_wtf import CSRFProtect
...
csrf = CSRFProtect()

...
from exts import csrf
...
csrf.init_app(app)

@bp.post("/upload/image")
@csrf.exempt
def upload_image():
...

from flask import session, g
from models.user import UserModel

def bbs_before_request():
    if "user_id" in session:
        user_id = session.get("user_id")
        try:
            user = UserModel.query.get(user_id)
            setattr(g, "user", user)
        except Exception:
            pass

import hooks
...
# 添加钩子函数
app.before_request(hooks.bbs_before_request)

from functools import wraps
from flask import redirect, url_for, g

def login_required(func):
    @wraps(func)
    def inner(*args, **kwargs):
        if not hasattr(g, "user"):
            return redirect(url_for("user.login"))
        else:
            return func(*args, **kwargs)
    return inner

from decorators import login_required
...
@bp.route("/post/public", methods=['GET', 'POST'])
@login_required
def public_post():
...

@bp.post("/upload/image")
@csrf.exempt
@login_required
def upload_image():
...

from .baseform import BaseForm
from wtforms import StringField, IntegerField
from wtforms.validators import InputRequired, Length

class PublicPostForm(BaseForm):
    title = StringField(validators=[Length(min=2, max=100, message='请输入正确长度的标题！')])
    content = StringField(validators=[Length(min=2, message="请输入正确长度的内容！")])
    board_id = IntegerField(validators=[InputRequired(message='请输入板块 id！')])

@bp.route("/post/public", methods=['GET', 'POST'])
@login_required
def public_post():
    if request.method == 'GET':
        boards = BoardModel.query.all()
        return render_template("front/public_post.html", boards=boards)
    else:
        form = PublicPostForm(request.form)
        if form.validate():
            title = form.title.data
            content = form.content.data
            board_id = form.board_id.data
            post = PostModel(title=title, content=content, board_id=board_id, author=g.user)
            db.session.add(post)
            db.session.commit()
            return restful.ok()
        else:
            message = form.messages[0]
            return restful.params_error(message=message)

$(function () {
    ...
    // 提交按钮单击事件
    $("#submit-btn").click(function (event) {
        event.preventDefault();

        var title = $("input[name='title']").val();
        var board_id = $("select[name='board_id']").val();
        var content = editor.txt.html();

        zlajax.post({
            url: "/post/public",
            data: { title, board_id, content }
        }).done(function (data) {
            setTimeout(function () {
                window.location = "/";
            }, 2000);
        }).fail(function (error) {
            alert(error.message);
        });
    });
});