feat: material增加count_mb字段

apps/inm/services.py

@@ -45,15 +45,16 @@ class InmService:
 
     @classmethod
     def cal_mat_count(cls, material: Material):
-        material_count = MaterialBatch.objects.filter(material=material).aggregate(total=Sum("count"))["total"]
+        mb_count = MaterialBatch.objects.filter(material=material).aggregate(total=Sum("count"))["total"]
         from apps.wpm.models import WMaterial
-
         wm_count = WMaterial.objects.filter(material=material).aggregate(total=Sum("count"))["total"]
-        if material_count is None:
-            material_count = 0
+        if mb_count is None:
+            mb_count = 0
         if wm_count is None:
             wm_count = 0
-        Material.objects.filter(id=material.id).update(count=material_count + wm_count)
+        Material.objects.filter(id=material.id).update(
+            count_mb=mb_count,
+            count=mb_count + wm_count)
 
     @classmethod
     def update_mb(cls, instance: MIO, in_or_out: int = 1):

apps/mtm/migrations/0029_auto_20240528_1428.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.2.12 on 2024-05-28 06:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mtm', '0028_shift_sort'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='material',
+            name='count_mb',
+            field=models.PositiveIntegerField(default=0, verbose_name='仓库总数'),
+        ),
+        migrations.AlterField(
+            model_name='material',
+            name='count',
+            field=models.PositiveIntegerField(default=0, verbose_name='总库存'),
+        ),
+    ]

apps/mtm/models.py

@@ -55,7 +55,8 @@ class Material(CommonAModel):
     testitems = models.JSONField('检测项目', default=list, blank=True)
     sort = models.PositiveSmallIntegerField('排序', default=1)
     unit = models.CharField('基准计量单位', default='个', max_length=10)
-    count = models.PositiveIntegerField('物料库存总数', default=0)
+    count = models.PositiveIntegerField('总库存', default=0)
+    count_mb = models.PositiveIntegerField('仓库总数', default=0)
     count_safe = models.PositiveIntegerField('安全库存总数', null=True, blank=True)
     week_esitimate_consume = models.PositiveIntegerField(
         '周消耗预估', null=True, blank=True)