Source code for ferrosoft.apps.dataimport.management.commands.dataimport_settings

from django.core.management import BaseCommand

from ferrosoft.apps.dataimport.models import Settings
from ferrosoft.apps.ferrobase.management.tenantcommand import TenantAwareCommand
from ferrosoft.apps.ferrobase.models import Directory




class Command(TenantAwareCommand, BaseCommand):


    def add_arguments(self, parser):
        super().add_arguments(parser)
        parser.add_argument(
            "--importfiledir",
            type=str,
            default="",
            help="Import file directory",
        )




    def handle_tenant(self, *args, **options):
        settings = Settings.get_instance()
        dir_name = options["importfiledir"]
        if dir_name != "":
            import_file_dir = Directory.get_or_create(dir_name)
            settings.import_file_dir = import_file_dir
            settings.save(update_fields=["import_file_dir"])

        self.stdout.write(self.style.SUCCESS("Settings saved."))