make sending queries via IPv6 work

Original author: Bob Halley <halley@dnspython.org>
Date: 2004-08-04 09:10:14

1 files changed, 18 insertions, 0 deletions

diff --git a/dns/inet.py b/dns/inet.py
index e8e3092..c12dbbf 100644
--- a/dns/inet.py
+++ b/dns/inet.py
@@ -70,3 +70,21 @@ def inet_ntop(family, address):
         return dns.ipv6.inet_ntoa(address)
     else:
         raise NotImplementedError
+
+def af_for_address(text):
+    """Determine the address family of a textual-form network address.
+
+    @param text: the textual address
+    @type text: string
+    @raises ValueError: the address family cannot be determined from the input.
+    @rtype int
+    """
+    try:
+        junk = dns.ipv4.inet_aton(text)
+        return AF_INET
+    except:
+        try:
+            junk = dns.ipv6.inet_aton(text)
+            return AF_INET6
+        except:
+            raise ValueError