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