nex/nonsensebot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Catch more types of query errors

Browse source
This commit is contained in:
Nexus 2024-09-18 18:15:55 +01:00
parent 2c0e61b831
commit 9f0ac830f2
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
app/modules/drill.py
View file

@ -10,6 +10,7 @@ import dns.asyncquery
import dns.nameserver import dns.nameserver
import dns.resolver import dns.resolver
import dns.message import dns.message
from dns.quic._common import UnexpectedEOF
from urllib.parse import urlparse from urllib.parse import urlparse
@ -116,7 +117,7 @@ class DNSCommand(niobot.Module):
end_ts = time.perf_counter() end_ts = time.perf_counter()
if isinstance(response, dns.message.Message): if isinstance(response, dns.message.Message):
response = (response, False) response = (response, False)
except dns.exception.DNSException as err: except (dns.exception.DNSException, UnexpectedEOF, httpx.HTTPError, ValueError) as err:
return await m.edit( return await m.edit(
"There was an error while querying the DNS server:\n```\n%s\n```" % err "There was an error while querying the DNS server:\n```\n%s\n```" % err
) )