Files
pear-admin-flask/applications/views/admin/file.py
T
2021-04-27 11:58:21 +08:00

85 lines
2.4 KiB
Python

import os
from flask import Blueprint, request, render_template, jsonify, current_app
from flask_login import login_required
from applications.models import db
from applications.models.admin_photo import Photo
from applications.service.admin.file import get_photo, upload_one, delete_photo_by_id
from applications.service.route_auth import authorize_and_log
admin_file = Blueprint('adminFile', __name__, url_prefix='/admin/file')
# 图片管理
@admin_file.route('/')
@authorize_and_log("admin:file:main")
def index():
return render_template('admin/photo/photo.html')
# 图片数据
@admin_file.route('/table')
@authorize_and_log("admin:file:main")
def table():
page = request.args.get('page', type=int)
limit = request.args.get('limit', type=int)
data, count = get_photo(page=page, limit=limit)
res = {
'msg': "",
'code': 0,
'data': data,
'count': count,
'limit': "10"
}
return jsonify(res)
# 上传接口
@admin_file.route('/upload', methods=['GET', 'POST'])
@authorize_and_log("admin:file:add")
def upload():
if request.method == 'POST' and 'file' in request.files:
photo = request.files['file']
mime = request.files['file'].content_type
file_url = upload_one(photo=photo, mime=mime)
res = {
"msg": "上传成功",
"code": 0,
"success":True,
"data":
{"src": file_url}
}
return jsonify(res)
return render_template('admin/photo/photo_add.html')
# 图片删除
@admin_file.route('/delete', methods=['GET', 'POST'])
@authorize_and_log("admin:file:delete")
def delete():
id = request.form.get('id')
res = delete_photo_by_id(id)
if res:
return jsonify(msg="删除成功", code=200)
else:
return jsonify(msg="删除失败", code=999)
# 图片批量删除
@admin_file.route('/batchRemove', methods=['GET', 'POST'])
@authorize_and_log("admin:file:delete")
def batchRemove():
ids = request.form.getlist('ids[]')
photo_name = Photo.query.filter(Photo.id.in_(ids)).all()
upload_url = current_app.config.get("UPLOADED_PHOTOS_DEST")
for p in photo_name:
os.remove(upload_url + '/' + p.name)
photo = Photo.query.filter(Photo.id.in_(ids)).delete(synchronize_session=False)
db.session.commit()
if photo:
return jsonify(msg="删除成功", code=200)
else:
res = {"msg": "删除失败", "code": 999}
return res