fix unsafe jinja use of __class__

hwhesselink · Mar 6, 2021 · 6942101 · 6942101
1 parent 814d0e8
commit 6942101
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/snmp/macros.jinja b/snmp/macros.jinja
@@ -4,14 +4,14 @@
 {%- macro v12c_communities(mode, proto='') -%}
   {% set communities = conf.get(mode+'communities'+proto, []) -%}
   {%- for community in communities %}
-    {%- if communities.__class__ == {}.__class__ and communities.get(community, {}) is  mapping %}
+    {%- if isinstance(communities, dict) and communities.get(community, {}) is  mapping %}
       {%- set source = communities.get(community).get('source', '') %}
       {%- set view = communities.get(community).get('view', None) %}
     {%- else %}
       {%- set source = '' %}
       {%- set view = '' %}
     {%- endif %}
-    {%- if not source.__class__ in (().__class__, [].__class__) %}
+    {%- if not isinstance(source, (tuple, list)) %}
       {%- set source = [source] %}
     {%- endif %}
     {%- for src in source -%}