NIR (#67) fix for py 2.x ipaddr calls

secynic · Jul 9, 2016 · 07bc367 · 07bc367
1 parent 340fc22
commit 07bc367
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/ipwhois/nir.py b/ipwhois/nir.py
@@ -293,8 +293,18 @@ def _get_nets_jpnic(self, response):
 
                 net = copy.deepcopy(BASE_NET)
                 tmp = ip_network(match.group(2))
+                try:
+                    network_address = tmp.network_address
+                except AttributeError:
+                    network_address = tmp.ip
+
+                try:
+                    broadcast_address = tmp.broadcast_address
+                except AttributeError:
+                    broadcast_address = tmp.broadcast
+
                 net['range'] = '{0} - {1}'.format(
-                    tmp.network_address + 1, tmp.broadcast_address
+                    network_address + 1, broadcast_address
                 )
 
                 cidr = ip_network(match.group(2).strip()).__str__()