--- a/src/iconolab_episteme/management/commands/importmetacategories.py Tue Jun 26 15:55:08 2018 +0200
+++ /dev/null Thu Jan 01 00:00:00 1970 +0000
@@ -1,90 +0,0 @@
-# -*- coding: UTF-8 -*-
-import json
-import logging
-import os
-import pprint
-import re
-import shutil
-
-from django.conf import settings
-from django.core.management.base import BaseCommand, CommandError
-
-from iconolab.management.commands.importimages import BaseImportImagesCommand
-from iconolab.models import (Collection, Folder, Image, ImageStats, Item,
- ItemMetadata, MetaCategory)
-
-if settings.IMPORT_LOGGER_NAME and settings.LOGGING['loggers'].get(settings.IMPORT_LOGGER_NAME, ''):
- logger = logging.getLogger(settings.IMPORT_LOGGER_NAME)
-else:
- logger = logging.getLogger(__name__)
-
-class Command(BaseImportImagesCommand):
- help = 'import metacategories files from a directory'
-
- def add_arguments(self, parser):
- parser.add_argument('source_dir')
- parser.add_argument(
- '--encoding',
- dest='encoding',
- default='utf-8',
- help='JSON file encoding'
-
- )
- parser.add_argument(
- '--collection-id',
- dest='collection_id',
- default=False,
- help='insert extracted data into the specified collection instead of trying to load a collection fixture',
- )
- parser.add_argument(
- '--metacategories-json',
- dest='metacategories_json',
- default=False,
- help='add metacategories to the collection from a json file (json must be a list of object with "label" and "triggers_notifications" fields)',
- )
-
- def handle(self, *args, **options):
-
- print('# Logging with logger '+logger.name)
- logger.debug('# Initializing command with args: %r', options)
-
- self.source_dir = options.get('source_dir')
-
- if options.get('collection_id'):
- print('## Finding collection with id ' +
- options.get('collection_id'))
- try:
- collection = Collection.objects.get(
- pk=options.get('collection_id'))
- except Collection.DoesNotExist:
- raise ValueError('!!! Collection with primary key ' +
- options.get('collection_id')+' was not found, aborting !!!')
- else:
- raise ValueError(
- '!!! No collection fixture or collection id, aborting because we can\'t properly generate data. !!!')
-
- if options.get('metacategories_json'):
- print('## Finding metacategories fixture json data in '+self.source_dir)
- metacategories_json_path = os.path.join(
- self.source_dir, options.get('metacategories_json'))
- if not os.path.isfile(metacategories_json_path):
- print('### No '+options.get('metacategories_json')+
- '.json file was found in the source directory')
- raise ValueError(
- '!!! Fixture file '+metacategories_json_path+' was not found !!!')
- with open(metacategories_json_path) as metacategories_json_file:
- metacategories_data = json.loads(
- metacategories_json_file.read())
- for metacategory in metacategories_data:
- if metacategory.get('label', None) is None:
- raise ValueError(
- '!!! Metacategory without label !!!')
-
- if options.get('metacategories_json'):
- for metacategory in metacategories_data:
- MetaCategory.objects.create(
- collection=collection,
- label=metacategory.get('label'),
- triggers_notifications=metacategory.get(
- 'triggers_notifications', 0)
- )
\ No newline at end of file