99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
import os
|
|
from rest_framework import viewsets, status
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from .models import FinancialRecord, Department, Category, ProofImage
|
|
from .serializers import (
|
|
FinancialRecordWriteSerializer,
|
|
FinancialRecordReadSerializer,
|
|
DepartmentSerializer,
|
|
CategorySerializer,
|
|
ProofImageSerializer
|
|
)
|
|
|
|
class FinancialRecordViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows financial records to be viewed or edited.
|
|
"""
|
|
queryset = FinancialRecord.objects.all()
|
|
|
|
def get_serializer_class(self):
|
|
"""根据操作类型返回不同的序列化器"""
|
|
if self.action in ['list', 'retrieve']:
|
|
return FinancialRecordReadSerializer
|
|
return FinancialRecordWriteSerializer
|
|
|
|
@action(detail=True, methods=['post'])
|
|
def upload_images(self, request, pk=None):
|
|
"""为财务记录上传多张凭证图片"""
|
|
record = self.get_object()
|
|
files = request.FILES.getlist('images')
|
|
|
|
if not files:
|
|
return Response({'error': '没有接收到图片文件'}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
created_images = []
|
|
for file in files:
|
|
proof_image = ProofImage.objects.create(
|
|
financial_record=record,
|
|
image=file,
|
|
description=request.data.get('description', '')
|
|
)
|
|
created_images.append(ProofImageSerializer(proof_image).data)
|
|
|
|
return Response({
|
|
'message': f'成功上传 {len(created_images)} 张图片',
|
|
'images': created_images
|
|
}, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
class ProofImageViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows proof images to be viewed or edited.
|
|
"""
|
|
queryset = ProofImage.objects.all()
|
|
serializer_class = ProofImageSerializer
|
|
|
|
def destroy(self, request, *args, **kwargs):
|
|
"""重写删除方法，确保删除图片记录时同时删除物理文件"""
|
|
instance = self.get_object()
|
|
|
|
# 获取文件路径
|
|
file_path = None
|
|
if instance.image:
|
|
try:
|
|
file_path = instance.image.path
|
|
except (ValueError, AttributeError):
|
|
# 如果文件路径无效或文件不存在，只删除数据库记录
|
|
pass
|
|
|
|
# 删除数据库记录
|
|
super().destroy(request, *args, **kwargs)
|
|
|
|
# 删除物理文件
|
|
if file_path and os.path.isfile(file_path):
|
|
try:
|
|
os.remove(file_path)
|
|
print(f"成功删除文件: {file_path}")
|
|
except OSError as e:
|
|
print(f"删除文件失败: {file_path}, 错误: {e}")
|
|
# 即使文件删除失败，也不抛出异常，因为数据库记录已经删除
|
|
|
|
return Response({'message': '图片删除成功'}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class DepartmentViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows departments to be viewed or edited.
|
|
"""
|
|
queryset = Department.objects.all()
|
|
serializer_class = DepartmentSerializer
|
|
|
|
|
|
class CategoryViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
API endpoint that allows categories to be viewed or edited.
|
|
"""
|
|
queryset = Category.objects.all()
|
|
serializer_class = CategorySerializer
|