Update src/pe_source/dnstwistscript.py

Co-authored-by: dav3r <david.redmin@trio.dhs.gov>
cisagov · Dec 29, 2022 · 1706415 · 1706415
1 parent 9dc3e6b
commit 1706415
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pe_source/dnstwistscript.py b/src/pe_source/dnstwistscript.py
@@ -60,7 +60,7 @@ def checkBlocklist(dom, sub_domain_uid, source_uid, pe_org_uid, perm_list):
         # Check IP in DShield API
         try:
             results = json.loads(
-                dshield.ip(str(dom["dns_a"][0]), return_format=dshield.JSON)
+                dshield.ip(domain_a, return_format=dshield.JSON)
             )
             threats = results["ip"]["threatfeeds"]
             dshield_attacks = int(results["ip"].get("attacks", 0))