forked from erp-dev/erp
379 lines
14 KiB
Python
379 lines
14 KiB
Python
from decimal import Decimal
|
|
|
|
from rest_framework import status, views, serializers, pagination
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from django.db.models import Sum
|
|
from django.utils import dateparse, timezone
|
|
|
|
from api_v1.views.stock_change_views.mixins import StockChangeViewMixin
|
|
from business import models as business_models
|
|
from business import pre_order_services
|
|
from business import allocation_services
|
|
|
|
|
|
class AllocationRecordSerializer(serializers.ModelSerializer):
|
|
stock_ids = serializers.SerializerMethodField(read_only=True)
|
|
scanned_by_name = serializers.CharField(source='scanned_by.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = business_models.AllocationRecord
|
|
fields = [
|
|
'id',
|
|
'pre_sales_order_item',
|
|
'stock_ids',
|
|
'quantity',
|
|
'unit',
|
|
'status',
|
|
'scanned_by',
|
|
'scanned_by_name',
|
|
'scanned_at',
|
|
'remarks',
|
|
'created_at',
|
|
]
|
|
read_only_fields = [
|
|
'id',
|
|
'pre_sales_order_item',
|
|
'scanned_by',
|
|
'scanned_by_name',
|
|
'scanned_at',
|
|
'created_at',
|
|
]
|
|
|
|
def get_stock_ids(self, obj: business_models.AllocationRecord):
|
|
return obj.stock_id_list
|
|
|
|
|
|
class PreSalesOrderItemSerializer(serializers.ModelSerializer):
|
|
allocation_records = AllocationRecordSerializer(many=True, read_only=True)
|
|
allocated_quantity = serializers.SerializerMethodField(read_only=True)
|
|
progress = serializers.SerializerMethodField(read_only=True)
|
|
allocation_status = serializers.SerializerMethodField(read_only=True)
|
|
|
|
class Meta:
|
|
model = business_models.PreSalesOrderItem
|
|
fields = [
|
|
'id',
|
|
'product_id',
|
|
'product_name',
|
|
'color',
|
|
'quantity',
|
|
'unit',
|
|
'spec',
|
|
'remarks',
|
|
'allocated_quantity',
|
|
'progress',
|
|
'allocation_status',
|
|
'allocation_records',
|
|
]
|
|
read_only_fields = ['id']
|
|
|
|
def _get_allocated_quantity(self, obj: business_models.PreSalesOrderItem) -> Decimal:
|
|
result = obj.allocation_records.filter(
|
|
status=business_models.AllocationRecordStatusEnum.ACTIVE
|
|
).aggregate(total=Sum('quantity'))
|
|
return result.get('total') or Decimal('0')
|
|
|
|
def get_allocated_quantity(self, obj: business_models.PreSalesOrderItem) -> str:
|
|
return str(self._get_allocated_quantity(obj))
|
|
|
|
def get_progress(self, obj: business_models.PreSalesOrderItem) -> str:
|
|
required = obj.quantity or Decimal('0')
|
|
allocated = self._get_allocated_quantity(obj)
|
|
if required <= 0:
|
|
return '0'
|
|
return str((allocated / required).quantize(Decimal('0.0001')))
|
|
|
|
def get_allocation_status(self, obj: business_models.PreSalesOrderItem) -> int:
|
|
required = obj.quantity or Decimal('0')
|
|
allocated = self._get_allocated_quantity(obj)
|
|
if allocated <= 0:
|
|
return 1
|
|
if allocated < required:
|
|
return 2
|
|
return 3
|
|
|
|
|
|
class PreSalesOrderSerializer(serializers.ModelSerializer):
|
|
human_id = serializers.CharField(read_only=True)
|
|
customer_name = serializers.CharField(source='customer.name', read_only=True)
|
|
warehouse_name = serializers.CharField(source='warehouse.name', read_only=True)
|
|
operator_name = serializers.CharField(source='operator.name', read_only=True)
|
|
items = PreSalesOrderItemSerializer(many=True, read_only=True)
|
|
|
|
class Meta:
|
|
model = business_models.PreSalesOrder
|
|
fields = [
|
|
'id',
|
|
'human_id',
|
|
'merchant',
|
|
'customer',
|
|
'customer_name',
|
|
'warehouse',
|
|
'warehouse_name',
|
|
'operator',
|
|
'operator_name',
|
|
'kind',
|
|
'remarks',
|
|
'created_at',
|
|
'items',
|
|
]
|
|
read_only_fields = [
|
|
'id',
|
|
'human_id',
|
|
'created_at',
|
|
'customer_name',
|
|
'warehouse_name',
|
|
'operator_name',
|
|
'items',
|
|
]
|
|
|
|
|
|
class PreSalesOrderPagination(pagination.LimitOffsetPagination):
|
|
default_limit = 20
|
|
max_limit = 100
|
|
|
|
|
|
class PreSalesOrderView(StockChangeViewMixin, views.APIView):
|
|
"""预销售单查询与创建"""
|
|
|
|
permission_classes = [IsAuthenticated]
|
|
pagination_class = PreSalesOrderPagination
|
|
|
|
def get(self, request):
|
|
if not self.check_employee_permission(request):
|
|
return self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
queryset = pre_order_services.list_pre_sales_orders(merchant=merchant)
|
|
|
|
customer_id = request.query_params.get('customer') or request.query_params.get('customer_id')
|
|
warehouse_id = request.query_params.get('warehouse') or request.query_params.get('warehouse_id')
|
|
kind = request.query_params.get('kind')
|
|
human_id_contains = request.query_params.get('human_id__icontains')
|
|
created_from = request.query_params.get('created_at_from')
|
|
created_to = request.query_params.get('created_at_to')
|
|
|
|
try:
|
|
if customer_id:
|
|
queryset = queryset.filter(customer_id=int(customer_id))
|
|
if warehouse_id:
|
|
queryset = queryset.filter(warehouse_id=int(warehouse_id))
|
|
if kind:
|
|
queryset = queryset.filter(kind=int(kind))
|
|
except (TypeError, ValueError):
|
|
return Response({'error': '筛选参数必须为整数'}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
if human_id_contains:
|
|
queryset = queryset.filter(human_id__icontains=str(human_id_contains).strip())
|
|
|
|
def _parse_datetime(value: str):
|
|
dt = dateparse.parse_datetime(value)
|
|
if dt is None:
|
|
d = dateparse.parse_date(value)
|
|
if d is None:
|
|
return None
|
|
dt = timezone.datetime.combine(d, timezone.datetime.min.time())
|
|
if timezone.is_naive(dt):
|
|
dt = timezone.make_aware(dt, timezone.get_current_timezone())
|
|
return dt
|
|
|
|
if created_from:
|
|
dt_from = _parse_datetime(created_from)
|
|
if dt_from is None:
|
|
return Response({'error': 'created_at_from 格式不正确'}, status=status.HTTP_400_BAD_REQUEST)
|
|
queryset = queryset.filter(created_at__gte=dt_from)
|
|
|
|
if created_to:
|
|
dt_to = _parse_datetime(created_to)
|
|
if dt_to is None:
|
|
return Response({'error': 'created_at_to 格式不正确'}, status=status.HTTP_400_BAD_REQUEST)
|
|
queryset = queryset.filter(created_at__lte=dt_to)
|
|
|
|
paginator = self.pagination_class()
|
|
page = paginator.paginate_queryset(queryset.order_by('-created_at'), request, view=self)
|
|
serializer = PreSalesOrderSerializer(page, many=True)
|
|
return paginator.get_paginated_response(serializer.data)
|
|
|
|
def post(self, request):
|
|
if not self.check_employee_permission(request):
|
|
return self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
employee = request.user.employee
|
|
data = request.data or {}
|
|
|
|
try:
|
|
pre_sales_order = pre_order_services.create_pre_sales_order(
|
|
merchant=merchant,
|
|
customer_id=data.get('customer') or data.get('customer_id'),
|
|
warehouse_id=data.get('warehouse') or data.get('warehouse_id'),
|
|
operator=employee,
|
|
created_by=request.user,
|
|
kind=data.get('kind'),
|
|
items=data.get('items'),
|
|
remarks=data.get('remarks', ''),
|
|
)
|
|
except ValueError as exc:
|
|
return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
return Response(PreSalesOrderSerializer(pre_sales_order).data, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
class PreSalesOrderDetailView(StockChangeViewMixin, views.APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def _get_order(self, request, pk: int):
|
|
if not self.check_employee_permission(request):
|
|
return None, self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
try:
|
|
order = pre_order_services.get_pre_sales_order(
|
|
merchant=merchant,
|
|
pre_sales_order_id=pk,
|
|
)
|
|
return order, None
|
|
except ValueError:
|
|
return None, self.not_found_response('预销售单不存在')
|
|
|
|
def get(self, request, pk: int):
|
|
order, error_response = self._get_order(request, pk)
|
|
if error_response:
|
|
return error_response
|
|
return Response(PreSalesOrderSerializer(order).data, status=status.HTTP_200_OK)
|
|
|
|
def put(self, request, pk: int):
|
|
return self._update(request, pk)
|
|
|
|
def patch(self, request, pk: int):
|
|
return self._update(request, pk)
|
|
|
|
def _update(self, request, pk: int):
|
|
order, error_response = self._get_order(request, pk)
|
|
if error_response:
|
|
return error_response
|
|
|
|
data = request.data or {}
|
|
|
|
try:
|
|
updated = pre_order_services.update_pre_sales_order(
|
|
pre_sales_order=order,
|
|
customer_id=data.get('customer') or data.get('customer_id'),
|
|
warehouse_id=data.get('warehouse') or data.get('warehouse_id'),
|
|
kind=data.get('kind'),
|
|
items=data.get('items'),
|
|
remarks=data.get('remarks'),
|
|
)
|
|
except ValueError as exc:
|
|
return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
return Response(PreSalesOrderSerializer(updated).data, status=status.HTTP_200_OK)
|
|
|
|
def delete(self, request, pk: int):
|
|
order, error_response = self._get_order(request, pk)
|
|
if error_response:
|
|
return error_response
|
|
|
|
pre_order_services.delete_pre_sales_order(pre_sales_order=order)
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class PreSalesOrderConvertToSalesOrderView(StockChangeViewMixin, views.APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def post(self, request, pk: int):
|
|
if not self.check_employee_permission(request):
|
|
return self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
employee = request.user.employee
|
|
|
|
try:
|
|
sales_order = pre_order_services.convert_pre_sales_order_to_sales_order(
|
|
merchant=merchant,
|
|
pre_sales_order_id=pk,
|
|
operator=employee,
|
|
created_by=request.user,
|
|
)
|
|
except PermissionError as exc:
|
|
return Response({'error': str(exc)}, status=status.HTTP_403_FORBIDDEN)
|
|
except business_models.PreSalesOrder.DoesNotExist:
|
|
return self.not_found_response('预销售单不存在')
|
|
except ValueError as exc:
|
|
return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
return Response(
|
|
{
|
|
'id': sales_order.id,
|
|
'human_id': sales_order.human_id,
|
|
'status': sales_order.status,
|
|
'message': '预销售单已转换为销售单,等待审批',
|
|
},
|
|
status=status.HTTP_201_CREATED,
|
|
)
|
|
|
|
|
|
class PreSalesOrderItemAllocationView(StockChangeViewMixin, views.APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def _get_item(self, request, item_id: int):
|
|
if not self.check_employee_permission(request):
|
|
return None, self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
try:
|
|
item = business_models.PreSalesOrderItem.objects.select_related(
|
|
'pre_sales_order',
|
|
).get(id=item_id, pre_sales_order__merchant=merchant)
|
|
return item, None
|
|
except business_models.PreSalesOrderItem.DoesNotExist:
|
|
return None, self.not_found_response('预销售单明细不存在')
|
|
|
|
def post(self, request, item_id: int):
|
|
item, error_response = self._get_item(request, item_id)
|
|
if error_response:
|
|
return error_response
|
|
|
|
data = request.data or {}
|
|
try:
|
|
record = allocation_services.create_allocation_record(
|
|
item=item,
|
|
stock_ids=data.get('stock_ids') or [],
|
|
quantity=data.get('quantity'),
|
|
unit=data.get('unit'),
|
|
scanned_by=request.user.employee,
|
|
remarks=data.get('remarks'),
|
|
)
|
|
except ValueError as exc:
|
|
return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
return Response(AllocationRecordSerializer(record).data, status=status.HTTP_201_CREATED)
|
|
|
|
|
|
class PreSalesOrderItemAllocationDetailView(StockChangeViewMixin, views.APIView):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def delete(self, request, item_id: int, pk: int):
|
|
if not self.check_employee_permission(request):
|
|
return self.permission_error_response('无权限访问')
|
|
|
|
merchant = request.user.employee.merchant
|
|
try:
|
|
record = business_models.AllocationRecord.objects.select_related(
|
|
'pre_sales_order_item',
|
|
'pre_sales_order_item__pre_sales_order',
|
|
).get(
|
|
id=pk,
|
|
pre_sales_order_item_id=item_id,
|
|
pre_sales_order_item__pre_sales_order__merchant=merchant,
|
|
)
|
|
except business_models.AllocationRecord.DoesNotExist:
|
|
return self.not_found_response('配货记录不存在')
|
|
|
|
updated = allocation_services.cancel_allocation_record_with_unfreeze(
|
|
record=record,
|
|
cancelled_by=request.user,
|
|
)
|
|
return Response(AllocationRecordSerializer(updated).data, status=status.HTTP_200_OK)
|