Need to convert to a boolean

bluesentry · Sep 9, 2019 · ecb2f84 · ecb2f84
1 parent 367eff0
commit ecb2f84
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scan.py b/scan.py
@@ -119,15 +119,16 @@ def sns_start_scan(s3_object):
         MessageStructure="json"
     )
 
+
 def sns_scan_results(s3_object, result):
     # Don't publish if SNS ARN has been supplied
     if AV_STATUS_SNS_ARN is None:
         return
     # Don't publish if result is CLEAN and CLEAN results should not be published
-    if result == AV_STATUS_CLEAN and not AV_STATUS_SNS_PUBLISH_CLEAN:
+    if result == AV_STATUS_CLEAN and not str_to_bool(AV_STATUS_SNS_PUBLISH_CLEAN):
         return
     # Don't publish if result is INFECTED and INFECTED results should not be published
-    if result == AV_STATUS_INFECTED and not AV_STATUS_SNS_PUBLISH_INFECTED:
+    if result == AV_STATUS_INFECTED and not str_to_bool(AV_STATUS_SNS_PUBLISH_INFECTED):
         return
     message = {
         "bucket": s3_object.bucket_name,