Flask實現圖片的上傳、下載及展示


用Flask處理圖片非常容易,這一篇學習一下圖片的上傳、下載及展示。還是以實例代碼演示為主。

首先,實現一個簡單的上傳(過程中未做任何處理,只是為了演示)
 

點擊選擇圖片,輸入李四:

HTML代碼:

  1.  
    <!DOCTYPE html>
  2.  
    <html lang="en">
  3.  
    <head>
  4.  
    <meta charset="UTF-8">
  5.  
    <title>Title</title>
  6.  
    </head>
  7.  
    <body>
  8.  
    <div>
  9.  
    <form method="post" action="http://localhost:5000/up_photo" enctype="multipart/form-data">
  10.  
    <input type="file" size="30" name="photo"/>
  11.  
    <br>
  12.  
    <input type="text" class="txt_input" name="name" style="margin-top:15px;"/>
  13.  
    <input type="submit" value="提交信息" class="button-new" style="margin-top:15px;"/>
  14.  
    </form>
  15.  
    </div>
  16.  
    </body>
  17.  
    </html>

 

點擊提交信息后,通過 http://localhost:5000/up_photo,傳遞到后台對應的路由處理,查看控制台:

 

后台接收的路由代碼:

  1.  
    basedir = os.path.abspath(os.path.dirname(__file__))
  2.  
     
  3.  
    @app.route('/up_photo', methods=['post'])
  4.  
    def up_photo():
  5.  
    img = request.files.get( 'txt_photo')
  6.  
    username = request.form.get( "name")
  7.  
    path = basedir+ "/static/photo/"
  8.  
    file_path = path+img.filename
  9.  
    img.save(file_path)
  10.  
    print '上傳頭像成功,上傳的用戶是:'+username
  11.  
    return render_template('index.html')

此時,我們查看static/photo時,已經可以看到我們點擊上傳的圖片了。

以上是最簡單的一個文件上傳示例,我們實際使用中基本不可能這樣去寫,通常會做一些處理。例如:上傳后做圖片名稱唯一性的處理、限制上傳文件的類型、無論上傳成功或者失敗,都去做一些返回提示等等。
 

 

限制上傳文件類型,例如,我們希望用戶上傳的文件為圖片類型

  1.  
    ALLOWED_EXTENSIONS = set([ 'png', 'jpg', 'JPG', 'PNG', 'gif', 'GIF'])
  2.  
     
  3.  
    def allowed_file(filename):
  4.  
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS

上面的代碼片段通過截取上傳文件的后綴名,判斷是否是屬於我們期望用戶上傳的類型。
 

 

更改上傳圖片名稱,防止重名

 

這里我們使用日期加隨機數生成圖片名,做一個生成字符串的工具類

 

  1.  
    #-*-coding:utf-8-*-
  2.  
    import datetime
  3.  
    import random
  4.  
    class Pic_str:
  5.  
    def create_uuid(self): #生成唯一的圖片的名稱字符串,防止圖片顯示時的重名問題
  6.  
    nowTime = datetime.datetime.now().strftime( "%Y%m%d%H%M%S"); # 生成當前時間
  7.  
    randomNum = random.randint( 0, 100); # 生成的隨機整數n,其中0<=n<=100
  8.  
    if randomNum <= 10:
  9.  
    randomNum = str( 0) + str(randomNum);
  10.  
    uniqueNum = str(nowTime) + str(randomNum);
  11.  
    return uniqueNum;

 

當我們點擊圖片上傳,選擇圖片點擊提交后,頁面返回提示信息如下圖:

 

此時看我們工程下的upload文件夾,唯一名稱的圖片已經上傳成功

 

圖片的下載

  1.  
    @app.route('/download/<string:filename>', methods=['GET'])
  2.  
    def download(filename):
  3.  
    if request.method == "GET":
  4.  
    if os.path.isfile(os.path.join('upload', filename)):
  5.  
    return send_from_directory('upload', filename, as_attachment=True)
  6.  
    pass

 

圖片的展示

  1.  
    # show photo
  2.  
    @app.route('/show/<string:filename>', methods=['GET'])
  3.  
    def show_photo(filename):
  4.  
    file_dir = os.path.join(basedir, app.config[ 'UPLOAD_FOLDER'])
  5.  
    if request.method == 'GET':
  6.  
    if filename is None:
  7.  
    pass
  8.  
    else:
  9.  
    image_data = open(os.path.join(file_dir, '%s' % filename), "rb").read()
  10.  
    response = make_response(image_data)
  11.  
    response.headers[ 'Content-Type'] = 'image/png'
  12.  
    return response
  13.  
    else:
  14.  
    pass

 

完整示例代碼

  1.  
    #encoding:utf-8
  2.  
    #!/usr/bin/env python
  3.  
    from werkzeug.utils import secure_filename
  4.  
    from flask import Flask, render_template, jsonify, request, make_response, send_from_directory, abort
  5.  
    import time
  6.  
    import os
  7.  
    from strUtil import Pic_str
  8.  
    import base64
  9.  
     
  10.  
    app = Flask(__name__)
  11.  
    UPLOAD_FOLDER = 'upload'
  12.  
    app.config[ 'UPLOAD_FOLDER'] = UPLOAD_FOLDER
  13.  
    basedir = os.path.abspath(os.path.dirname(__file__))
  14.  
    ALLOWED_EXTENSIONS = set([ 'png', 'jpg', 'JPG', 'PNG', 'gif', 'GIF'])
  15.  
     
  16.  
    def allowed_file(filename):
  17.  
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
  18.  
     
  19.  
     
  20.  
    @app.route('/upload')
  21.  
    def upload_test():
  22.  
    return render_template('up.html')
  23.  
     
  24.  
     
  25.  
    # 上傳文件
  26.  
    @app.route('/up_photo', methods=['POST'], strict_slashes=False)
  27.  
    def api_upload():
  28.  
    file_dir = os.path.join(basedir, app.config[ 'UPLOAD_FOLDER'])
  29.  
    if not os.path.exists(file_dir):
  30.  
    os.makedirs(file_dir)
  31.  
    f = request.files[ 'photo']
  32.  
    if f and allowed_file(f.filename):
  33.  
    fname = secure_filename(f.filename)
  34.  
    print fname
  35.  
    ext = fname.rsplit( '.', 1)[1]
  36.  
    new_filename = Pic_str().create_uuid() + '.' + ext
  37.  
    f.save(os.path.join(file_dir, new_filename))
  38.  
     
  39.  
    return jsonify({"success": 0, "msg": "上傳成功"})
  40.  
    else:
  41.  
    return jsonify({"error": 1001, "msg": "上傳失敗"})
  42.  
     
  43.  
    @app.route('/download/<string:filename>', methods=['GET'])
  44.  
    def download(filename):
  45.  
    if request.method == "GET":
  46.  
    if os.path.isfile(os.path.join('upload', filename)):
  47.  
    return send_from_directory('upload', filename, as_attachment=True)
  48.  
    pass
  49.  
     
  50.  
     
  51.  
    # show photo
  52.  
    @app.route('/show/<string:filename>', methods=['GET'])
  53.  
    def show_photo(filename):
  54.  
    file_dir = os.path.join(basedir, app.config[ 'UPLOAD_FOLDER'])
  55.  
    if request.method == 'GET':
  56.  
    if filename is None:
  57.  
    pass
  58.  
    else:
  59.  
    image_data = open(os.path.join(file_dir, '%s' % filename), "rb").read()
  60.  
    response = make_response(image_data)
  61.  
    response.headers[ 'Content-Type'] = 'image/png'
  62.  
    return response
  63.  
    else:
  64.  
    pass
  65.  
     
  66.  
     
  67.  
    if __name__ == '__main__':
  68.  
    app.run(debug= True)

基本示例就以上這些了!
原文章地址:https://blog.csdn.net/weixin_36380516/article/details/80347192


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM