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