converters/Range.py

from singleton_decorator import singleton
import re
from .Cardinal import CardinalVietnamese


@singleton
class Range:
    """
    Steps:
    - Check for - splitting numbers

    Note:
    Punctuation always stays the same
    """

    def __init__(self):
        super().__init__()
        self.cardinal = CardinalVietnamese()

    def convert(self, token: str) -> str:
        numbers = re.split('-', token)
        if len(numbers) == 1:
            token = self.cardinal.convert(numbers[0])
        elif len(numbers) == 2:

            token = self.cardinal.convert(numbers[0])
            token += ' đến '
            token += self.cardinal.convert(numbers[1])

        else:
            token = ''
            for number in numbers:
                token += self.cardinal.convert(number)
                token += ' '

        return token