48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
import ipaddress
|
||
|
|
|
||
|
|
from django.core.exceptions import ValidationError
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
|
||
|
|
|
||
|
|
def clean_ipv6_address(
|
||
|
|
ip_str, unpack_ipv4=False, error_message=_("This is not a valid IPv6 address.")
|
||
|
|
):
|
||
|
|
"""
|
||
|
|
Clean an IPv6 address string.
|
||
|
|
|
||
|
|
Raise ValidationError if the address is invalid.
|
||
|
|
|
||
|
|
Replace the longest continuous zero-sequence with "::", remove leading
|
||
|
|
zeroes, and make sure all hextets are lowercase.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
ip_str: A valid IPv6 address.
|
||
|
|
unpack_ipv4: if an IPv4-mapped address is found,
|
||
|
|
return the plain IPv4 address (default=False).
|
||
|
|
error_message: An error message used in the ValidationError.
|
||
|
|
|
||
|
|
Return a compressed IPv6 address or the same value.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
addr = ipaddress.IPv6Address(int(ipaddress.IPv6Address(ip_str)))
|
||
|
|
except ValueError:
|
||
|
|
raise ValidationError(error_message, code="invalid")
|
||
|
|
|
||
|
|
if unpack_ipv4 and addr.ipv4_mapped:
|
||
|
|
return str(addr.ipv4_mapped)
|
||
|
|
elif addr.ipv4_mapped:
|
||
|
|
return "::ffff:%s" % str(addr.ipv4_mapped)
|
||
|
|
|
||
|
|
return str(addr)
|
||
|
|
|
||
|
|
|
||
|
|
def is_valid_ipv6_address(ip_str):
|
||
|
|
"""
|
||
|
|
Return whether or not the `ip_str` string is a valid IPv6 address.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
ipaddress.IPv6Address(ip_str)
|
||
|
|
except ValueError:
|
||
|
|
return False
|
||
|
|
return True
|