8. Django系列之上传文件与下载-djang为服务端,requests为客户端

preface

运维平台新上线一个探测功能，需要上传文件到服务器上和下载文件从服务器上，那么我们就看看requests作为客户端，django作为服务器端怎么去处理？ 对于静态文件我们不建议通过django下载，而是建议通过django返回一个重定向URL（下载文件的URL）给client，这个url是nginx提供下载，众所周知，nginx是非常牛逼的静态web-server。
以下代码都是简单的代码，可以正常使用，只是实现最基本的要上传文件到服务器上和下载文件从服务器上,不涉及到公司的业务逻辑。文件可以通过md5校验，反正我测试都是一致的MD5值。

Author：温柔易淡

代码如下

使用的python2.7的版本，其他的如果没有做特殊说明是python2.7的版本，那么我的其他博客都是python3.5的版本

1. 先看客户端的代码：

#!/usr/bin/env python
# encode:utf-8
import requests

def post_file():   #文件上传（post方法）
    post_url='http://127.0.0.1/file/post/'

    data=[]
    fr=open(r'/tmp/Python-2.7.13.tgz','rb')
    file = {'file':('Python-2.7.13.tgz',open(r'/tmp/Python-2.7.13.tgz','rb'),'application/x-tar')}
    m = {'act': 'avatar', 'save': '1'}


    rt =requests.post(url=post_url,files=file,data=m)
    print(rt.status_code,rt.text)

def get_file():   #文件下载（get方法）
    get_url = 'http://127.0.0.1/file/get/'
    get_params = {'filename':'Python-2.7.13.tgz'}
    rt = requests.get(url=get_url,params=get_params)
    with open('/tmp/vmware-root/Python-2.7.13.tgz','wb') as f:
        # it's not img or others and so on ,if it is ,we need to use BytesIO to save
        # for example:
        #   from io import BytesIO
        #   i = Image.open(BytesIO(r.content))  # r is a  return object

        for chunk in rt.iter_content():
            f.write(chunk)
    print('ok')

#post_file()
get_file()

2. 再看服务器端的代码：

from django.shortcuts import render,HttpResponse
from django.http import StreamingHttpResponse

# Create your views here.


def file_get(request):   #文件下载（get方法）
    print(request.GET)
    file_name = request.GET.get('filename')
    file_path = "/root/%s"%(file_name)

    def file_read(file_name,chunk_size=1024):
        # itertor return the file's chunk, if the file size is very large,it must be useful, so server won't OOM 
        with open(file_name,'rb') as f:
            while True:
                chunks = f.read(chunk_size)
                if chunks:
                    print(chunks)
                    yield chunks
                else:
                    break

    response_data = StreamingHttpResponse(file_read(file_path))
    response_data['Content-Type'] = 'application/octet-stream'   # set the type as stream then PC will save it in their disk
    response_data['Content-Disposition'] = 'attachment;filename="%s"'%(file_name)   # set the file name

    return response_data

def file_post(request):  
    print(request.POST,request.FILES)
    file_obj = request.FILES.get('file')
    print(type(file_obj))
    if file_obj:
        save_file_path='/root/%s'%(file_obj.name)
        with open(save_file_path,'wb') as f:
            for chunks in file_obj.chunks():
                f.write(chunks)

    return HttpResponse('ok')

其他的url代码就不赘述了。