from django.core.management.base import BaseCommand from django.apps import apps from ._readcsv import read_csv import os def dir_path(string): if os.path.isdir(string): return string else: raise NotADirectoryError(string) class Command(BaseCommand): help = 'Loads a test dataset into the database from a folder.' def add_arguments(self, parser): parser.add_argument('folder_path', nargs=1, type=dir_path) def handle(self, *args, **options): directory = options['folder_path'][0] if directory: read_csv(os.path.normpath(directory))