9bbeb35c08
Fixed an issue that prevented the password reset tokens from working. Added email templates for password reset success and new account creation. Added more dynamic email template support.
122 lines
3.3 KiB
Python
122 lines
3.3 KiB
Python
import codecs
|
|
import re
|
|
from typing import Any, Optional, Tuple
|
|
|
|
from .core import IDNAError, alabel, decode, encode, ulabel
|
|
|
|
_unicode_dots_re = re.compile("[\u002e\u3002\uff0e\uff61]")
|
|
|
|
|
|
class Codec(codecs.Codec):
|
|
def encode(self, data: str, errors: str = "strict") -> Tuple[bytes, int]:
|
|
if errors != "strict":
|
|
raise IDNAError('Unsupported error handling "{}"'.format(errors))
|
|
|
|
if not data:
|
|
return b"", 0
|
|
|
|
return encode(data), len(data)
|
|
|
|
def decode(self, data: bytes, errors: str = "strict") -> Tuple[str, int]:
|
|
if errors != "strict":
|
|
raise IDNAError('Unsupported error handling "{}"'.format(errors))
|
|
|
|
if not data:
|
|
return "", 0
|
|
|
|
return decode(data), len(data)
|
|
|
|
|
|
class IncrementalEncoder(codecs.BufferedIncrementalEncoder):
|
|
def _buffer_encode(self, data: str, errors: str, final: bool) -> Tuple[bytes, int]:
|
|
if errors != "strict":
|
|
raise IDNAError('Unsupported error handling "{}"'.format(errors))
|
|
|
|
if not data:
|
|
return b"", 0
|
|
|
|
labels = _unicode_dots_re.split(data)
|
|
trailing_dot = b""
|
|
if labels:
|
|
if not labels[-1]:
|
|
trailing_dot = b"."
|
|
del labels[-1]
|
|
elif not final:
|
|
# Keep potentially unfinished label until the next call
|
|
del labels[-1]
|
|
if labels:
|
|
trailing_dot = b"."
|
|
|
|
result = []
|
|
size = 0
|
|
for label in labels:
|
|
result.append(alabel(label))
|
|
if size:
|
|
size += 1
|
|
size += len(label)
|
|
|
|
# Join with U+002E
|
|
result_bytes = b".".join(result) + trailing_dot
|
|
size += len(trailing_dot)
|
|
return result_bytes, size
|
|
|
|
|
|
class IncrementalDecoder(codecs.BufferedIncrementalDecoder):
|
|
def _buffer_decode(self, data: Any, errors: str, final: bool) -> Tuple[str, int]:
|
|
if errors != "strict":
|
|
raise IDNAError('Unsupported error handling "{}"'.format(errors))
|
|
|
|
if not data:
|
|
return ("", 0)
|
|
|
|
if not isinstance(data, str):
|
|
data = str(data, "ascii")
|
|
|
|
labels = _unicode_dots_re.split(data)
|
|
trailing_dot = ""
|
|
if labels:
|
|
if not labels[-1]:
|
|
trailing_dot = "."
|
|
del labels[-1]
|
|
elif not final:
|
|
# Keep potentially unfinished label until the next call
|
|
del labels[-1]
|
|
if labels:
|
|
trailing_dot = "."
|
|
|
|
result = []
|
|
size = 0
|
|
for label in labels:
|
|
result.append(ulabel(label))
|
|
if size:
|
|
size += 1
|
|
size += len(label)
|
|
|
|
result_str = ".".join(result) + trailing_dot
|
|
size += len(trailing_dot)
|
|
return (result_str, size)
|
|
|
|
|
|
class StreamWriter(Codec, codecs.StreamWriter):
|
|
pass
|
|
|
|
|
|
class StreamReader(Codec, codecs.StreamReader):
|
|
pass
|
|
|
|
|
|
def search_function(name: str) -> Optional[codecs.CodecInfo]:
|
|
if name != "idna2008":
|
|
return None
|
|
return codecs.CodecInfo(
|
|
name=name,
|
|
encode=Codec().encode,
|
|
decode=Codec().decode,
|
|
incrementalencoder=IncrementalEncoder,
|
|
incrementaldecoder=IncrementalDecoder,
|
|
streamwriter=StreamWriter,
|
|
streamreader=StreamReader,
|
|
)
|
|
|
|
|
|
codecs.register(search_function)
|