Merge pull request #215 from matclab/fix/214

Check for None before dereferencing
afewmail · Jan 2, 2019 · 9bc4da4 · 9bc4da4
2 parents 9488e7f + 787324d
commit 9bc4da4
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/afew/filters/DMARCReportInspectionFilter.py b/afew/filters/DMARCReportInspectionFilter.py
@@ -80,10 +80,15 @@ def read_auth_results(document):
     root = ET.fromstring(document)
     for record in root.findall('record'):
         auth_results = record.find('auth_results')
-        dkim = auth_results.find('dkim').find('result')
-        spf = auth_results.find('spf').find('result')
-        results['dkim'] &= not has_failed(dkim)
-        results['spf'] &= not has_failed(spf)
+        if auth_results:
+            dkim = auth_results.find('dkim')
+            if dkim:
+                dkim = dkim.find('result')
+                results['dkim'] &= not has_failed(dkim)
+            spf = auth_results.find('spf')
+            if spf:
+                spf = spf.find('result')
+                results['spf'] &= not has_failed(spf)
 
     return results