How to remove extra data from JSONField in django rest framework?

Question

I have a model like this:

class Things(models.Model):
    data = models.JSONField(default=dict)

and structure of data is like this:

{
   "item" : "sugar",
   "quantity" : "2",
   "cost" : 220
}

Note that it is not mandatory for data to contain all 3 keys, all of them are optional.

The difficult part is that I don't want any other keys, apart from these 3 keys, to be present in data field.

Creating a serializer can help to ensure presence and format of these 3 fields but it will not ensure absence of other fields. How can I achieve this thing in django rest framework?

can you show me some examples of input and output to be expected? — binpy, Apr 15 '21 at 04:13
You can use *field level validator* to validate the incoming data — JPG, Apr 15 '21 at 04:19
create a validate_ function in the serializer. In your could would be `def validate_data(self, incoming_data)` — Kurohige, Apr 15 '21 at 08:18

score 0 · Answer 1 · answered Apr 15 '21 at 04:21

0

You can validate the field in your serializer:

class ThingsSerializer(serializers.ModelSerializer):
    class Meta:
        model = Things
        fields = ['data']

    def validate_data(self, value):
        # You can validate, clean and change your data
        final_data = {
            "item": value.get("item"),
            "quantity": value.get("quantity"),
            "cost": value.get("cost", 0)
        }
        return final_data

You can also create a separate serializer for your data field and pass the value to it, validate each field there and return the final data. But this should do if your fields are limited and simple.

answered Apr 15 '21 at 04:21

Navid Zarepak

4,148
1
12
26

What if one of the keys in ("item", "quantity", "cost") does not exist? – Puneet Garg Apr 15 '21 at 08:34
Also what if there are a large number of keys, let's say 100. – Puneet Garg Apr 15 '21 at 08:35
If a key doesn't exist, then the `.get()` returns a `None` so you can just check for that and return an error if you wish. If the number of keys is a lot, then instead of checking for them one by one, create a list or a tuple and put the allowed keys there, and use a for loop to check and validate them. – Navid Zarepak Apr 15 '21 at 10:34

score 0 · Answer 2 · answered Apr 15 '21 at 04:34

By using validate_<field_name> in serializer level, you can also create an logic to makesure that all fields inside the data is required to fill.

class ThingsSerializer(serializers.ModelSerializer):
     data = serializers.JSONField()
 
     class Meta:
         model = Things
         fields = ('data', )
 
     def validate_data(self, data):
         valid_keys = ('item', 'quantity', 'cost')
         errors = []
         out_data = {}
 
         for key in valid_keys:
             if key in data.keys():
                 if data.get(key):
                     out_data[key] = data.get(key)
                 else:
                     errors.append({key: 'This field should be filled.'})
             else:
                 errors.append({key: 'This field is required'})
 
         if len(errors) > 0:
             raise serializers.ValidationError(errors)
         return out_data

Here is the tests:

In [25]: s = ThingsSerializer(data={'data': {'item': 'sugar', 'quantity': 2, 'cost': 20}})

In [26]: s.is_valid()
Out[26]: True

In [27]: s.errors
Out[27]: {}



In [28]: s = ThingsSerializer(data={'data': {'item': 'sugar', 'quantity': 2}})

In [29]: s.is_valid()
Out[29]: False

In [30]: s.errors
Out[30]: {'data': [{'cost': ErrorDetail(string='This field is required', code='invalid')}]}



In [34]: s = ThingsSerializer(data={'data': {'item': 'sugar', 'quantity': 2, 'cost': None}})

In [35]: s.is_valid()
Out[35]: False

In [36]: s.errors
Out[36]: {'data': [{'cost': ErrorDetail(string='This field should be filled.', code='invalid')}]}

How to remove extra data from JSONField in django rest framework?

2 Answers2