Altera o destino dos documentos migrados

Signed-off-by: Luciano Almeida <lucianoalmeida@interlegis.leg.br>

sapl/legacy/migracao_documentos.py

@@ -59,15 +59,15 @@ DOCS = {
     MateriaLegislativa: [(
         'texto_original',
         'materia/{}_texto_integral',
-        'materialegislativa/{0}/{0}_texto_integral{1}')],
+        'materialegislativa/{2}/{0}/{0}_texto_integral{1}')],
     DocumentoAcessorio: [(
         'arquivo',
         'materia/{}',
-        'documentoacessorio/{0}/{0}{1}')],
+        'documentoacessorio/{2}/{0}/{0}{1}')],
     NormaJuridica: [(
         'texto_integral',
         'norma_juridica/{}_texto_integral',
-        'normajuridica/{0}/{0}_texto_integral{1}')],
+        'normajuridica/{2}/{0}/{0}_texto_integral{1}')],
     SessaoPlenaria: [
         ('upload_ata',
          'ata_sessao/{}_ata_sessao',
@@ -169,15 +169,22 @@ def migrar_docs_por_ids(tipo):
         for arq in os.listdir(dir_origem):
             match = pat.match(arq)
             if match:
+                # associa documento ao objeto
+                try:
                     origem = os.path.join(dir_origem, match.group(0))
                     id = match.group(1)
+                    obj = tipo.objects.get(pk=id)
+
                     extensao = get_extensao(origem)
+                    if hasattr(obj, "ano"):
+                        destino = base_destino.format(id, extensao, obj.ano)
+                    elif hasattr(obj, "data"):
+                        destino = base_destino.format(
+                            id, extensao, obj.data.year)
+                    else:
                         destino = base_destino.format(id, extensao)
                     mover_documento(origem, destino)
 
-                # associa documento ao objeto
-                try:
-                    obj = tipo.objects.get(pk=id)
                     setattr(obj, campo, destino)
                     obj.save()
                 except tipo.DoesNotExist: