import re

from mistletoe.span_token import SpanToken


class NonbreakingSpaces(SpanToken):
    """
    Insert unbreakable space into span token (§ 3)
    Identifies "§$number", "Abs.$number", "Klasse [A,E,N]" w/ and w\ space between.
    Identifies "..." with and without leading and trailing space
    """
    pattern = re.compile(r"(§|Abs\.|Klasse)\s?(\d|[NEA])")

    def __init__(self, match_object):
        self.first = match_object.group(1)
        self.second = match_object.group(2)

class NonbreakingSpacesDots(SpanToken):
    pattern = re.compile(r"\s?(\.\.\.)\s?")

    def __init__(self, match_object):
        self.first = match_object.group(1)