用Flask寫一個(gè)上傳圖片的API

話說(shuō)星期一的時(shí)候,我們數(shù)據(jù)組開(kāi)會(huì)布置這個(gè)禮拜的人。組長(zhǎng)這樣跟我說(shuō):
“這兩天你把上禮拜爬的那幾萬(wàn)張圖片都上傳到阿里云去”
“嗯”
“還有把里面的廣告啊水印啊都去掉,我這里買個(gè)網(wǎng)易云盾讓你用”
“嗯”
“這些星期三大概就完成了”
“嗯”
“然后你用flask寫一個(gè)上傳圖片的API,方便以后使用,這樣就不需要每次都再爬蟲那里處理圖片”
“哈?”
“哈!”
“什么?”
“你不是會(huì)flask的么”
“會(huì)啊!”
“那不就得了”
“。。。。?!?br> 我當(dāng)時(shí)心里是這樣想的:
1.寫一個(gè)專門用來(lái)上傳圖片的程序,部署到服務(wù)器?難道是亞馬遜的lambada那個(gè)按次收費(fèi)的服務(wù),跑一次就給一次錢?可是,圖片上傳為什么要用flask?
2.flask的API不是用來(lái)返回json格式的么,怎么上傳圖片啊?我先傳好圖片,寫進(jìn)數(shù)據(jù)庫(kù),然后flask從數(shù)據(jù)庫(kù)讀取?
怎么都想不通,到底是要干什么啊。然后,我就糾結(jié)了一個(gè)晚上。結(jié)果第二天睡醒,一道靈光打進(jìn)我天靈蓋,突然醒悟了:MDZZ。。。。誰(shuí)說(shuō)flask不能寫上傳圖片的功能的。。。而且還想用亞馬遜那個(gè)按次收費(fèi)跑代碼的服務(wù)。。。我二啊,直接把flask部署到服務(wù)器,愛(ài)怎么用就怎么用了。

我的思路清晰了:
Flask的API就使用用來(lái)返回的json。我要把原圖片地址作為參數(shù)傳遞給Flask,然后經(jīng)過(guò)里面的函數(shù)處理把圖片上傳,最后返回上傳后的圖片地址給我。
下面是效果圖

IMG_3818.PNG

思路有了,那就開(kāi)始寫代碼了

先寫個(gè)連接服務(wù)器和上傳功能的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 = #認(rèn)證的秘鑰
        endpoint = #服務(wù)器域名
        bucket_name =#服務(wù)器的盤名
        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

然后寫個(gè)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()#連接服務(wù)器
    dir_name = 'location'#設(shè)置好服務(wù)器保存的文件夾
    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)

把這個(gè)API寫完并部署好以后,我突然覺(jué)得自己吊吊的。以前沒(méi)工作時(shí)總聽(tīng)一些程序員前輩說(shuō)寫API的,原來(lái)這就是寫API啊!今天算是第一次啦!
我正洋洋得意的時(shí)候,突然聽(tīng)到外面總經(jīng)理妹子在哪里吼“剛才面試的人呢?把他叫回來(lái),忘記讓他做筆試題了?。?!”
我弱弱地問(wèn)“郵箱寄給他不行么?”
“不現(xiàn)場(chǎng)做,難道讓他回去百度呀?”
我的心里突然拔涼拔涼的。。。。。你當(dāng)初不就是直接手機(jī)發(fā)了份題讓我在家百度的,做完以后直接錄用我的么?
<strong>難道,你不是看上我的才華才請(qǐng)我的?`</strong>

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容