flask 之上传本地图片

1
2
3
4
5
6
7
8
9
10
11
12
13
14

import osclass Config(object):
    DEBUG = True
    SQLALCHEMY_DATABASE_URI = 'mysql://root:mysql@127.0.0.1:3306/day08'
    SQLALCHEMY_TRACK_MODIFICATIONS = False    # 构建项目所在的 绝对路径，也就是 day08 的绝对路径
    BASE_DIR = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
    # 静态资源存放路径
    STATIC_ROOT = os.path.join(BASE_DIR, 'static/')
    # 自定义的 图片上传路径
    MEDIA_ROOT = os.path.join(BASE_DIR, 'media/')

1

app = Flask(__name__, static_folder=Config.STATIC_ROOT, static_url_path='/static')

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

import osfrom uuid import uuid4
from flask_restful import Resource, reqparse
from werkzeug.datastructures import FileStoragefrom app.config import Config
from .models import *def img_upload(img):
    if not img:
        return None    # 将图片名按照 .  进行切分， 找到最后一个元素，也就是  文件的后缀名
    end_name = img.filename.rsplit('.')[-1]    # 通过文件的后缀名判断 身份为 合法的  图片
    if end_name not in ['jpg', 'png', 'gif', 'jpeg']:
        return None    filename = str(uuid4()) + '.' + end_name  # 为了生成一个不重复的 文件名
    img_path = os.path.join(Config.STATIC_ROOT, filename)  # 将路径和文件名拼接在一起，方便保存文件    img.save(img_path)  # 将图片对象保存到 本地    return filenameclass NewsView(Resource):
    def post(self):
        # 1. 创建解析参数的对象
        parser = reqparse.RequestParser()
        # 2. 指明需要解析的参数
        parser.add_argument('title', type=str, location='form', required=True)
        parser.add_argument('img', type=FileStorage, location='files')        # 3. 获取具体的参数
        args = parser.parse_args()        title = args.get('title')
        img = args.get('img')        # 利用自定义函数，将图片保存到本地
        filename = img_upload(img)        # 4. 创建对象, 注意：图片存储的只是 从media之后的  图片路径
        news = News(title=title, img=filename)
        # 5. 添加到 事务中
        db.session.add(news)
        # 6. 提交事务
        try:
            db.session.commit()
        except:
            return {
                       'msg': '添加失败'
                   }, 500
        # 7. 返回响应
        return {
                   'id': news.id,
                   'title': news.title,
                   'img': news.img
               }, 201

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

import os
from flask import make_response
from app.config import Configclass ImgView(Resource):
    def get(self, filename):
        # 1. 拼接除图片的完成路径
        img_path = os.path.join(Config.MEDIA_ROOT, filename)        # 2. 按照二进制方式打开文件，读到的内容为 二进制文件流，方便接下来的网络传输
        try:
            with open(img_path, 'rb') as f:
                img = f.read()
        except FileNotFoundError:
            return None, 404        # 3. 自定义响应
        resp = make_response(img)
        # 4. 声明响应体的类型 为  图片
        resp.headers['Content-Type'] = 'image/png'
        # 5. 返回响应
        return resp

1
2
3
4
5
6
7
8
9
10
11
12
13

import os
from flask import send_from_directoryfrom app.config import Configclass ImgView(Resource):
    def get(self, filename):
        return send_from_directory(Config.MEDIA_ROOT, filename)