75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import hashlib
|
|
import mimetypes
|
|
|
|
from rest_framework import serializers
|
|
|
|
from application import dispatch
|
|
from dvadmin.system.models import FileList
|
|
from dvadmin.utils.serializers import CustomModelSerializer
|
|
from dvadmin.utils.viewset import CustomModelViewSet
|
|
|
|
|
|
class FileSerializer(CustomModelSerializer):
|
|
url = serializers.SerializerMethodField(read_only=True)
|
|
|
|
def get_url(self, instance):
|
|
# return 'media/' + str(instance.url)
|
|
return instance.file_url or (f'media/{str(instance.url)}')
|
|
|
|
class Meta:
|
|
model = FileList
|
|
fields = "__all__"
|
|
|
|
def create(self, validated_data):
|
|
file_engine = dispatch.get_system_config_values("fileStorageConfig.file_engine")
|
|
file_backup = dispatch.get_system_config_values("fileStorageConfig.file_backup")
|
|
file = self.initial_data.get('file')
|
|
file_size = file.size
|
|
validated_data['name'] = str(self.initial_data.get('file'))
|
|
validated_data['size'] = file_size
|
|
validated_data['md5sum'] = hashlib.md5().hexdigest()
|
|
validated_data['engine'] = file_engine
|
|
validated_data['mime_type'] = mimetypes.guess_type(str(file))[0]
|
|
if file_backup:
|
|
validated_data['url'] = file
|
|
if file_engine =='oss':
|
|
from dvadmin_cloud_storage.views.aliyun import ali_oss_upload
|
|
file_path = ali_oss_upload(file)
|
|
if file_path:
|
|
validated_data['file_url'] = file_path
|
|
else:
|
|
raise ValueError("上传失败")
|
|
elif file_engine == 'cos':
|
|
from dvadmin_cloud_storage.views.tencent import tencent_cos_upload
|
|
file_path = tencent_cos_upload(file)
|
|
if file_path:
|
|
validated_data['file_url'] = file_path
|
|
else:
|
|
raise ValueError("上传失败")
|
|
else:
|
|
validated_data['url'] = file
|
|
# 审计字段
|
|
try:
|
|
request_user = self.request.user
|
|
validated_data['dept_belong_id'] = request_user.dept.id
|
|
validated_data['creator'] = request_user.id
|
|
validated_data['modifier'] = request_user.id
|
|
except:
|
|
pass
|
|
return super().create(validated_data)
|
|
|
|
|
|
class FileViewSet(CustomModelViewSet):
|
|
"""
|
|
文件管理接口
|
|
list:查询
|
|
create:新增
|
|
update:修改
|
|
retrieve:单例
|
|
destroy:删除
|
|
"""
|
|
queryset = FileList.objects.all()
|
|
serializer_class = FileSerializer
|
|
filter_fields = ['name', ]
|
|
permission_classes = []
|