feat: mission category

basic_info/admin.py

@@ -194,3 +194,33 @@ class VehicleTransportRecordAdmin(AdminBase):
     list_display = ('driver_name', 'contact_number', 'delivery_date')
     search_fields = ('driver_name', 'contact_number')
     list_filter = ('vehicle_type',)
+
+
+class TransportVehicleMaterialCapacityInline(admin.TabularInline):
+    model = models.TransportVehicleMaterialCapacity
+    extra = 0
+    fields = ('material_name', 'capacity')
+
+
+@admin.register(models.TransportVehicle)
+class TransportVehicleAdmin(AdminBase):
+    inlines = [TransportVehicleMaterialCapacityInline]
+    list_display = ('name', 'license_plate')
+    search_fields = ('name', 'license_plate')
+
+    def save_formset(self, request, form, formset, change):
+        instances = formset.save(commit=False)
+        for obj in formset.deleted_objects:
+            obj.delete()
+        for instance in instances:
+            if isinstance(instance, models.TransportVehicleMaterialCapacity):
+                instance.merchant = form.instance.merchant
+            instance.save()
+        formset.save_m2m()
+
+
+@admin.register(models.TransportVehicleMaterialCapacity)
+class TransportVehicleMaterialCapacityAdmin(AdminBase):
+    list_display = ('transport_vehicle', 'material_name', 'capacity')
+    search_fields = ('transport_vehicle__name', 'transport_vehicle__license_plate', 'material_name')
+    list_filter = ('transport_vehicle',)