用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 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>

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,100评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,308评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,718评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,275评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,376评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,454评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,464评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,248评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,686评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,974评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,150评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,817评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,484评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,140评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,374评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,012评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,041评论 2 351

推荐阅读更多精彩内容