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