图片上传

import os
import random

import cv2
from flask import Flask, request, send_from_directory, render_template, jsonify, make_response
from flask_uploads import UploadSet, configure_uploads, IMAGES, \
    patch_request_class
from output import out_path, web_path, file_name, direct_path
from program import task

app = Flask(__name__)
app.config['UPLOADED_PHOTOS_DEST'] = os.getcwd() + '\\static\\uploads'  # 文件储存地址

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)
patch_request_class(app)  # 文件大小限制，默认为16MB

# 存储的最开始的图片的名字 后面上传就会改变
photo_name = 'Tsunami_by_hokusai_19th_century.jpg'
app.config['PHOTO_URL'] = os.getcwd() + '\\static\\uploads\\' + 'Tsunami_by_hokusai_19th_century.jpg'
# 现在的图片的版本
app.config['cnt'] = 0

html = '''
    <!DOCTYPE html>
    <title>Upload File</title>
    <h1>图片上传</h1>
    <form method=post enctype=multipart/form-data>
         <input type=file name=photo>
         <input type=submit value=上传>
    </form>
    '''

pure_html = '''
	<!DOCTYPE html>
	<title>Processed Image</title>
	'''


# 主页面
@app.route('/upload_file', methods=['GET', 'POST'])
def upload_file():
    cnt = 0
    del_all()
    if request.method == 'POST' and 'photo' in request.files:
        filename = photos.save(request.files['photo'])
        # app.name = filename
        app.photo_name = filename
        file_url = photos.url(filename)
        print(file_url)
        app.config['PHOTO_URL'] = os.getcwd() + '\\static\\uploads\\' + filename
        print(app.config['PHOTO_URL'] + 'photourl')
        img = cv2.imread(app.config['PHOTO_URL'])  # 根据路径读取一张图片
        cv2.imwrite(out_path(0), img)
        print(file_url)
        res = {
            'code': 0,
            'url_before': direct_path(app.config['cnt']) + '?' + str(random.random()),
            'url_after': ''
        }
        print(res)
        return jsonify(res)
    res = {
        'code': -1,
        'url_before': '',
        'url_after': ''
    }
    print(res)
    return jsonify(res)


@app.route('/upload_style', methods=['GET', 'POST'])
def upload_style():
    cnt = 0
    del_all()
    if request.method == 'POST' and 'photo' in request.files:
        filename = photos.save(request.files['photo'])
        # app.name = filename
        app.photo_name = filename
        file_url = photos.url(filename)
        print(file_url)
        app.config['PHOTO_URL'] = os.getcwd() + '\\static\\uploads\\' + filename
        print(app.config['PHOTO_URL'] + 'photourl')
        img = cv2.imread(app.config['PHOTO_URL'])  # 根据路径读取一张图片
        cv2.imwrite(out_path(0), img)
        print(file_url)
        cv2.imwrite('./img_src/styles/style0.jpg', img)
        res = {
            'code': 0,
        }
        print(res)
        return jsonify(res)
    res = {
        'code': -1,
    }
    print(res)
    return jsonify(res)


# 删除uploads文件夹       （不用设置接口）
@app.route('/delete_all')
def del_all():
    ls = os.listdir(app.config['UPLOADED_PHOTOS_DEST'])
    for i in ls:
        c_path = os.path.join(app.config['UPLOADED_PHOTOS_DEST'], i)
        if os.path.isdir(c_path):
            del_file(c_path)
        else:
            os.remove(c_path)
    app.config['cnt'] = 0


# 用在delete_all里面，不用设置接口    （不用设置接口）
@app.route('/delete_file/<path>')
def del_file(path):
    if os.path.exists(path):
        os.remove(path)


# 通过commmand_id来选择处理的方式，commmand_id具体内容再program.py里面找
@app.route('/deal/<command_id>', methods=['GET', 'POST'])
def deal(command_id):
    img = cv2.imread(app.config['PHOTO_URL'])  # 根据路径读取一张图片

    res = {
        'code': -1,
        'url_before': '',
        'url_after': ''
    }

    cmd = command_id
    if (cmd == 'b' or cmd == 'back'):
        # 撤销操作
        if (app.config['cnt'] > 1):
            if os.path.exists(out_path(app.config['cnt'])):
                os.remove(out_path(app.config['cnt']))
            print('back success')
            print(app.config['cnt'])
            app.config['cnt'] = app.config['cnt'] - 1
            res = {
                'code': 0,
                'url_before': direct_path(app.config['cnt'] - 1) + '?' + str(random.random()),
                'url_after': direct_path(app.config['cnt']) + '?' + str(random.random())
            }
        elif (app.config['cnt'] == 1):
            res = {
                'code': 0,
                'url_before': direct_path(app.config['cnt'] - 1) + '?' + str(random.random()),
                'url_after': ''
            }
            app.config['cnt'] = app.config['cnt'] - 1
        elif (app.config['cnt'] == 0):
            res = {
                'code': 1,
                'url_before': direct_path(app.config['cnt'] - 1) + '?' + str(random.random()),
                'url_after': ''
            }

    elif (cmd == 's' or cmd == 'save'):
        # 保存操作
        cnt_max = app.config['cnt']
        img = cv2.imread(out_path(app.config['cnt']))
        cv2.imwrite(os.getcwd() + '\\static', img)
        cnt = 0
        cv2.imwrite(out_path(cnt), img)
        for id in range(1, cnt_max + 1):
            if os.path.exists(out_path(id)):
                os.remove(out_path(id))
        print('save success')

    else:
        # 图像处理操作
        success = task(cmd, app.config['cnt'])
        if (success == 0):
            app.config['cnt'] = app.config['cnt'] + 1
            res = {
                'code': 0,
                'url_before': direct_path(app.config['cnt'] - 1) + '?' + str(random.random()),
                'url_after': direct_path(app.config['cnt']) + '?' + str(random.random()),
            }
        str_cnt = str(app.config['cnt'])
        print(type(str_cnt))
        print(app.config['cnt'])
    return jsonify(res)


# 这个download会下载最新的照片
@app.route('/download', methods=['POST', 'GET'])
def download():
    direct_path = os.getcwd() + '\\static\\uploads'
    # return send_from_directory(direct_path, file_name(app.config['cnt']), as_attachment=True)
    response = send_from_directory(direct_path, file_name(app.config['cnt']), as_attachment=True)
    response.headers["Access-Control-Expose-Headers"] = "Content-disposition"
    return response


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