volume-visualization/automaticimport/views.py

import glob
import os
from django.conf import settings
from django.http import HttpResponse

from multiprocessing import Process

from .models import AutomaticImport

from volumes.models import Volume
from volumes.processing.service import SliceGenerator

from mongoengine import ValidationError

import json

import logging

import pdb

logger = logging.getLogger(__name__)

DEFAULT_SLICE_FOLDER = 'slices_8bit'

def trigger(request):
    volumes_to_process = []
    paths = sorted(glob.glob(os.path.join(settings.DEFAULT_IMPORT_PATH, '*')))

    # process dirs with dir DEFAULT_SLICE_FOLDER only
    dir_paths = []
    for path in paths:
        if os.path.isdir(path) and DEFAULT_SLICE_FOLDER in os.listdir(path):
            dir_paths.append(path)

    paths = dir_paths

    for path in paths:
        automatic_import = AutomaticImport.objects(path=path).first()
        if automatic_import is None:
            volume = Volume()
            volume.imageSequence = True
            volume.path = os.path.join(path, DEFAULT_SLICE_FOLDER)
            volume.name = path.replace(settings.DEFAULT_IMPORT_PATH, '')
            volume = volume.save()
            volumes_to_process.append(volume.id)

            automatic_import = AutomaticImport()
            automatic_import.path = path
            automatic_import.volume_id = str(volume.id)
            automatic_import.save()

            logger.info("Sequence in dir: " + path + " will be processed")
        
    if len(volumes_to_process) > 0:
        logger.info("Processing sequences in dir: " + settings.DEFAULT_IMPORT_PATH + " ...")
        
        process = Process(target=SliceGenerator.generateListOfVolumes, args=(volumes_to_process,))
        process.start()

    jsonDump = json.dumps({
        'triggered': True,
        'number_of_new_volumes': len(volumes_to_process)
    })
    return HttpResponse(jsonDump, content_type="json")

def clear(request):
    models = AutomaticImport.objects()

    if len(models) == 0:
        logger.info("No automatic imported volumes!")
        
    for automatic_import in models:
        try:
            volume = Volume.objects(id=automatic_import.volume_id)[0]
            volume.delete()
        except IndexError as e:
            logger.info("Volume with id = " + str(automatic_import.volume_id + " which was binded to AutomaticImport for path = " + automatic_import.path + " not found!"))

        automatic_import.delete()

        logger.info("Automaticly generated volume with id: " + automatic_import.volume_id + " deleted!")
        

    return HttpResponse(json.dumps({'deleted': len(models)}), content_type="json")