HBlog/view/oss.py

from flask import Flask, Blueprint, redirect, render_template, request, abort, flash, url_for
from flask_login import login_required, current_user
from typing import Optional
from flask_wtf import FlaskForm
from wtforms import FileField, SubmitField
from wtforms.validators import DataRequired

from aliyun import aliyun
from view.base import App

oss = Blueprint("oss", __name__)
app: Optional[Flask] = None


class UploadForm(FlaskForm):
    file = FileField("选择文件", validators=[DataRequired()])
    submit = SubmitField("上传")


@oss.before_request
def check_aliyun():
    if aliyun is None:
        abort(404)
        return


@oss.route('get/<string:name>')
def get_page(name: str):
    return redirect(aliyun.shared_obj(name))


@oss.route('upload', methods=["GET", "POST"])
@login_required
def upload_page():
    if not current_user.check_role("ConfigureSystem"):
        abort(403)
        return

    form = UploadForm()
    if form.validate_on_submit():
        file = request.files["file"]
        aliyun.upload_file(file.filename, file)
        flash(f"文件 {file.filename} 已上传: {url_for('oss.get_page', name=file.filename, _external=True)}")
        return redirect(url_for("oss.upload_page"))
    return render_template("oss/upload.html", UploadForm=form)


class OSSApp(App):
    def __init__(self, import_name):
        super(OSSApp, self).__init__(import_name)

        global app
        app = self._app
        app.register_blueprint(oss, url_prefix="/oss")