Browse Source

Fix bug left over from work on #3569

Jeremy Stretch 6 years ago
parent
commit
590bbbce7f
1 changed files with 2 additions and 2 deletions
  1. 2 2
      netbox/extras/models.py

+ 2 - 2
netbox/extras/models.py

@@ -506,11 +506,11 @@ class ExportTemplate(models.Model):
             'queryset': queryset
             'queryset': queryset
         }
         }
 
 
-        if self.template_language == TEMPLATE_LANGUAGE_DJANGO:
+        if self.template_language == ExportTemplateLanguageChoices.LANGUAGE_DJANGO:
             template = Template(self.template_code)
             template = Template(self.template_code)
             output = template.render(Context(context))
             output = template.render(Context(context))
 
 
-        elif self.template_language == TEMPLATE_LANGUAGE_JINJA2:
+        elif self.template_language == ExportTemplateLanguageChoices.LANGUAGE_JINJA2:
             template = Environment().from_string(source=self.template_code)
             template = Environment().from_string(source=self.template_code)
             output = template.render(**context)
             output = template.render(**context)