From c88d7f6f83a05a19f7f8e6ed4f02063e525f77bd Mon Sep 17 00:00:00 2001
From: Luciano Almeida <lucianoalmeida@interlegis.leg.br>
Date: Wed, 17 May 2017 14:40:17 -0300
Subject: [PATCH] Altera o destino dos documentos migrados

Signed-off-by: Luciano Almeida <lucianoalmeida@interlegis.leg.br>
---
 sapl/legacy/migracao_documentos.py | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/sapl/legacy/migracao_documentos.py b/sapl/legacy/migracao_documentos.py
index 7bb3c0347..7b4b36c41 100644
--- a/sapl/legacy/migracao_documentos.py
+++ b/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',
@@ -83,10 +83,10 @@ DOCS = {
     DocumentoAdministrativo: [(
         'texto_integral',
         'administrativo/{}_texto_integral',
-        'documentoadministrativo/{0}/{0}_texto_integral{1}')       
+        '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: