python flask解決上傳下載的問題

　　記錄瞬間

 

最近為了解決一些新的需求，簡單介入了flask對文件的上傳和下載的方法，並分別使用python和curl模擬發送

 

代碼：

#! /usr/bin/env python3
# coding:utf-8
import platform

from werkzeug.utils import secure_filename
from flask import Flask, jsonify, request, Response
import os

app = Flask(__name__)
UPLOAD_FOLDER = 'upload'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 20 * 1024 * 1024   # 定義最大上傳文件大小為：20M

ALLOWED_EXTENSIONS = set(['txt', 'png', 'jpg', 'xls', 'JPG', 'PNG', 'zip', 'gif', 'GIF'])
run_path = "./"                         # 根據不同的操作系統，定義基礎運行路徑
if platform.system() == "Linux":
    run_path = r'/opt/AutoUpload/'
if platform.system() == "Windows":
    run_path = r'D:/PythonWorkSpace/'
msg = 'niGEin!'


# 用於判斷文件后綴
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS


# 上傳文件-upload-file
@app.route('/uf', methods=['POST'], strict_slashes=False)
def api_upload():
    file_dir = run_path + UPLOAD_FOLDER
    if not os.path.exists(file_dir):
        os.makedirs(file_dir)
    f = request.files['file']                             # 獲取上傳文件

    print(request.values.get("filePath"))
    

    fname = secure_filename(f.filename)
    ext = fname.rsplit('.', 1)[1]                          # 獲取文件后綴

    f.save(os.path.join(file_dir, fname))                  # 保存文件到upload目錄
    if ext == 'zip':
        pass
    return jsonify({"errno": "000000", "errmsg": u"success"})


# 下載文件-download-file
@app.route('/df', methods=['GET', 'POST'])
def api_download():
    if request.method == 'GET':
        fullfilename = request.json['fileName']
        print(fullfilename)
    filepath = run_path + 'tools/' + fullfilename
    print(filepath)
    if not os.path.isfile(filepath):
        print("nononononono!!!")
        return
    # 普通下載
    # response = make_response(send_from_directory(filepath, fullfilename, as_attachment=True))
    # response.headers["Content-Disposition"] = "attachment; filename={}".format(filepath.encode().decode('latin-1'))
    # return response

    # 流式讀取
    def send_file():
        store_path = filepath
        with open(store_path, 'rb') as targetfile:
            while 1:
                data = targetfile.read(1 * 1024 * 1024)  # 每次讀取1M
                if not data:
                    break
                yield data

    response = Response(send_file(), content_type='application/octet-stream')
    response.headers["Content-disposition"] = 'attachment; filename=%s' % fullfilename 
    return response


if __name__ == '__main__':
    app.run(debug=True, port=5002, host='0.0.0.0')
    # 默認127.0.0.1:5000，這里修改了地址和端口方便自己使用

 

調用方式：

# coding:utf-8
import requests
from urllib3 import encode_multipart_formdata


url = "http://localhost:5002/up"
data = {
            "filePath": "123123123"
        }
header = {}
data['file'] = ("xx.zip", open(r"./basedir/xx.zip", 'rb').read())
encode_data = encode_multipart_formdata(data)
data = encode_data[0]
header['Content-Type'] = encode_data[1]
try:
    result = requests.request(method='POST', url=url, headers=header, data=data, timeout=(3, 100))

    if "true" in result.text:
        analyse_json = result.json()
        print("向服務器發送文件並解壓成功")
        result_path = analyse_json["data"]
        print("服務器端的地址為 {}".format(result_path))
    else:
        print("向服務器發送文件並解壓Failed {}".format(result.text))
except Exception as e:
    print("執行發送數據失敗.{}".format(e))



#--------------------------------------------

url = "http://localhost:5002/df"
data = {
            "fileName": "xx.jar"
        }

result = requests.request(method="GET", url=url, json=data, stream=True)
f = open(data['fileName'], "wb")
for chunk in result.iter_content(chunk_size=512):
    if chunk:
        f.write(chunk)

#---------------------------------------------

 

使用curl命令進行發送文件的方式：

 

curl ${URL} -X POST -F "file=@${app_path}/${APP_NAME}.zip"  -F "ip1=${IP}" -F "ip2=${get_ip}" -F "port=${port}" -F "num=${num}"