| | 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 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') |
| |
|
