From 76194239c3d4f64cceff7d9a0161aaf777d4cf18 Mon Sep 17 00:00:00 2001
From: Roberto Bochet <robertobochet@gmail.com>
Date: Fri, 24 Dec 2021 23:47:32 +0100
Subject: [PATCH] Added support for AXFR query

---
 pydig/query_type.py | 1 +
 pydig/resolver.py   | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/pydig/query_type.py b/pydig/query_type.py
index c17e441..d871ffc 100644
--- a/pydig/query_type.py
+++ b/pydig/query_type.py
@@ -29,6 +29,7 @@ class QueryType(Enum):
     DNSKEY = 48
     CDS = 59
     CDNSKEY = 60
+    AXFR = 252
     CAA = 257
 
     @classmethod
diff --git a/pydig/resolver.py b/pydig/resolver.py
index 18e6d6d..71467c1 100644
--- a/pydig/resolver.py
+++ b/pydig/resolver.py
@@ -60,7 +60,7 @@ def _args(self, domain, query_type):
         yield query_type.name
 
         # We only care about the result
-        yield '+short'
+        yield from ('+noall', '+answer') if query_type == query_type.AXFR else ('+short',)
 
         # Add in any additional args
         yield from self.additional_args