From: Bob Halley <halley@dnspython.org>
Date: Sun, 30 Jun 2013 19:34:27 +0000 (-0700)
Subject: add set_flags()
X-Git-Tag: v1.11.0-py3~5
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=027e8fba185f26eb520054bf782d6ea459962113;p=thirdparty%2Fdnspython.git

add set_flags()
---

diff --git a/dns/resolver.py b/dns/resolver.py
index 9c4bce09..938ca939 100644
--- a/dns/resolver.py
+++ b/dns/resolver.py
@@ -459,6 +459,8 @@ class Resolver(object):
     @type ednsflags: int
     @ivar payload: The EDNS payload size.  The default is 0.
     @type payload: int
+    @ivar flags: The message flags to use.  The default is None (i.e. not overwritten)
+    @type flags: int
     @ivar cache: The cache to use.  The default is None.
     @type cache: dns.resolver.Cache object
     @ivar retry_servfail: should we retry a nameserver if it says SERVFAIL?
@@ -503,6 +505,7 @@ class Resolver(object):
         self.ednsflags = 0
         self.payload = 0
         self.cache = None
+        self.flags = None
         self.retry_servfail = False
 
     def read_resolv_conf(self, f):
@@ -795,6 +798,8 @@ class Resolver(object):
                 request.use_tsig(self.keyring, self.keyname,
                                  algorithm=self.keyalgorithm)
             request.use_edns(self.edns, self.ednsflags, self.payload)
+            if self.flags is not None:
+                request.flags = self.flags
             response = None
             #
             # make a copy of the servers list so we can alter it later.
@@ -935,6 +940,13 @@ class Resolver(object):
         self.ednsflags = ednsflags
         self.payload = payload
 
+    def set_flags(self, flags):
+        """Overrides the default flags with your own
+
+        @param flags: The flags to overwrite the default with
+        @type flags: int"""
+        self.flags = flags
+
 default_resolver = None
 
 def get_default_resolver():