[Python知识库]2021-09-18

django更改存储方式，实现连接sftp上传文件到其他服务器

下面只是简单地实现方法(demo)，具体内容视自己需求具体使用

关于使用drf-yasg中的swagger实现能够上传文件的测试功能

需要在视图文件中使用解析器，为了更改上传文件的头部信息，重点在’parser_classes’ 详情，可以参考drf-yasg的git中issues，有相关内容，

安装django-storages:

? (django-storages封装了对python连接sftp的连接)

pip install django-storages

setting.py中配置SFTPStorages作为后端存储文件的方式

# 更改django的默认存储
DEFAULT_FILE_STORAGE = 'storages.backends.sftp.SFTPStorage'
# 配置sftp的服务器连接时的信息
SFTP_STORAGE_HOST = '10.69.39.25'
SFTP_STORAGE_ROOT = 'home/apadmin/'# 文件在sftp服务器上的存储位置
SFTP_STORAGE_PARAMS = {
    'username': 'ROOT',
    'password': 'ROOT',
    'allow_agent': False,# 禁用連接到SSH代理
    'look_for_keys': False,# 禁用搜索可發現的私鑰文件 ~/.ssh/
}
SFTP_STORAGE_INTERACTIVE = False # 一個布爾值，指示在無法使用密鑰建立連接時是否提示輸入密碼

配置model.py

from django.db import modelsfrom storages.backends.sftpstorage import SFTPStorage

# Create your models here.
# 导入存放文件的位置
from drf_vue_blog.settings import SFTP_STORAGE_ROOT
# SFTP 作为存储的方式
sftp = SFTPStorage()
class Test(models.Model):
    # 表中存放文件名字和文件路径，path指向sftp服务器存放路径
    filename = models.CharField(max_length=100,verbose_name='存放文件的名字',default='')
    filepath = models.FilePathField(path=SFTP_STORAGE_ROOT,default='', verbose_name='存放文件的路徑')

views.py

class TestView(APIView):
	parser_classes = (MultiPartParser,)
	@swagger_auto_schema(operation_description='Upload File',
                         operation_id='Upload File',
                         manual_parameters=[openapi.Parameter(name='file',
                                                              in_=openapi.IN_FORM,
                                                              type=openapi.TYPE_FILE,
                                                              required=True,
                                                              description='Upload File'),]
    def post(self,request):
	# 获取用户上传的文件
    obj = request.FILES.get('file')
    # file_path = os.path.join(obj.name)
    # f = open(file_path,'wb')
    # 打开文件循环写入到sftp服务器中
    f = sftp.sftp.open(obj.name,'wb')
    for i in obj.chunks():
        f.write(i)
    f.close()
    # 数据库表中插入数据，存放文件名字和路径位置
    Test.objects.create(filename=obj.name,
                        filepath=os.path.join(SFTP_STORAGE_ROOT,obj.name))
    
# urls.py
from django.contrib import admin
from django.urls import path, include

from article.views import TestView

urlpatterns = [
    # 上传文件到SFTP
    path('upload', TestView.as_view())
    # 下载文件从SFTP
    path('download',DownloadView.as_view())
]

class productpnserializer(serializers.Serializer):
    file_name = serializers.CharField(label='下載文件的名字', required=True)
class DownloadView(APIView):
    @swagger_auto_schema(query_serializer=productpnserializer,operation_summary='下載文件')
    def get(self,request):
        last = Test.objects.last()
        print(last.filename,last.filepath) # 通信錄更新.xlsx home/apadmin/通信錄更新.xlsx或者DRF.md home/apadmin/DRF.md
        
        
        file_name = request.query_params.get('file_name')
        if not file_name:
            pass
            # 邏輯處理
            # TODO

        # with sftp.sftp.open(file_name, 'rb') as f:

        response = FileResponse(sftp.sftp.open(file_name, 'rb'))
        response['Content-Type'] = 'application/octet-stream'
        # response['Content-Disposition'] = 'attachment;filename*=UTF-8"{}"'.format(escape_uri_path(file_name))
        response["Content-Disposition"] = "attachment; filename*=UTF-8''{}".format(escape_uri_path(file_name))
        return response