diff --git a/sapl/legacy/management/commands/migracao_25_31.py b/sapl/legacy/management/commands/migracao_25_31.py index ba9a80224..93834443d 100644 --- a/sapl/legacy/management/commands/migracao_25_31.py +++ b/sapl/legacy/management/commands/migracao_25_31.py @@ -5,16 +5,27 @@ from sapl.legacy.migracao import migrar class Command(BaseCommand): - help = 'Migração de dados do SAPL 2.5 para o SAPL 3.1' + help = "Migração de dados do SAPL 2.5 para o SAPL 3.1" def add_arguments(self, parser): parser.add_argument( - '-a', - action='store_true', + "-a", + action="store_true", default=False, - dest='apagar_do_legado', - help='Apagar entradas migradas do legado', + dest="apagar_do_legado", + help="Apagar entradas migradas do legado", + ) + + parser.add_argument( + "-f", + action="store_true", + default=False, + dest="flush", + help="Apagar banco antes de migrar", ) def handle(self, *args, **options): - migrar(apagar_do_legado=options['apagar_do_legado']) + migrar( + apagar_do_legado=options["apagar_do_legado"], + flush=options["flush"], + ) diff --git a/sapl/legacy/management/commands/migracao_documentos.py b/sapl/legacy/management/commands/migracao_documentos.py deleted file mode 100644 index 7cba09d5a..000000000 --- a/sapl/legacy/management/commands/migracao_documentos.py +++ /dev/null @@ -1,11 +0,0 @@ -from django.core.management.base import BaseCommand - -from sapl.legacy.migracao_documentos import migrar_documentos - - -class Command(BaseCommand): - - help = 'Migração documentos do SAPL 2.5 para o SAPL 3.1' - - def handle(self, *args, **options): - migrar_documentos()