Code formatting

Author: seanthegeek

checkdmarc/spf.py

@@ -616,13 +616,15 @@ def parse_spf_record(
                 seen.append(value.lower())
 
                 if "%{" in value:
-                    include = OrderedDict([
-                        ("action", action),
-                        ("mechanism", mechanism),
-                        ("value", value),
-                        ("dns_lookups", mechanism_dns_lookups),
-                        ("void_dns_lookups", mechanism_void_dns_lookups),
-                    ])
+                    include = OrderedDict(
+                        [
+                            ("action", action),
+                            ("mechanism", mechanism),
+                            ("value", value),
+                            ("dns_lookups", mechanism_dns_lookups),
+                            ("void_dns_lookups", mechanism_void_dns_lookups),
+                        ]
+                    )
                     parsed["mechanisms"].append(include)
                     continue
                 try:

tests.py

@@ -307,13 +307,13 @@ def testSPFMXMechanism(self):
         results = checkdmarc.spf.parse_spf_record(spf_record, domain)
         for mechanism in results["parsed"]["mechanisms"]:
             if mechanism["mechanism"] == "mx":
-                self.assertTrue(len(mechanism["hosts"] ) > 0 )
+                self.assertTrue(len(mechanism["hosts"]) > 0)
             for host in mechanism["hosts"]:
                 self.assertTrue(len(host) > 0)
-    
+
     def testSPFMacrosExists(self):
         """SPF macros can be used with the exists mechanism"""
-        record ="v=spf1 exists:exists:%{i}.spf.hc0000-xx.iphmx.com ~all"
+        record = "v=spf1 exists:exists:%{i}.spf.hc0000-xx.iphmx.com ~all"
         domain = "example.com"
         results = checkdmarc.spf.parse_spf_record(record, domain)
         self.assertTrue(len(results["parsed"]["mechanisms"]) > 0)
@@ -325,15 +325,14 @@ def testSPFMacrosInclude(self):
         results = checkdmarc.spf.parse_spf_record(record, domain)
         self.assertTrue(len(results["parsed"]["mechanisms"]) > 0)
 
-
     def testSPFAMechanism(self):
         """Addresses are included in the output for SPF records with an a lookup"""
         spf_record = "v=spf1 a ~all"
         domain = "example.com"
         results = checkdmarc.spf.parse_spf_record(spf_record, domain)
         for mechanism in results["parsed"]["mechanisms"]:
             if mechanism["mechanism"] == "a":
-                self.assertTrue(len(mechanism["addresses"] ) > 0 )
+                self.assertTrue(len(mechanism["addresses"]) > 0)
 
     @unittest.skipUnless(os.path.exists("/etc/resolv.conf"), "no network")
     def testDMARCPctLessThan100Warning(self):