From ba1c70e86b93b6695a990e9a5df2333bab3e9efc Mon Sep 17 00:00:00 2001
From: Oliver <oliver.henry.walters@gmail.com>
Date: Sat, 26 Jun 2021 20:48:09 +1000
Subject: [PATCH] Intercept is_valid() method to set default values
---
InvenTree/InvenTree/serializers.py | 30 ++++++++++++++++++++++++++++++
1 file changed, 30 insertions(+)
diff --git a/InvenTree/InvenTree/serializers.py b/InvenTree/InvenTree/serializers.py
index fa7674723c..5be6d44c0c 100644
--- a/InvenTree/InvenTree/serializers.py
+++ b/InvenTree/InvenTree/serializers.py
@@ -7,6 +7,7 @@ Serializers used in various InvenTree apps
from __future__ import unicode_literals
from rest_framework import serializers
+from rest_framework.utils import model_meta
import os
@@ -39,6 +40,35 @@ class InvenTreeModelSerializer(serializers.ModelSerializer):
but also ensures that the underlying model class data are checked on validation.
"""
+ def is_valid(self, raise_exception=False):
+ """
+ Override the 'is_valid' method of the underlying ModelSerializer class.
+
+ - This is so we can intercept the data before save() is called.
+ - If we are creating a *new* model, replace any "missing" fields with the default values
+ - Default values are those specified by the database model
+ """
+
+ # This serializer is *not* associated with a model instance
+ # This means that we are trying to *create* a new model
+ if self.instance is None:
+ ModelClass = self.Meta.model
+
+ fields = model_meta.get_field_info(ModelClass)
+
+ for field_name, field in fields.fields.items():
+
+ # Check if the field has a default value
+ if field.has_default():
+
+ # Value not specified?
+ if field_name not in self.initial_data:
+
+ self.initial_data[field_name] = field.default
+
+
+ return super().is_valid(raise_exception=raise_exception)
+
def validate(self, data):
""" Perform serializer validation.
In addition to running validators on the serializer fields,