summaryrefslogtreecommitdiff
path: root/examples/doh.py
blob: eff9ae7579fbac567902dfc9be6ebf919a2d117b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#!/usr/bin/env python3
#
# This is an example of sending DNS queries over HTTPS (DoH) with dnspython.
# Requires use of the requests module's Session object.
#
# See https://2.python-requests.org//en/latest/user/advanced/#session-objects
# for more details about Session objects
import requests

import dns.message
import dns.query
import dns.rdatatype


def main():
    where = '1.1.1.1'
    qname = 'example.com.'
    # one method is to use context manager, session will automatically close
    with requests.sessions.Session() as session:
        q = dns.message.make_query(qname, dns.rdatatype.A)
        r = dns.query.https(q, where, session=session)
        for answer in r.answer:
            print(answer)

        # ... do more lookups

    where = 'https://dns.google/dns-query'
    qname = 'example.net.'
    # second method, close session manually
    session = requests.sessions.Session()
    q = dns.message.make_query(qname, dns.rdatatype.A)
    r = dns.query.https(q, where, session=session)
    for answer in r.answer:
        print(answer)

    # ... do more lookups

    # close the session when you're done
    session.close()

if __name__ == '__main__':
    main()