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 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,
            "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