Spaces:
Runtime error
Runtime error
| from src.model.container import Container | |
| INFINITE = 99999 | |
| def create_dic_levels(c:Container,dict_of_levels : dict = {}): | |
| if c.level == 0: | |
| dict_of_levels[c.level] = [0] | |
| for child in c.children: | |
| if child.level not in dict_of_levels: | |
| dict_of_levels[child.level] = [1 for _ in range(child.level)] | |
| create_dic_levels(child, dict_of_levels) | |
| if INFINITE in dict_of_levels.keys(): | |
| dict_of_levels[INFINITE] = [1] | |
| return dict_of_levels | |
| def create_good_indexes(c:Container, dict_of_levels : dict): | |
| actual_level = c.level | |
| c.index = dict_of_levels[actual_level].copy() | |
| actual_len = len(dict_of_levels[actual_level]) | |
| temp_update = dict_of_levels[actual_level][-1] | |
| dict_of_levels[actual_level][-1] += 1 | |
| for i in dict_of_levels.values(): | |
| if len(i) > actual_len: | |
| i[actual_len - 1] = temp_update | |
| for child in c.children: | |
| c_lvl = child.level | |
| for i in dict_of_levels.values(): | |
| if len(i) > c_lvl: | |
| i[c_lvl:] = [1 for _ in range(len(i[c_lvl:]))] | |
| create_good_indexes(child, dict_of_levels) # Apply the function recursively to all children | |
| def create_good_indexes_not_ordered_titles(c:Container, dict_of_levels : dict): | |
| actual_level = c.level | |
| c.index = dict_of_levels[actual_level].copy() | |
| actual_len = len(dict_of_levels[actual_level]) | |
| temp_update = dict_of_levels[actual_level][-1] | |
| dict_of_levels[actual_level][-1] += 1 | |
| for i in dict_of_levels.values(): | |
| if len(i) > actual_len: | |
| i[actual_len - 1] = temp_update | |
| for child in c.children: | |
| c_lvl = child.level | |
| for i in dict_of_levels.values(): | |
| if len(i) > c_lvl: | |
| i[c_lvl:] = [1 for _ in range(len(i[c_lvl:]))] | |
| create_good_indexes(child, dict_of_levels) # Apply the function recursively to all children | |
| def set_good_block_indexes(c:Container): | |
| for i in c.containers: | |
| for b in i.blocks: | |
| b.index = i.index | |
| def set_indexes(c:Container): | |
| dict_levels = create_dic_levels(c) | |
| myKeys = list(dict_levels.keys()) | |
| myKeys.sort() | |
| dict_levels = {key: dict_levels[key] for key in myKeys} | |
| if c.children and c.children[0] and (c.children[0].level > min(list(dict_levels.keys())[1:])): | |
| c.children[0].level = min(list(dict_levels.keys())[1:]) | |
| create_good_indexes_not_ordered_titles(c, dict_levels) | |
| else: | |
| create_good_indexes(c, dict_levels) | |
| set_good_block_indexes(c) | |