From 1ea008805a4508db632011f9549d438234ac3a10 Mon Sep 17 00:00:00 2001
From: Czw <459749926@qq.com>
Date: Fri, 12 Nov 2021 15:42:40 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BB=93=E5=BA=93=E9=94=81=E5=AE=9A/?=
 =?UTF-8?q?=E8=A7=A3=E9=94=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/data/views.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/apps/data/views.py b/apps/data/views.py
index 56fb505..92c408e 100644
--- a/apps/data/views.py
+++ b/apps/data/views.py
@@ -43,6 +43,30 @@ class WarehouseViewSet(BaseViewSet, ReadWriteMixin):
         number = Warehouse.get_number(self.team)
         return Response(data={'number': number}, status=status.HTTP_200_OK)
 
+    @extend_schema(responses={200: WarehouseSerializer})
+    @action(detail=True, methods=['post'])
+    def lock(self, request, *args, **kwargs):
+        """锁定仓库"""
+
+        warehouse = self.get_object()
+        warehouse.is_locked = False
+        warehouse.save(update_fields=['is_locked'])
+
+        serializer = WarehouseSerializer(instance=warehouse)
+        return Response(data=serializer.data, status=status.HTTP_200_OK)
+
+    @extend_schema(responses={200: WarehouseSerializer})
+    @action(detail=True, methods=['post'])
+    def unlock(self, request, *args, **kwargs):
+        """解锁仓库"""
+
+        warehouse = self.get_object()
+        warehouse.is_locked = True
+        warehouse.save(update_fields=['is_locked'])
+
+        serializer = WarehouseSerializer(instance=warehouse)
+        return Response(data=serializer.data, status=status.HTTP_200_OK)
+
 
 class ClientViewSet(BaseViewSet, ReadWriteMixin):
     """客户"""