Allow customizing dns.resolver.Resolver class attributes

ecederstrand · Oct 21, 2020 · 62d1af4 · 62d1af4
1 parent 17cc10e
commit 62d1af4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,7 @@ Change Log
 
 HEAD
 ----
+-   Allow overriding `dns.resolver.Resolver` class attributes via `Autodiscovery.DNS_RESOLVER_ATTRS`.
 
 
 3.3.0

diff --git a/exchangelib/autodiscover/discovery.py b/exchangelib/autodiscover/discovery.py
@@ -75,6 +75,9 @@ class Autodiscovery:
     INITIAL_RETRY_POLICY = FailFast()
     RETRY_WAIT = 10  # Seconds to wait before retry on connection errors
     MAX_REDIRECTS = 10  # Maximum number of URL redirects before we give up
+    DNS_RESOLVER_ATTRS = {
+        'timeout': AutodiscoverProtocol.TIMEOUT,
+    }
 
     def __init__(self, email, credentials=None, auth_type=None, retry_policy=None):
         """
@@ -149,7 +152,8 @@ def _cache_key(self):
     @threaded_cached_property
     def resolver(self):
         resolver = dns.resolver.Resolver()
-        resolver.timeout = AutodiscoverProtocol.TIMEOUT
+        for k, v in self.DNS_RESOLVER_ATTRS.items():
+            setattr(resolver, k, v)
         return resolver
 
     def _build_response(self, ad_response):