import video sliced directories to beaverdam

import_images_dir.py

from django.core.management.base import BaseCommand, CommandError
from annotator.models import Video
import os
import json
import re
import os
import shutil

# referring to https://stackoverflow.com/questions/5967500/how-to-correctly-sort-a-string-with-a-number-inside for "human sorting"
def atoi(text):
	    return int(text) if text.isdigit() else text

def natural_keys(text):
	    return [ atoi(c) for c in re.split('(\d+)', text) ]

class Command(BaseCommand):
	help = "Imports a directory of images as a new view object"

	def add_arguments(self, parser):
		parser.add_argument('directory')
		parser.add_argument('host_url')
		parser.add_argument('--recursive', action="store_true")

	def handle(self, *args, **options):
		if (options['directory'] and options['host_url']):
			directory = options['directory']
			if not options['recursive']:
				self.create_entry_from_dir(options['directory'],
						options['host_url'])
			else:
				for subdir in os.listdir(options['directory']):
					host_url = ''.join([options['host_url'], subdir])
					self.create_entry_from_dir(os.path.join(directory, subdir),
							host_url)
				self.stdout.write("You got it!")

	def create_entry_from_dir(self, directory, host_url):
		f_list = []
		for f in os.listdir(directory):
			f_path = os.path.join(directory, f)
			if (os.path.isfile(f_path)):
				f_list.append(f)
		f_list.sort(key = natural_keys)
		image_list_json = json.dumps(f_list)
		source = os.path.basename(directory)
		v = Video(source = source, image_list = image_list_json, host = host_url)
		v.save()
		#self.stdout.write(f"Video {v.id} was added.")
		self.stdout.write("Video {} was added.".format(v.id))
		self.stdout.write("Video.source = {}".format(v.source))
		self.stdout.write("Video.host = {}".format(v.host))
		self.stdout.write("Video.image_list = {}".format(v.image_list))