created TreeTag model, fixes for addtaxonomy command

This commit is contained in:
Sam
2021-02-10 13:03:10 +00:00
parent 41e68031fa
commit 02f6b29e5f
3 changed files with 16 additions and 3 deletions

View File

@@ -3,6 +3,8 @@ import logging
from django.core.management.base import BaseCommand
from django.conf import settings
from core.models import TreeTag
class Command(BaseCommand):
@@ -11,16 +13,20 @@ class Command(BaseCommand):
def handle(self, *args, **kwargs):
print(self.help)
print("Deleting existing instances")
TreeTag.objects.all().delete()
file_path = settings.BASE_DIR + '/../datasets/' + settings.TAXONOMY_FILE
counter = 0
with open(file_path, 'rt') as data_file:
print(f"Reading from {settings.TAXONOMY_FILE}")
for line in data_file.readlines():
try:
tag = TagModel.objects.create(name=line)
tag = TreeTag.objects.create(name=line)
counter += 1
print('.', end='')
logging.debug(f"{tag} created from {line}")
except Exception as e:
logging.error(f"{type(e)} while creating tags from {settings.TAXONOMY_FILE}")
print(f"{counter} new TagModel instances created")
print(f"\n{counter} new TreeTag instances created")
print('Shutting down\n')