Skip to content

Commit

Permalink
Allow injection of context data for template rendered by import_actio…
Browse files Browse the repository at this point in the history
…n() and export_action()
  • Loading branch information
thauk-copperleaf committed Nov 15, 2016
1 parent c7e00b4 commit b84a770
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions import_export/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,12 @@ def add_success_message(self, result, request):

messages.success(request, success_message)

def get_import_context_data(self, **kwargs):
return self.get_context_data(**kwargs)

def get_context_data(self, **kwargs):
return {}

def import_action(self, request, *args, **kwargs):
'''
Perform a dry_run of the import to make sure the import will not
Expand All @@ -228,7 +234,7 @@ def import_action(self, request, *args, **kwargs):
'''
resource = self.get_import_resource_class()(**self.get_import_resource_kwargs(request, *args, **kwargs))

context = {}
context = self.get_import_context_data()

import_formats = self.get_import_formats()
form = ImportForm(import_formats,
Expand Down Expand Up @@ -378,6 +384,12 @@ def get_export_data(self, file_format, queryset, *args, **kwargs):
export_data = file_format.export_data(data)
return export_data

def get_export_context_data(self, **kwargs):
return self.get_context_data(**kwargs)

def get_context_data(self, **kwargs):
return {}

def export_action(self, request, *args, **kwargs):
formats = self.get_export_formats()
form = ExportForm(formats, request.POST or None)
Expand All @@ -401,7 +413,7 @@ def export_action(self, request, *args, **kwargs):
post_export.send(sender=None, model=self.model)
return response

context = {}
context = self.get_export_context_data()

if django.VERSION >= (1, 8, 0):
context.update(self.admin_site.each_context(request))
Expand Down

0 comments on commit b84a770

Please sign in to comment.