ChiPy Algorithms SIGDescription
Write a function to check whether an input string is a valid IPv4 address or IPv6 address or neither. An IPv4 address is for example 172.16.254.1. The numbers can be from 0 to 255 and there must be no leading zeros (e.g. 01).
IPv6 addresses are represented as eight groups of four hexadecimal digits, each group representing 16 bits. The groups are separated by colons (“:”). For example, the address 2001:0db8:85a3:0000:0000:8a2e:0370:7334 is a valid one. Also, we could omit some leading zeros among four hexadecimal digits and some low-case characters in the address to upper-case ones, so 2001:db8:85a3:0:0:8A2E:0370:7334.
Input: IP = "172.16.254.1"
Output: "IPv4"
Explanation: This is a valid IPv4 address, return "IPv4".
Input: IP = "256.256.256.256"
Output: "Neither"
Explanation: This is neither a IPv4 address nor a IPv6 address.
Solution
def is_ipv4(address):
splitted = address.split('.')
if len(splitted) != 4:
return False
for part in splitted:
try:
num = int(part)
except:
return False
if num < 0:
return False
if num > 255:
return False
if str(num) != part: # str('01')
return False
return True
def is_ipv6(address):
splitted = address.split(':')
if len(splitted) != 8:
return False
for part in splitted:
if len(part) > 4:
return False
try:
int(part, 16)
except:
return False
return True
class Solution:
def validIPAddress(self, IP: str) -> str:
if is_ipv4(IP):
return 'IPv4'
elif is_ipv6(IP):
return 'IPv6'
else:
return 'Neither'