用Flask写一个上传图片的API-原创手记-慕课网

话说星期一的时候，我们数据组开会布置这个礼拜的人。组长这样跟我说：
“这两天你把上礼拜爬的那几万张图片都上传到阿里云去”
“嗯”
“还有把里面的广告啊水印啊都去掉，我这里买个网易云盾让你用”
“嗯”
“这些星期三大概就完成了”
“嗯”
“然后你用flask写一个上传图片的API，方便以后使用，这样就不需要每次都再爬虫那里处理图片”
“哈？”
“哈！”
“什么？”
“你不是会flask的么”
“会啊！”
“那不就得了”
“。。。。。”
我当时心里是这样想的：
1.写一个专门用来上传图片的程序，部署到服务器？难道是亚马逊的lambada那个按次收费的服务，跑一次就给一次钱？可是，图片上传为什么要用flask？
2.flask的API不是用来返回json格式的么，怎么上传图片啊？我先传好图片，写进数据库，然后flask从数据库读取？
怎么都想不通，到底是要干什么啊。然后，我就纠结了一个晚上。结果第二天睡醒，一道灵光打进我天灵盖，突然醒悟了：MDZZ。。。。谁说flask不能写上传图片的功能的。。。而且还想用亚马逊那个按次收费跑代码的服务。。。我二啊，直接把flask部署到服务器，爱怎么用就怎么用了。

我的思路清晰了：
Flask的API就使用用来返回的json。我要把原图片地址作为参数传递给Flask，然后经过里面的函数处理把图片上传，最后返回上传后的图片地址给我。
下面是效果图

IMG_3818.PNG

思路有了，那就开始写代码了

先写个连接服务器和上传功能的utils
utils

import sys
import os
import oss2
import uuid
import requestsclass QssClient(object):
    def __new__(cls, *args, **kw):        if not hasattr(cls, '_instance'):
            orig = super(QssClient, cls)
            cls._instance = orig.__new__(cls, *args, **kw)        return cls._instance    def __init__(self):
        auth = #认证的秘钥
        endpoint = #服务器域名
        bucket_name =#服务器的盘名
        self.bucket = oss2.Bucket(auth, endpoint, bucket_name, connect_timeout=30)        super(QssClient, self).__init__()    def upload(self, file_path, dir_name='文件夹名'):
        file_name = str(uuid.uuid1())
        with open(file_path, 'rb') as f:
            remote = '%s/%s.jpg' % (dir_name, str(file_name))
            result = self.bucket.put_object(remote, f)
            file_url = result.resp.response.url
            f.close()
        os.remove(file_path)        return file_url    def upload_list(self, file_list, dir_name='location'):
        file_url_list = []        for f in file_list:
            file_url = self.upload(f, dir_name=dir_name)
            file_url_list.append(file_url)        return file_url_list

然后写个Flask的run
run

from flask import Flask, jsonify,requestfrom utils import QssClientimport urllibimport os

app = Flask(__name__)

tasks = [
    {        'id': 1,        'image_url': 'url from website',        'ali_url': 'url from ali',        'done': False
    },
    {        'id': 2,         'image_url': 'url from website',        'ali_url': 'url from ali',        'done': False
    }
]@app.route('/tasks', methods=['GET'])def get_tasks():
    return jsonify({'tasks': tasks})@app.route('/pics', methods=['POST'])def create_task():
    if not request.json or not 'image_url' in request.json:
        abort(400)



    urllib.urlretrieve(request.json['image_url'], 'pics\\' + request.json['image_url'].split('/')[-1] )#先保存图片到本地
    qs = QssClient()#连接服务器
    dir_name = 'location'#设置好服务器保存的文件夹
    full_file_lis = []
    file_path = os.path.join('pics\\', request.json['image_url'].split('/')[-1])#从本地的保存路径获得图片
    full_file_lis.append(file_path)
    url = qs.upload_list(full_file_lis, dir_name)[0]从本地上传图片并获得上传后的地址

    task = {        'id': tasks[-1]['id'] + 1,        'image_url': request.json['image_url'],        'ali_url': url,        'done': False
    }
    tasks.append(task)    return jsonify({'task': task}), 201#返回图片地址的json@app.route('/')def index():
    return "Hello, World!"if __name__ == '__main__':
    app.run(debug=True)

把这个API写完并部署好以后，我突然觉得自己吊吊的。以前没工作时总听一些程序员前辈说写API的，原来这就是写API啊！今天算是第一次啦！
我正洋洋得意的时候，突然听到外面总经理妹子在哪里吼“刚才面试的人呢？把他叫回来，忘记让他做笔试题了！！！”
我弱弱地问“邮箱寄给他不行么？”
“不现场做，难道让他回去百度呀？”
我的心里突然拔凉拔凉的。。。。。你当初不就是直接手机发了份题让我在家百度的，做完以后直接录用我的么？
<strong>难道，你不是看上我的才华才请我的？`</strong>

作者：LEONYao
链接：https://www.jianshu.com/p/36133ee1a1c2