created TreeTag model, fixes for addtaxonomy command
This commit is contained in:
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user