small blog views

import os
from functools import wraps
from datetime import datetime
from models import app, db, User, Article
from forms import LoginForm, RegisterForm, ArticleAddForm, ArticleEditForm
from flask import session, redirect, url_for, request, \
    flash, render_template, make_response
from uuid import uuid4
from werkzeug.utils import secure_filename


# settings
def user_login_request(f):
    """
    login wrapper
    """

    @wraps(f)
    def login_request(*args, **kwargs):
        if 'user' not in session:
            return redirect(url_for('login', next_url=request.url))
        return f(*args, **kwargs)

    return login_request


def change_name(name):
    info = os.path.splitext(name)
    name_in = datetime.now().strftime('%Y%m%d%H%M%S') + str(uuid4().hex) + info[-1]
    return name_in


@app.route('/')
def jump():
    return redirect(url_for('login'))


@app.route('/login/', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        data = form.data
        session['user'] = data['account']
        flash('success login', 'ok')
        return redirect(url_for('art_list', page=1))
    return render_template('login.html', title='user login', form=form)


@user_login_request
@app.route('/logout/', methods=['GET', 'POST'])
def logout():
    session.pop('user', None)
    return redirect(url_for('login'))


@app.route('/register/', methods=['GET', 'POST'])
def register():
    form = RegisterForm()
    if form.validate_on_submit():
        data = form.data
        user = User(
            username=data['account'],
            password=data['re_pwd'],
        )
        db.session.add(user)
        db.session.commit()
        flash('success register', 'ok')
        return redirect(url_for('login'))
    return render_template('register.html', title='register', form=form)


@user_login_request
@app.route('/article/add/', methods=['GET', 'POST'])
def article_add():
    # print('article add')
    form = ArticleAddForm()

    if form.validate_on_submit():
        data = form.data
        # print(data)

        file_name = secure_filename(form.logo.data.filename)
        logo_filename = change_name(file_name)
        if not os.path.exists(app.config['uploads']):
            os.makedirs(app.config['uploads'])
        form.logo.data.save(os.path.join(app.config['uploads'], logo_filename))
        user = User.query.filter_by(username=session['user']).first()
        user_id = user.id
        article = Article(
            title=data['title'],
            user_id=user_id,
            logo=logo_filename,
            content=data['content']
        )
        db.session.add(article)
        db.session.commit()
        flash('article publish success', 'ok')
        return redirect(url_for('art_list', page=1))
    return render_template('article_add.html', title='publish article', form=form)


@user_login_request
@app.route("/article/edit/<int:pk>/", methods=["GET", "POST"])  # 编辑文章
def art_edit(pk):
    form = ArticleEditForm()
    article = Article.query.get_or_404(ident=pk)
    if request.method == 'GET':
        form.content.data = article.content

    if form.validate_on_submit():
        if form.logo.data:
            print(form.logo.data)
            get_name = secure_filename(form.logo.data.filename)
            filename = change_name(get_name)
            if not os.path.exists(app.config['uploads']):
                os.makedirs(app.config['uploads'])
            form.logo.data.save(os.path.join(app.config['uploads'], filename))
            article.logo = filename
        if form.content.data != article.content:
            article.content = form.content.data

        db.session.add(article)
        db.session.commit()
        return redirect(url_for('art_list', page=1))

    return render_template("article_edit.html", form=form, title="编辑文章", article=article)


@user_login_request
@app.route("/article/list/<int:page>/", methods=["GET"])  # 文章列表
def art_list(page):
    if not page:
        page = 1
    # 只展示当前用户才能看到的内容
    user = User.query.filter_by(username=session["user"]).first()
    user_id = user.id
    page_data = Article.query.filter_by(user_id=user_id) \
        .order_by(Article.add_time.desc()).paginate(page=page, per_page=5)
    # print(page_data)
    return render_template("article_list.html", title="文章列表", page_data=page_data)


@user_login_request
@app.route("/article/del/<int:pk>/", methods=["GET"])  # 删除文章
def art_del(pk):
    article = Article.query.get_or_404(ident=pk)
    db.session.delete(article)
    db.session.commit()
    flash("删除 %s 成功!" % article.title, "ok")
    return redirect(url_for('art_list', page=1))


@app.route("/captcha/", methods=["GET"])
def captcha():
    from app.generate_img import generate_buf_picture
    img_bin, text = generate_buf_picture()

    session['captcha'] = text
    response = make_response(img_bin)
    response.headers['Content-Type'] = 'image/jpeg'
    return response


if __name__ == '__main__':
    app.run(debug=True)