Clean up with collections.defaultdict

validate.py

@@ -6,6 +6,8 @@
 import colorama
 import tldextract
 
+from collections import defaultdict
+
 # Use: ./validate.py old.json new.json
 KEYS = ['action_map', 'snitch_map', 'version']
 old_path = sys.argv[1]
@@ -53,26 +55,20 @@
 
 BLOCKED = ("block", "cookieblock")
 
-blocked_old = {}
+blocked_old = defaultdict(list)
 for domain in old_js['action_map'].keys():
     if old_js['action_map'][domain]['heuristicAction'] not in BLOCKED:
         continue
 
     base = extract(domain).registered_domain
-
-    if base not in blocked_old:
-        blocked_old[base] = []
     blocked_old[base].append(domain)
 
-blocked_new = {}
+blocked_new = defaultdict(list)
 for domain in new_js['action_map'].keys():
     if new_js['action_map'][domain]['heuristicAction'] not in BLOCKED:
         continue
 
     base = extract(domain).registered_domain
-
-    if base not in blocked_new:
-        blocked_new[base] = []
     blocked_new[base].append(domain)
 
 blocked_bases_old = set(blocked_old.keys())