def populate_thesaurus(): nodes = load_csv_treemap(TREEMAP_FILE) rows = [] for node in nodes: label = node.breadcrumb.split('\u00BB')[-1].strip() row = ThesaurusClass(id=node.id, root=node.root, label=label[:100], level=node.level, size=node.size, ratio=node.ratio, x=node.x, y=node.y, width=node.w, height=node.h, sort=node.sort, breadcrumb=node.breadcrumb[:300]) rows.append(row) ThesaurusClass.objects.bulk_create(rows)
def load_level4_nodes(): nodes = load_csv_treemap(TREEMAP_FILE) return set([n.id for n in nodes if n.level == 4])