From 6e4298873fe86346187101071e823b323e3bd4ed Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Wed, 20 Mar 2024 17:39:44 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E5=91=A8=E9=A2=84?=
 =?UTF-8?q?=E4=BC=B0=E6=B6=88=E8=80=97=E9=87=8F=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../0026_material_week_esitimate_consume.py    | 18 ++++++++++++++++++
 apps/mtm/models.py                             |  2 ++
 apps/mtm/views.py                              | 13 +++++++++++++
 3 files changed, 33 insertions(+)
 create mode 100644 apps/mtm/migrations/0026_material_week_esitimate_consume.py

diff --git a/apps/mtm/migrations/0026_material_week_esitimate_consume.py b/apps/mtm/migrations/0026_material_week_esitimate_consume.py
new file mode 100644
index 00000000..e663acd7
--- /dev/null
+++ b/apps/mtm/migrations/0026_material_week_esitimate_consume.py
@@ -0,0 +1,18 @@
+# Generated by Django 3.2.12 on 2024-03-20 09:35
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mtm', '0025_auto_20231120_1139'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='material',
+            name='week_esitimate_consume',
+            field=models.PositiveIntegerField(blank=True, null=True, verbose_name='周消耗预估'),
+        ),
+    ]
diff --git a/apps/mtm/models.py b/apps/mtm/models.py
index 60b2e271..3d0af08c 100644
--- a/apps/mtm/models.py
+++ b/apps/mtm/models.py
@@ -57,6 +57,8 @@ class Material(CommonAModel):
     unit = models.CharField('基准计量单位', default='个', max_length=10)
     count = models.PositiveIntegerField('物料库存总数', default=0)
     count_safe = models.PositiveIntegerField('安全库存总数', null=True, blank=True)
+    week_esitimate_consume = models.PositiveIntegerField(
+        '周消耗预估', null=True, blank=True)
     process = models.ForeignKey(
         Process, verbose_name='所用工序', on_delete=models.CASCADE, null=True, blank=True)
     parent = models.ForeignKey(
diff --git a/apps/mtm/views.py b/apps/mtm/views.py
index 61b05a92..21aeda45 100644
--- a/apps/mtm/views.py
+++ b/apps/mtm/views.py
@@ -44,6 +44,19 @@ class MaterialViewSet(CustomModelViewSet):
         daoru_material(settings.BASE_DIR + request.data.get('path', ''))
         return Response()
 
+    @action(methods=['put'], detail=True, serializer_class=Serializer, perms_map={'put': '*'})
+    @transaction.atomic
+    def set_week_esitimate_consume(self, request, *args, **kwargs):
+        """设置周预估消耗量
+
+        设置周预估消耗量
+        """
+        obj: Material = self.get_object()
+        obj.week_esitimate_consume = request.data.get(
+            'week_esitimate_consume', None)
+        obj.save()
+        return Response()
+
 
 class ShiftViewSet(ListModelMixin, CustomGenericViewSet):
     """