from .core import encode, decode, alabel, ulabel, IDNAError

import codecs

import re

​

_unicode_dots_re = re.compile(u'[\u002e\u3002\uff0e\uff61]')

​

class Codec(codecs.Codec):

​

    def encode(self, data, errors='strict'):

​

        if errors != 'strict':

            raise IDNAError("Unsupported error handling \"{0}\"".format(errors))

​

        if not data:

            return "", 0

​

        return encode(data), len(data)

​

    def decode(self, data, errors='strict'):

​

        if errors != 'strict':

            raise IDNAError("Unsupported error handling \"{0}\"".format(errors))

​

        if not data:

            return u"", 0

​

        return decode(data), len(data)

​

class IncrementalEncoder(codecs.BufferedIncrementalEncoder):

    def _buffer_encode(self, data, errors, final):

        if errors != 'strict':

            raise IDNAError("Unsupported error handling \"{0}\"".format(errors))

​

        if not data:

            return ("", 0)

​

        labels = _unicode_dots_re.split(data)

        trailing_dot = u''

        if labels:

            if not labels[-1]:

                trailing_dot = '.'

                del labels[-1]

            elif not final: