add option to addlocalfiles

import sys

from django.core.management.base import BaseCommand, CommandError

import os

from django.contrib.auth.models import User

from resources.models import Project, FileExtension

from concurrent.futures import ThreadPoolExecutor

import multiprocessing

from pathlib import Path

@lru_cache(maxsize=None)

                yield os.path.join(root, file)

def _add_file(file_path: str, project: Project, current_user: User, start_dir: str):

def _add_file(

    file_path: str,

    project: Project,

    current_user: User,

    start_dir: str,

    delete_source: bool = False,

):

    resource_id = handle_local_file(file_path, project)

    collection_path = os.path.dirname(file_path)[len(start_dir) :]

    if collection_path:

            current_user, collection_path, project.pk

        )

        collection_id = hierarchy_of_collections[-1]["id"]

        add_resource_to_collection(current_user, resource_id, collection_id)

        if add_resource_to_collection(current_user, resource_id, collection_id):

            if delete_source:

                Path(file_path).unlink(missing_ok=True)

class Command(BaseCommand):

        parser.add_argument("user", type=str)

        parser.add_argument("project", type=int)

        parser.add_argument("--extensions", nargs="+", type=str)

        parser.add_argument("--delete", action="store_true")

    def handle(self, *args, **options):

        if options.get("extensions"):

            allowed_extensions = options.get("extensions")

        else:

            allowed_extensions = ALLOWED_EXTENSIONS

        delete_source = options.get("delete", False)

        start_dir = options.get("start_dir")

        if not os.path.isdir(start_dir):

            raise CommandError("{} is not a directory".format(start_dir))

            _, extension = os.path.splitext(file_path)

            if extension.lower() not in allowed_extensions:

                continue

            executor.submit(_add_file, file_path, project, current_user, start_dir)

            executor.submit(

                _add_file, file_path, project, current_user, start_dir, delete_source

            )