Flask实现图片的上传、下载及展示示例代码

更新时间：2018年08月03日 13:49:45 作者：阿木侠

这篇文章主要介绍了Flask实现图片的上传、下载及展示示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

用Flask处理图片非常容易，这一篇学习一下图片的上传、下载及展示。还是以实例代码演示为主。

首先，实现一个简单的上传（过程中未做任何处理，只是为了演示）

点击选择图片，输入李四：

HTML代码：

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8">
  <title>Title</title>
</head>
  <body>
    <div>
      <form method="post" action="http://localhost:5000/up_photo" enctype="multipart/form-data">
      <input type="file" size="30" name="photo"/>
      <br>
      <input type="text" class="txt_input" name="name" style="margin-top:15px;"/>
      <input type="submit" value="提交信息" class="button-new" style="margin-top:15px;"/>
      </form>
    </div>
  </body>
</html>

点击提交信息后，通过 http://localhost:5000/up_photo，传递到后台对应的路由处理，查看控制台：

后台接收的路由代码：

basedir = os.path.abspath(os.path.dirname(__file__))
 
@app.route('/up_photo', methods=['post'])
def up_photo():
  img = request.files.get('txt_photo')
  username = request.form.get("name")
  path = basedir+"/static/photo/"
  file_path = path+img.filename
  img.save(file_path)
  print '上传头像成功，上传的用户是：'+username
  return render_template('index.html')

此时，我们查看static/photo时，已经可以看到我们点击上传的图片了。

以上是最简单的一个文件上传示例，我们实际使用中基本不可能这样去写，通常会做一些处理。例如：上传后做图片名称唯一性的处理、限制上传文件的类型、无论上传成功或者失败，都去做一些返回提示等等。

限制上传文件类型，例如，我们希望用户上传的文件为图片类型

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

上面的代码片段通过截取上传文件的后缀名，判断是否是属于我们期望用户上传的类型。

更改上传图片名称，防止重名

这里我们使用日期加随机数生成图片名，做一个生成字符串的工具类

#-*-coding:utf-8-*-
import datetime
import random
class Pic_str:
  def create_uuid(self): #生成唯一的图片的名称字符串，防止图片显示时的重名问题
    nowTime = datetime.datetime.now().strftime("%Y%m%d%H%M%S"); # 生成当前时间
    randomNum = random.randint(0, 100); # 生成的随机整数n，其中0<=n<=100
    if randomNum <= 10:
      randomNum = str(0) + str(randomNum);
    uniqueNum = str(nowTime) + str(randomNum);
    return uniqueNum;

当我们点击图片上传，选择图片点击提交后，页面返回提示信息如下图：

此时看我们工程下的upload文件夹，唯一名称的图片已经上传成功

图片的下载

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

图片的展示

# show photo
@app.route('/show/<string:filename>', methods=['GET'])
def show_photo(filename):
  file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
  if request.method == 'GET':
    if filename is None:
      pass
    else:
      image_data = open(os.path.join(file_dir, '%s' % filename), "rb").read()
      response = make_response(image_data)
      response.headers['Content-Type'] = 'image/png'
      return response
  else:
    pass

完整示例代码

#encoding:utf-8
#!/usr/bin/env python
from werkzeug.utils import secure_filename
from flask import Flask, render_template, jsonify, request, make_response, send_from_directory, abort
import time
import os
from strUtil import Pic_str
import base64
 
app = Flask(__name__)
UPLOAD_FOLDER = 'upload'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
basedir = os.path.abspath(os.path.dirname(__file__))
ALLOWED_EXTENSIONS = set(['png', 'jpg', 'JPG', 'PNG', 'gif', 'GIF'])
 
def allowed_file(filename):
  return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
 
 
@app.route('/upload')
def upload_test():
  return render_template('up.html')
 
 
# 上传文件
@app.route('/up_photo', methods=['POST'], strict_slashes=False)
def api_upload():
  file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
  if not os.path.exists(file_dir):
    os.makedirs(file_dir)
  f = request.files['photo']
  if f and allowed_file(f.filename):
    fname = secure_filename(f.filename)
    print fname
    ext = fname.rsplit('.', 1)[1]
    new_filename = Pic_str().create_uuid() + '.' + ext
    f.save(os.path.join(file_dir, new_filename))
 
    return jsonify({"success": 0, "msg": "上传成功"})
  else:
    return jsonify({"error": 1001, "msg": "上传失败"})
 
@app.route('/download/<string:filename>', methods=['GET'])
def download(filename):
  if request.method == "GET":
    if os.path.isfile(os.path.join('upload', filename)):
      return send_from_directory('upload', filename, as_attachment=True)
    pass
  
  
# show photo
@app.route('/show/<string:filename>', methods=['GET'])
def show_photo(filename):
  file_dir = os.path.join(basedir, app.config['UPLOAD_FOLDER'])
  if request.method == 'GET':
    if filename is None:
      pass
    else:
      image_data = open(os.path.join(file_dir, '%s' % filename), "rb").read()
      response = make_response(image_data)
      response.headers['Content-Type'] = 'image/png'
      return response
  else:
    pass
 
 
if __name__ == '__main__':
  app.run(debug=True)

基本示例就以上这些了！

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python的重要技能输入与输出字符串格式化使用详解
这篇文章主要为大家介绍了python的重要技能输入与输出格式化使用详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2023-06-06
Python语言实现机器学习的K-近邻算法
今天学习的算法是KNN近邻算法。KNN算法是一个监督学习分类器类别的算法。下面我们来详细的探讨下
2015-06-06
Python对FTP交互封装的实现
本文主要介绍了Python对FTP交互封装的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-06-06
解决Keras TensorFlow 混编中 trainable=False设置无效问题
这篇文章主要介绍了解决Keras TensorFlow 混编中 trainable=False设置无效问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
通过python-turtle库实现绘制图画
turtle库是python的基础绘图库，经常被用来给孩子们介绍编程知识的方法库，是标准库之一，利用turtle可以制作很多复杂的绘图。本文将为大家介绍通过turtle库绘制的一些有趣的图画，感兴趣的小伙伴可以学习一下
2021-12-12
pytorch实现特殊的Module--Sqeuential三种写法
今天小编就为大家分享一篇pytorch实现特殊的Module--Sqeuential三种写法。具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
利用PyQt5+Matplotlib 绘制静态/动态图的实现代码
这篇文章主要介绍了利用PyQt5+Matplotlib 绘制静态/动态图的实现代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-07-07
Python中的 dir() 函数示例详解
dir()函数是Python 中一个非常有用的工具，可以用于查找对象的所有属性和方法，如获取当前作用域的变量和方法、查找模块中的导出内容、动态查找对象属性等，通过本文的介绍和示例代码，大家可以更全面地了解 dir() 函数的用法和注意事项，需要的朋友参考下吧
2022-03-03
django Layui界面点击弹出对话框并请求逻辑生成分页的动态表格实例
这篇文章主要介绍了django Layui界面点击弹出对话框并请求逻辑生成分页的动态表格实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-05-05
Python Numpy学习之索引及切片的使用方法
数组中的元素可以通过索引以及切片的手段进行访问或者修改，和列表的切片操作一样。本文将详细为大家介绍一下Python中的科学计算库-Numpy的索引及切片的使用方法
2022-01-01

Flask实现图片的上传、下载及展示示例代码

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具