使用python flask框架開發圖片上傳介面的案例詳解

2022-04-14 10:01:47

python版本：3.6+

需要模組：flask，pillow

需求：開發一個支援多格式圖片上傳的介面，並且將圖片壓縮，支援線上預覽圖片。

目錄結構：

app.py編輯內容：

from flask import Flask, request, Response, render_template
from werkzeug.utils import secure_filename
import os
import uuid
from PIL import Image, ExifTags
 
app = Flask(__name__)  # 範例Flask應用
# 設定允許上傳的檔案格式
ALLOW_EXTENSIONS = ['png', 'jpg', 'jpeg']
# 設定圖片儲存資料夾
app.config['UPLOAD_FOLDER'] = './static/image/'
# 設定圖片返回的域名字首
image_url = "http://127.0.0.1:8002/image/"
# 設定圖片壓縮尺寸
image_c = 1000
# 跨域支援
def after_request(resp):
    resp.headers['Access-Control-Allow-Origin'] = '*'
    return resp
app.after_request(after_request)
# 判斷檔案字尾是否在列表中
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[-1] in ALLOW_EXTENSIONS
# 首頁
@app.route('/')
def hello_world():
    return render_template('index.html')
# 心跳檢測
@app.route("/check", methods=["GET"])
def check():
    return 'Im live'
# 圖片獲取地址 用於存放靜態檔案
@app.route("/image/<imageId>")
def get_frame(imageId):
    # 圖片上傳儲存的路徑
    try:
        with open(r'./static/image/{}'.format(imageId), 'rb') as f:
            image = f.read()
            result = Response(image, mimetype="image/jpg")
            return result
    except BaseException as e:
        return {"code": '503', "data": str(e), "message": "圖片不存在"}
# 上傳圖片
@app.route("/upload_image", methods=['POST', "GET"])
def uploads():
    if request.method == 'POST':
        # 獲取檔案
        file = request.files['file']
        # 檢測檔案格式
        if file and allowed_file(file.filename):
            # secure_filename方法會去掉檔名中的中文，獲取檔案的字尾名
            file_name_hz = secure_filename(file.filename).split('.')[-1]
            # 使用uuid生成唯一圖片名
            first_name = str(uuid.uuid4())
            # 將 uuid和字尾拼接為 完整的檔名
            file_name = first_name + '.' + file_name_hz
            # 儲存原圖
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], file_name))
            # 以下是壓縮圖片的過程，在原圖的基礎上
            file_min = Image.open(file)
            # exif讀取原始方位資訊 防止圖片壓縮後發生旋轉
            try:
                for orientation in ExifTags.TAGS.keys():
                    if ExifTags.TAGS[orientation] == 'Orientation': break
                exif = dict(file_min._getexif().items())
                if exif[orientation] == 3:
                    file_min = file_min.rotate(180, expand=True)
                elif exif[orientation] == 6:
                    file_min = file_min.rotate(270, expand=True)
                elif exif[orientation] == 8:
                    file_min = file_min.rotate(90, expand=True)
            except:
                pass
            # 獲取原圖尺寸
            w, h = file_min.size
            # 計算壓縮比
            bili = int(w / image_c)
            # 按比例對寬高壓縮
            file_min.thumbnail((w // bili, h // bili))
            # 生成縮圖的完整檔名
            file_name_min = first_name + '_min.' + file_name_hz
            # 儲存縮圖
            file_min.save(os.path.join(app.config['UPLOAD_FOLDER'], file_name_min))
            # 返回原本和縮圖的 完整瀏覽連結
            return {"code": '200', "image_url": image_url + file_name, "image_url_min": image_url + file_name_min,
                    "message": "上傳成功"}
        else:
            return "格式錯誤，僅支援jpg、png、jpeg格式檔案"
    return {"code": '503', "data": "", "message": "僅支援post方法"}
if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8002, debug=True)  # 專案入口

執行app.py