adding env filter

MozillaFoundation · Oct 1, 2018 · 64f93ae · 64f93ae
1 parent 6bcebf0
commit 64f93ae
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/network-api/networkapi/buyersguide/templates/product_page.html b/network-api/networkapi/buyersguide/templates/product_page.html
@@ -3,13 +3,13 @@
 {% load criterion %}
 {% load product-update %}
 {% load yes_no %}
+{% load env %}
 
 {% block body-id %}TODO-CATEGORY{% endblock %}
 
 {% block guts %}
 <div class="container-fluid text-center product-header bg-gray">
-
-  <img class="mb-0" src="{{mediaUrl}}{{AWS_LOCATION}}{{product.image}}" width="500"/>
+  <img class="mb-0" src="{{mediaUrl}}{{"AWS_LOCATION"|env}}{{product.image}}" width="500"/>
 </div>
 <div class="container">
 

diff --git a/network-api/networkapi/buyersguide/templatetags/env.py b/network-api/networkapi/buyersguide/templatetags/env.py
@@ -0,0 +1,12 @@
+from django import template
+from django.conf import settings
+
+register = template.Library()
+
+
+@register.filter
+def env(value):
+    if hasattr(settings, value):
+        return getattr(settings, value)
+    else:
+        return ""