import os

from flask import request, jsonify, current_app
from flask_restful import Api, Resource, marshal
from sqlalchemy import desc

from applications.api import api_bp
from applications.common.serialization import photo_fields
from applications.common.utils.http import fail_api, success_api, table_api
from applications.common.utils.rights import authorize
from applications.common.utils.upload import upload_one, delete_photo_by_id
from applications.extensions import db
from applications.models import Photo

file_api = Api(api_bp, prefix='/file')


@file_api.resource('/photos')
class FilePhotos(Resource):

    @authorize("admin:file:main", log=True)
    def get(self):
        page = request.args.get('page', type=int)
        limit = request.args.get('limit', type=int)
        photo_paginate = Photo.query.order_by(desc(Photo.create_time)).paginate(page=page, per_page=limit,
                                                                                error_out=False)
        data = marshal(photo_paginate.items, photo_fields)
        return table_api(data=data, count=photo_paginate.total, code=0)

    @authorize("admin:file:add", log=True)
    def post(self):
        if '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 fail_api()

    @authorize("admin:file:delete", log=True)
    def delete(self):
        """图片批量删除"""
        # TODO bugs 图片删除失败
        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 success_api(msg="删除成功")
        else:
            return fail_api(msg="删除失败")


@file_api.resource('/photo/<int:photo_id>')
class FileTable(Resource):
    """图片数据"""

    @authorize("admin:file:delete", log=True)
    def delete(self, photo_id):
        res = delete_photo_by_id(photo_id)
        if res:
            return success_api(msg="删除成功")
        else:
            return fail_api(msg="删除失败")