doctests for utils.anonymize(remote_addr)

This commit is contained in:
Martin Zimmermann 2013-10-27 13:58:50 +01:00
parent 527b349fbe
commit 89b1ca8846

View File

@ -11,14 +11,23 @@ import ipaddress
def anonymize(remote_addr): def anonymize(remote_addr):
"""
Anonymize IPv4 and IPv6 :param remote_addr: to /24 (zero'd)
and /48 (zero'd).
>>> anonymize(u'12.34.56.78')
u'12.34.56.0'
>>> anonymize(u'1234:5678:90ab:cdef:fedc:ba09:8765:4321')
u'1234:5678:90ab:0000:0000:0000:0000:0000'
"""
try: try:
ipv4 = ipaddress.IPv4Address(remote_addr) ipv4 = ipaddress.IPv4Address(remote_addr)
return ''.join(ipv4.exploded.rsplit('.', 1)[0]) + '.' + '0' return u''.join(ipv4.exploded.rsplit('.', 1)[0]) + '.' + '0'
except ipaddress.AddressValueError: except ipaddress.AddressValueError:
ipv6 = ipaddress.IPv6Address(remote_addr) ipv6 = ipaddress.IPv6Address(remote_addr)
if ipv6.ipv4_mapped is not None: if ipv6.ipv4_mapped is not None:
return anonymize(ipv6.ipv4_mapped) return anonymize(ipv6.ipv4_mapped)
return ipv6.exploded.rsplit(':', 5)[0] + ':' + ':'.join(['0000']*3) return u'' + ipv6.exploded.rsplit(':', 5)[0] + ':' + ':'.join(['0000']*5)
def salt(value, s=u'\x082@t9*\x17\xad\xc1\x1c\xa5\x98'): def salt(value, s=u'\x082@t9*\x17\xad\xc1\x1c\xa5\x98'):