rules: fix is_active function

StamusNetworks · Mar 12, 2018 · 88f31fe · 88f31fe
1 parent d753e7e
commit 88f31fe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rules/models.py b/rules/models.py
@@ -1391,6 +1391,8 @@ def parse_flowbits(self, source, flowbits, addition = False):
                             flowbits['added']['through_set'].append(through_elt)
 
     def is_active(self, ruleset):
+        SUPPRESSED = Transformation.SUPPRESSED
+        S_SUPPRESSED = Transformation.S_SUPPRESSED
         if self.state and self.category in ruleset.categories.all() and self not in ruleset.get_transformed_rules(key=SUPPRESSED, value=S_SUPPRESSED):
             return True
         return False