from rest_framework import viewsets from utils.page import MyPageNumberPagination from utils.md5 import Md5 from rest_framework.filters import OrderingFilter from django_filters.rest_framework import DjangoFilterBackend from rest_framework.response import Response from rest_framework.exceptions import APIException from django.http import StreamingHttpResponse from stock.models import StockshelfModel as shelflist from . import serializers from .filter import StockBinFilter,StockshelfFilter from .files import FileListRenderCN, FileListRenderEN, FileBinListRenderCN, FileBinListRenderEN from rest_framework.settings import api_settings from django.db import transaction from rest_framework import status class stockshelfViewSet(viewsets.ModelViewSet): filter_backends = [DjangoFilterBackend, OrderingFilter, ] ordering_fields = ['id', "create_time", "update_time", ] filter_class = StockshelfFilter pagination_class = MyPageNumberPagination """ list: Response a data list（all） post: Create a new data（create） """ def get_project(self): try: id = self.kwargs.get('pk') return id except: return None def get_queryset(self): id = self.get_project() if self.request.user: if id is None: return shelflist.objects.filter() else: return shelflist.objects.filter(id=id) else: return shelflist.objects.none() def get_serializer_class(self): if self.action in ['list', 'create', ]: return serializers.stockShelfSerializer elif self.action in ['retrieve','update',]: return serializers.stockShelfpartialSerializer else: return self.http_method_not_allowed(request=self.request) def create(self, request, *args, **kwargs): data = request.data.copy() warehouse_code = data.get('warehouse_code') warehouse_name = data.get('warehouse_name') shelf_type = data.get('shelf_type') shelf_name = data.get('shelf_name') rows = int(data.get('rows', 0)) cols = int(data.get('cols', 0)) layers = int(data.get('layers', 0)) existing_positions = shelflist.get_existing_positions(warehouse_code,shelf_name, rows, cols, layers) # 生成所有可能的坐标 instances = [] for r in range(1, rows+1): for c in range(1, cols+1): for l in range(1, layers+1): if (r, c, l) not in existing_positions : instances.append(shelflist( warehouse_code=warehouse_code, warehouse_name=warehouse_name, shelf_type=shelf_type, shelf_name=shelf_name, row=r, col=c, layer=l, )) try: with transaction.atomic(): shelflist.objects.bulk_create(instances, batch_size=1000) except Exception as e: return Response( {"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR ) return Response({ "created": len(instances), "skipped": (rows*cols*layers) - len(instances) }, status=status.HTTP_201_CREATED) def update(self, request, pk): qs = self.get_object() print(qs.id) # return Response({"detail": "暂不支持修改"}, status=status.HTTP_400_BAD_REQUEST) data = self.request.data print(data) serializer = self.get_serializer(qs, data=data) serializer.is_valid(raise_exception=True) serializer.save() headers = self.get_success_headers(serializer.data) return Response(serializer.data, status=200, headers=headers) def retrieve(self, request, pk): qs = self.get_object() serializer = self.get_serializer(qs) return Response(serializer.data) class StockBinViewSet(viewsets.ModelViewSet): filter_backends = [DjangoFilterBackend, OrderingFilter, ] ordering_fields = ['id', "create_time", "update_time", ] filter_class = StockBinFilter pagination_class = MyPageNumberPagination """ list: Response a data list（all） post: Create a new data（create） """ def get_project(self): try: id = self.kwargs.get('pk') return id except: return None def get_queryset(self): id = self.get_project() if self.request.user: if id is None: return shelflist.objects.filter() else: return shelflist.objects.filter(id=id) else: return shelflist.objects.none() def get_serializer_class(self): if self.action in ['list', 'create', ]: return serializers.stocklistSerializer # elif self.action in ['retrieve','update',]: # return serializers.stocklistpartialSerializer else: return self.http_method_not_allowed(request=self.request) # def create(self, request, *args, **kwargs): # data = request.data.copy() # warehouse_code = data.get('warehouse_code') # warehouse_name = data.get('warehouse_name') # shelf_type = data.get('shelf_type') # shelf_name = data.get('shelf_name') # rows = int(data.get('rows', 0)) # cols = int(data.get('cols', 0)) # layers = int(data.get('layers', 0)) # existing_positions = shelflist.get_existing_positions(warehouse_code,shelf_name, rows, cols, layers) # # 生成所有可能的坐标 # instances = [] # for r in range(1, rows+1): # for c in range(1, cols+1): # for l in range(1, layers+1): # if (r, c, l) not in existing_positions : # instances.append(shelflist( # warehouse_code=warehouse_code, # warehouse_name=warehouse_name, # shelf_type=shelf_type, # shelf_name=shelf_name, # row=r, # col=c, # layer=l, # )) # try: # with transaction.atomic(): # shelflist.objects.bulk_create(instances, batch_size=1000) # except Exception as e: # return Response( # {"error": str(e)}, # status=status.HTTP_500_INTERNAL_SERVER_ERROR # ) # return Response({ # "created": len(instances), # "skipped": (rows*cols*layers) - len(instances) # }, status=status.HTTP_201_CREATED) # def update(self, request, pk): # qs = self.get_object() # print(qs.id) # # return Response({"detail": "暂不支持修改"}, status=status.HTTP_400_BAD_REQUEST) # data = self.request.data # print(data) # serializer = self.get_serializer(qs, data=data) # serializer.is_valid(raise_exception=True) # serializer.save() # headers = self.get_success_headers(serializer.data) # return Response(serializer.data, status=200, headers=headers) # def retrieve(self, request, pk): # qs = self.get_object() # serializer = self.get_serializer(qs) # return Response(serializer.data)