def do_init(self): file = open(self.file_path, "r") lines = file.readlines() file.close() for line in lines[1:]: info = line.decode("utf-8")[:-1] tag_name, score, parents_str, equal_to, items_str = info.split("\t") print(tag_name) score = 1.0 items = [] parents = parents_str.split(",") for item in items_str.split(","): name, class_name = item.split("__") if class_name == "NORMAL": print("Save normal tag %s" % name) n = Normal(slug=name, score=score) n.save() items.append({"slug": name, "class": class_name}) tag = Tag(name=tag_name, score=score, parents=parents, equal_to=equal_to, items=items) tag.save()
def load(self): with open(name=self.path, mode='r') as f: for each_line in f.readlines(): name, parents_str = each_line.decode('utf-8')[:-1].split('\t') parents = parents_str.split(',') print(name, parents) # To normal model exists = Normal.get_by_slug(name) if not exists: item = Normal(slug=name) print("Saving Normal Item %s" % name) item.save() for parent in parents: exists = Normal.get_by_slug(parent) if not exists: parent_item = Normal(slug=parent) print("Saving Normal Item %s" % parent) parent_item.save() # To tag model exists = Tag.get_by_name(name) if not exists: tag = Tag(name=name, parents=parents, score=self.score, items=[{'slug': name, 'class': 'NORMAL'},], proxy='NORMAL') print("Tag Item, %s" % name) tag.save() for parent in parents: exists = Tag.get_by_name(parent) if not exists: tag = Tag(name=parent, score=self.score, items=[{'slug': parent, 'class': 'NORMAL'},], proxy='NORMAL') print("Tag item %s " % parent) tag.save()
def callback(info): normal = Normal(**info) normal.save() print(info.get('slug', ''))