django import-export, export multiple many to many models

Question

I have Rule Model which has multiple RuleCondtion and RuleAction. I want to export these into a csv file. I am using django import-export for this.

Example:

name, priority, tags, conditions, actions
Rule 1, 1000, "tag1,tag2", "[{"identifier":"A", "operator":"B"..}]", "[{"identifier":"A", "operator":"B"..}]"

My Models:

    class Rule(models.Model):
        name = models.CharField(max_length=128, help_text="Name of Rule")
        description = models.TextField(help_text="Brief Description of Rule", blank=True)
        priority = models.IntegerField(default=1000, help_text="Priority of rule, lesser applies first")
        created_at = models.DateTimeField(auto_now_add=True)
        updated_at = models.DateTimeField(auto_now=True)
        tags = models.ManyToManyField('Tag', blank=True)
        disabled = models.BooleanField(default=True)

        def __str__(self):
            return self.name

    class RuleCondition(models.Model):
        identifier = models.CharField(max_length=128, help_text="Select a Property", blank=True)
        operator = models.CharField(max_length=128, help_text="Select an Operator", blank=True, choices=CONDITION_OPERATOR_CHOICES)
        value = models.TextField(help_text="Content to match the rule")
        rule = models.ForeignKey('Rule', on_delete=models.CASCADE, related_name='conditions')
        created_at = models.DateTimeField(auto_now_add=True)
        updated_at = models.DateTimeField(auto_now=True)

        def __str__(self):
            return 'Rule Condition ' + str(self.id)

    class RuleAction(models.Model):
        identifier = models.CharField(max_length=128, help_text="Select a Property", blank=True)
        operator = models.CharField(max_length=128, help_text="Select an Operator", blank=True, choices=ACTION_OPERATOR_CHOICES)
        value = models.TextField(help_text="Content to apply on the rule")
        rule = models.ForeignKey('Rule', on_delete=models.CASCADE, related_name='actions')
        created_at = models.DateTimeField(auto_now_add=True)
        updated_at = models.DateTimeField(auto_now=True)

        def __str__(self):
            return 'Rule Action ' + str(self.id)

How can I achieve this, there is no option in the django import-export to do this.

Check out this https://stackoverflow.com/questions/18685223/how-to-export-django-model-data-into-csv-file — Rohit Sharma, Aug 09 '19 at 06:54

score 3 · Accepted Answer · answered Aug 09 '19 at 07:08

Figured it out. Here is the code. Override the internal function of django-import-export.

import json
from import_export import resources, fields
from django.core import serializers
from .models import Rule, Tag, RuleCondition, RuleAction
from import_export.widgets import JSONWidget, ManyToManyWidget, ForeignKeyWidget

class RuleOperationsWidget(ManyToManyWidget):

    def render(self, value, obj=None):
        return json.dumps(
            list(value.values('identifier', 'operator', 'value')),
        )

class RuleResource(resources.ModelResource):
    tags = fields.Field(
        attribute='tags',
        widget=ManyToManyWidget(model=Tag, separator=',', field='name'),
    )

    conditions = fields.Field(
        attribute='conditions',
        widget=RuleOperationsWidget(model=RuleCondition),
    )

    actions = fields.Field(
        attribute='actions',
        widget=RuleOperationsWidget(model=RuleAction),
    )

    class Meta:
        model = Rule
        exclude = ('created_at', 'updated_at', 'id',)

django import-export, export multiple many to many models

1 Answers1