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',
@@ -86,7 +86,7 @@ DOCS = {
         'documentoadministrativo/{0}/{0}_texto_integral{1}')
     ],
     DocumentoAcessorioAdministrativo: [(
-    	 'arquivo',
+        'arquivo',
         'administrativo/{}',
         'documentoacessorioadministrativo/{0}/{0}_acessorio_administrativo{1}')
     ],
@@ -169,15 +169,22 @@ def migrar_docs_por_ids(tipo):
         for arq in os.listdir(dir_origem):
             match = pat.match(arq)
             if match:
-                origem = os.path.join(dir_origem, match.group(0))
-                id = match.group(1)
-                extensao = get_extensao(origem)
-                destino = base_destino.format(id, extensao)
-                mover_documento(origem, destino)
-
                 # 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)
+
                     setattr(obj, campo, destino)
                     obj.save()
                 except tipo.DoesNotExist: