fix: org record bug

This commit is contained in:
Matt Keeley
2023-04-05 15:33:47 -07:00
parent c7a8897bf0
commit 1767184533

View File

@@ -2,6 +2,9 @@ import dns.resolver, tldextract
def get_dmarc_record(domain, dns_server):
"""Returns the DMARC record for a given domain."""
subdomain = tldextract.extract(domain).registered_domain
if subdomain != domain:
return get_dmarc_record(subdomain, dns_server)
try:
resolver = dns.resolver.Resolver()
resolver.nameservers = [dns_server]