幾行代碼教你用python web框架做文件流下載

前不久公司有個需求,把時序數據庫中的日誌下載到本地,大家都知道,數據庫裏的數據都是存在數據庫裏的(廢話),想把他下載到客戶的本地, 有的同學第一反應是: 只有文件才能下載,所以大多數同學會想到先把數據從數據庫中讀出來,然後寫入到服務器中的某個文件夾下生成文件, 然後再下載。其實這是非常不效率的方法,最簡單的方法是,我們從數據庫中讀取到文件後,直接以流的形式讓用戶去下載。

幾行代碼教你用python web框架做文件流下載

 

這裏我拿python flask框架來做例子,其實非常簡單,步驟一共有3個

1: 取出數據, 我們默認爲json文件吧

2: 在response中加入指定的header

3: 將數據返回.

那麼廢話少說,我們直接上代碼:

# coding:utf-8

import json

 

from flask import Flask, make_response

 

app = Flask(__name__)

 

@app.route('/download', methods=["GET"])

def download():user = {'name': 'dewei', 'age': 33}

 

  data = json.dumps(user)

  response = make_response(data)

  response.headers['content-type'] = 'application/octet-stream;charset=utf-8'

  response.headers['content-disposition'] = 'attachment;filename=user.json'

  return response

 

if __name__ == '__main__':

  app.run(host='0.0.0.0', port=5005, debug=True)

這是一個非常簡單的demo.

flask的定義我們不去關心了. 這裏最重要的是獲取response, 並且在response的headers中,加入content-type 定義爲octet-stream 流的操作. 另外一個key是 content-disposition ,這裏主要定義文件的名稱. 這兩部完成之後,我們只需要return response即可, 重點就在這兩個header上,然後傳入數據信息(這裏我們定義了一個json,大家也可以定義其他文件,比如excel,當然 excel請大家查詢一下對應的content-type. 還是非常簡單的, 這個方法對於任何web框架都是用, 比如 我們常用的 django, tornado web.py 都是一樣, 只不過語法在不同的框架裏,方法名和用法可能有簡單區別,但換湯不換藥,原理是一樣的.

最後說一下用法,一般來說 下載的鏈接都是 get 所以前端只需要定義個 a標籤 添加 download屬性即可, demo如下:

<a href="xxxx/download" download>文件下載</a>

好了,很簡單的一個分享,希望對大家有幫助

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章