import math

class Block:
    def __init__(self, doc: str = '', title: str = '', content: str = '', content_fr: str = '',
                 index: str = '', rank: int = 0, level: int = 0, distance: float = 99999):
        self.doc = doc
        self.title = title
        self.title_fr = ""
        self.content = content
        self.content_fr = content_fr
        self.specials = []
        self.index = index
        self.rank = rank
        self.level = level
        self.distance = distance


    def separate_1_block_in_n(self, max_size=4500):
        """
        Separate a block in n blocks of equal size
        """
        content_length = len(self.content)
        n = math.ceil(content_length / max_size)
        block_size = content_length // n
        new_blocks = []
        for i in range(n):
            start = i * block_size
            end = (i + 1) * block_size if i < n - 1 else None
            new_blocks.append(Block(doc=self.doc,
                                    title=self.title + f"_part{i}",
                                    content=self.content[start:end],
                                    index=self.index + f"_{i}",
                                    rank=self.rank,
                                    level=self.level))
        return new_blocks

    def to_dict(self) -> {}:
        block_dict = {'doc': self.doc,
                      'title': self.title,
                      'title_fr': self.title_fr,
                      'content': self.content,
                      'content_fr': self.content_fr,
                      'index': self.index,
                      'rank': self.rank,
                      'level': self.level,
                      'distance': self.distance}
        for i, s in enumerate(self.specials):
            special_key = 'special_'+str(i)
            block_dict[special_key] = s
        block_dict['specials_len'] = len(self.specials)
        return block_dict

    def from_dict(self, block_dict: {}):
        self.doc = block_dict['doc']
        self.title = block_dict['title']
        self.title_fr = block_dict['title_fr']
        self.content = block_dict['content']
        self.content_fr = block_dict['content_fr']
        self.index = block_dict['index']
        self.rank = block_dict['rank']
        self.level = block_dict['level']
        self.distance = block_dict['distance']
        self.specials = []
        for i in range(block_dict['specials_len']):
            special_key = 'special_' + str(i)
            self.specials.append(block_dict[special_key])
        return self

    @property
    def distance_str(self) -> str:
        return format(self.distance, '.2f')