用Flask写一个上传图片的API

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

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

《用Flask写一个上传图片的API》 IMG_3818.PNG

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

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

import sys
import os
import oss2
import uuid
import requests





class 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,request
from utils import QssClient
import urllib
import 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
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞