mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
33 lines
900 B
Python
33 lines
900 B
Python
#
|
|
#
|
|
#
|
|
|
|
from idna import decode as _decode, encode as _encode
|
|
|
|
|
|
def idna_encode(name):
|
|
# Based on https://github.com/psf/requests/pull/3695/files
|
|
# #diff-0debbb2447ce5debf2872cb0e17b18babe3566e9d9900739e8581b355bd513f7R39
|
|
try:
|
|
name.encode('ascii')
|
|
# No utf8 chars, just use as-is
|
|
return name
|
|
except UnicodeEncodeError:
|
|
if name.startswith('*'):
|
|
# idna.encode doesn't like the *
|
|
name = _encode(name[2:]).decode('utf-8')
|
|
return f'*.{name}'
|
|
return _encode(name).decode('utf-8')
|
|
|
|
|
|
def idna_decode(name):
|
|
pieces = name.lower().split('.')
|
|
if any([p.startswith('xn--') for p in pieces]):
|
|
# it's idna
|
|
if name.startswith('*'):
|
|
# idna.decode doesn't like the *
|
|
return f'*.{_decode(name[2:])}'
|
|
return _decode(name)
|
|
# not idna, just return as-is
|
|
return name
|