Upload docx_formatters.py

docx_formatters.py

@@ -0,0 +1,482 @@
+"""
+docx_formatters.py
+
+Funções auxiliares para formatação de documentos DOCX.
+Não contém conteúdo/texto - apenas lógica de formatação.
+"""
+
+from docx.shared import Pt, Inches, Cm, RGBColor
+from docx.enum.text import WD_ALIGN_PARAGRAPH
+from docx.enum.table import WD_TABLE_ALIGNMENT, WD_CELL_VERTICAL_ALIGNMENT
+from docx.oxml.ns import qn
+from docx.oxml import OxmlElement
+
+WD_ALIGN_VERTICAL = WD_CELL_VERTICAL_ALIGNMENT
+
+
+# ============================================================================
+# NUMERAÇÃO AUTOMÁTICA DE SEÇÕES
+# ============================================================================
+
+class NumeradorSecoes:
+    """
+    Gerencia numeração automática hierárquica de seções.
+    
+    Uso:
+        num = NumeradorSecoes()
+        num.secao("SOLICITAÇÃO")        # 1. SOLICITAÇÃO
+        num.subsecao("Considerações")   # 1.1 Considerações
+        num.subsecao("Documentação")    # 1.2 Documentação
+        num.secao("IMÓVEL")             # 2. IMÓVEL
+        num.subsecao("Descrição")       # 2.1 Descrição
+        num.subsubsecao("Detalhe")      # 2.1.1 Detalhe
+    """
+    
+    def __init__(self, secao_inicial=0):
+        """
+        Args:
+            secao_inicial: Número inicial para seções (0 = começa em 1)
+        """
+        self.contadores = [secao_inicial, 0, 0, 0]  # [secao, subsecao, subsubsecao, subsubsubsecao]
+    
+    def _reset_niveis_abaixo(self, nivel):
+        """Reseta contadores de níveis abaixo do especificado."""
+        for i in range(nivel + 1, len(self.contadores)):
+            self.contadores[i] = 0
+    
+    def secao(self):
+        """Incrementa e retorna número da seção (ex: '1')."""
+        self.contadores[0] += 1
+        self._reset_niveis_abaixo(0)
+        return str(self.contadores[0])
+    
+    def subsecao(self):
+        """Incrementa e retorna número da subseção (ex: '1.1')."""
+        self.contadores[1] += 1
+        self._reset_niveis_abaixo(1)
+        return f"{self.contadores[0]}.{self.contadores[1]}"
+    
+    def subsubsecao(self):
+        """Incrementa e retorna número da sub-subseção (ex: '1.1.1')."""
+        self.contadores[2] += 1
+        self._reset_niveis_abaixo(2)
+        return f"{self.contadores[0]}.{self.contadores[1]}.{self.contadores[2]}"
+    
+    def subsubsubsecao(self):
+        """Incrementa e retorna número da sub-sub-subseção (ex: '1.1.1.1')."""
+        self.contadores[3] += 1
+        return f"{self.contadores[0]}.{self.contadores[1]}.{self.contadores[2]}.{self.contadores[3]}"
+    
+    def numero_atual(self, nivel=0):
+        """Retorna o número atual de um nível sem incrementar."""
+        if nivel == 0:
+            return str(self.contadores[0])
+        elif nivel == 1:
+            return f"{self.contadores[0]}.{self.contadores[1]}"
+        elif nivel == 2:
+            return f"{self.contadores[0]}.{self.contadores[1]}.{self.contadores[2]}"
+        else:
+            return f"{self.contadores[0]}.{self.contadores[1]}.{self.contadores[2]}.{self.contadores[3]}"
+    
+    def definir_secao(self, numero):
+        """Define manualmente o número da seção atual."""
+        self.contadores[0] = numero
+        self._reset_niveis_abaixo(0)
+
+
+# ============================================================================
+# HELPERS BÁSICOS
+# ============================================================================
+
+def aplicar_cor_run(run, cor):
+    """
+    Aplica cor RGB a um run.
+    
+    Args:
+        run: Run do python-docx
+        cor: tuple (r,g,b) ou RGBColor
+    """
+    if cor:
+        if isinstance(cor, tuple):
+            run.font.color.rgb = RGBColor(cor[0], cor[1], cor[2])
+        else:
+            run.font.color.rgb = cor
+
+
+def set_cell_shading(cell, color):
+    """
+    Define a cor de fundo de uma célula.
+    
+    Args:
+        cell: Célula da tabela
+        color: Cor em hex (ex: "E6E6E6")
+    """
+    shading = OxmlElement('w:shd')
+    shading.set(qn('w:fill'), color)
+    cell._tc.get_or_add_tcPr().append(shading)
+
+
+# ============================================================================
+# CRIAÇÃO DE PARÁGRAFOS
+# ============================================================================
+
+def criar_paragrafo_formatado(doc, texto, negrito=False, sublinhado=False, italico=False,
+                               tamanho=11, cor=None, alinhamento=WD_ALIGN_PARAGRAPH.JUSTIFY,
+                               espaco_antes=0, espaco_depois=6, recuo_esq=0, 
+                               recuo_primeira_linha=0, fonte='Arial'):
+    """
+    Cria um parágrafo com formatação completa.
+    
+    Args:
+        doc: Documento python-docx
+        texto: Texto do parágrafo
+        negrito, sublinhado, italico: Formatação do texto
+        tamanho: Tamanho da fonte em pontos
+        cor: RGBColor ou tuple (r,g,b)
+        alinhamento: WD_ALIGN_PARAGRAPH constant
+        espaco_antes, espaco_depois: Espaçamento em pontos
+        recuo_esq: Recuo esquerdo em inches
+        recuo_primeira_linha: Recuo da primeira linha em cm
+        fonte: Nome da fonte
+    
+    Returns:
+        Parágrafo criado
+    """
+    p = doc.add_paragraph()
+    run = p.add_run(texto)
+    
+    run.bold = negrito
+    run.underline = sublinhado
+    run.italic = italico
+    run.font.size = Pt(tamanho)
+    run.font.name = fonte
+    aplicar_cor_run(run, cor)
+    
+    p.alignment = alinhamento
+    p.paragraph_format.space_before = Pt(espaco_antes)
+    p.paragraph_format.space_after = Pt(espaco_depois)
+    if recuo_esq:
+        p.paragraph_format.left_indent = Inches(recuo_esq)
+    if recuo_primeira_linha:
+        p.paragraph_format.first_line_indent = Cm(recuo_primeira_linha)
+    
+    return p
+
+
+# ============================================================================
+# TÍTULOS E SEÇÕES
+# ============================================================================
+
+def add_heading_custom(doc, text, level=1):
+    """Adiciona título customizado."""
+    configs = {
+        1: {'tamanho': 14, 'antes': 18, 'depois': 12},
+        2: {'tamanho': 12, 'antes': 12, 'depois': 6},
+    }
+    cfg = configs.get(level, {'tamanho': 11, 'antes': 6, 'depois': 6})
+    
+    return criar_paragrafo_formatado(
+        doc, text, negrito=True, sublinhado=True, tamanho=cfg['tamanho'],
+        espaco_antes=cfg['antes'], espaco_depois=cfg['depois'],
+        alinhamento=WD_ALIGN_PARAGRAPH.LEFT
+    )
+
+
+def add_section_title(doc, number, text):
+    """Adiciona título de seção principal (ex: 1. SOLICITAÇÃO)."""
+    return criar_paragrafo_formatado(
+        doc, f"{number}. {text}", negrito=True, tamanho=12,
+        espaco_antes=18, espaco_depois=12, alinhamento=WD_ALIGN_PARAGRAPH.LEFT
+    )
+
+
+def add_subsection_title(doc, number, text):
+    """Adiciona título de subseção (ex: 1.1 Considerações)."""
+    return criar_paragrafo_formatado(
+        doc, f"{number} {text}", negrito=True, tamanho=11,
+        espaco_antes=12, espaco_depois=6, recuo_esq=0.3,
+        alinhamento=WD_ALIGN_PARAGRAPH.LEFT
+    )
+
+
+def add_subsubsection_title(doc, number, text):
+    """Adiciona título de sub-subseção (ex: 3.2.1)."""
+    return criar_paragrafo_formatado(
+        doc, f"{number} {text}", negrito=True, tamanho=11,
+        espaco_antes=10, espaco_depois=4, recuo_esq=0.5,
+        alinhamento=WD_ALIGN_PARAGRAPH.LEFT
+    )
+
+
+def add_subsubsubsection_title(doc, number, text):
+    """Adiciona título de sub-sub-subseção (ex: 3.2.1.1)."""
+    return criar_paragrafo_formatado(
+        doc, f"{number} {text}", negrito=True, tamanho=10,
+        espaco_antes=8, espaco_depois=4, recuo_esq=0.7,
+        alinhamento=WD_ALIGN_PARAGRAPH.LEFT
+    )
+
+
+# ============================================================================
+# TEXTO DO CORPO
+# ============================================================================
+
+def add_body_text(doc, text, cor=None, indent=True):
+    """Adiciona texto do corpo com formatação padrão."""
+    return criar_paragrafo_formatado(
+        doc, text, tamanho=11, cor=cor,
+        espaco_depois=6, recuo_esq=0.3 if indent else 0,
+        recuo_primeira_linha=1.25 if indent else 0
+    )
+
+
+def add_placeholder_text(doc, text="[Preencher informações]", indent=True):
+    """Adiciona texto placeholder em vermelho."""
+    return criar_paragrafo_formatado(
+        doc, text, tamanho=11, cor=RGBColor(255, 0, 0),
+        espaco_depois=6, recuo_esq=0.3 if indent else 0,
+        recuo_primeira_linha=1.25 if indent else 0
+    )
+
+
+def add_bullet_text(doc, text, indent_level=1):
+    """Adiciona texto com bullet/recuo."""
+    return criar_paragrafo_formatado(
+        doc, f"• {text}", tamanho=11,
+        espaco_depois=3, recuo_esq=0.3 + (indent_level * 0.2)
+    )
+
+
+# ============================================================================
+# TABELAS
+# ============================================================================
+
+def formatar_celula_tabela(cell, texto, negrito=False, cor=None, shading_color=None):
+    """
+    Formata uma célula de tabela com configurações padrão.
+    
+    Args:
+        cell: Célula da tabela
+        texto: Texto a inserir
+        negrito: Se o texto deve ser negrito
+        cor: Cor do texto (tuple ou RGBColor)
+        shading_color: Cor de fundo da célula (hex string)
+    
+    Returns:
+        Run criado
+    """
+    p = cell.paragraphs[0]
+    
+    p.alignment = WD_ALIGN_PARAGRAPH.CENTER
+    p.paragraph_format.left_indent = Cm(0)
+    p.paragraph_format.right_indent = Cm(0)
+    p.paragraph_format.first_line_indent = Cm(0)
+    p.paragraph_format.space_before = Pt(0)
+    p.paragraph_format.space_after = Pt(0)
+    
+    cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
+    
+    run = p.add_run(texto)
+    run.font.size = Pt(9)
+    run.font.name = 'Arial'
+    run.bold = negrito
+    
+    if cor:
+        aplicar_cor_run(run, cor)
+    
+    if shading_color:
+        set_cell_shading(cell, shading_color)
+    
+    return run
+
+from docx.shared import Inches
+from docx.enum.table import WD_TABLE_ALIGNMENT
+
+def add_simple_table(doc, dados_tabela, header_row=True, largura_colunas=None):
+    """
+    Adiciona uma tabela simples ao documento.
+    
+    Args:
+        doc: Documento
+        dados_tabela: Lista de listas. Cada célula pode ser string ou dict {'texto', 'cor'}
+        header_row: Se True, primeira linha é cabeçalho (negrito com fundo cinza)
+        largura_colunas: Lista de inteiros representando proporção da largura de cada coluna.
+                         Por padrão, todas iguais (1).
+    
+    Returns:
+        Tabela criada ou None se dados vazios
+    """
+    if not dados_tabela or not dados_tabela[0]:
+        return None
+    
+    num_colunas = len(dados_tabela[0])
+    
+    # Se não passou largura_colunas, define 1 para todas
+    if largura_colunas is None:
+        largura_colunas = [1] * num_colunas
+    elif len(largura_colunas) != num_colunas:
+        raise ValueError("largura_colunas deve ter o mesmo número de colunas de dados_tabela")
+    
+    # Normaliza a largura relativa
+    total = sum(largura_colunas)
+    proporcoes = [x / total for x in largura_colunas]
+    
+    # Cria tabela
+    table = doc.add_table(rows=len(dados_tabela), cols=num_colunas)
+    table.style = 'Table Grid'
+    table.alignment = WD_TABLE_ALIGNMENT.CENTER
+    
+    # Determinar largura total da tabela (aprox. 6.5 polegadas)
+    largura_total = Inches(6.5)
+    larguras_em_colunas = [largura_total * p for p in proporcoes]
+    
+    for i, row_data in enumerate(dados_tabela):
+        row = table.rows[i]
+        for j, cell_data in enumerate(row_data):
+            if j >= len(row.cells):
+                continue
+            
+            cell = row.cells[j]
+            
+            if isinstance(cell_data, dict):
+                texto = cell_data.get('texto', '')
+                cor = cell_data.get('cor')
+            else:
+                texto = str(cell_data) if cell_data else ""
+                cor = None
+            
+            is_header = header_row and i == 0
+            
+            # Formata a célula (sua função existente)
+            formatar_celula_tabela(
+                cell, texto, 
+                negrito=is_header, 
+                cor=cor,
+                shading_color="E6E6E6" if is_header else None
+            )
+            
+            # Define largura da coluna
+            cell.width = larguras_em_colunas[j]
+    
+    doc.add_paragraph()
+    return table
+
+def configurar_linha_tabela_altura(row, altura_cm=0.6):
+    """
+    Configura altura mínima de uma linha de tabela.
+    
+    Args:
+        row: Linha da tabela
+        altura_cm: Altura em centímetros
+    """
+    tr = row._tr
+    trPr = tr.get_or_add_trPr()
+    trHeight = OxmlElement('w:trHeight')
+    trHeight.set(qn('w:val'), str(int(Cm(altura_cm).twips)))
+    trHeight.set(qn('w:hRule'), 'atLeast')
+    trPr.append(trHeight)
+
+
+def criar_celula_cabecalho_tabela(cell, texto, recuo_primeira_linha_cm=1.0):
+    """
+    Formata uma célula de cabeçalho de seção na tabela principal.
+    
+    Args:
+        cell: Célula da tabela
+        texto: Texto do cabeçalho
+        recuo_primeira_linha_cm: Recuo da primeira linha
+    """
+    cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
+    
+    p = cell.paragraphs[0]
+    p.paragraph_format.left_indent = Cm(0)
+    p.paragraph_format.first_line_indent = Cm(recuo_primeira_linha_cm)
+    p.alignment = WD_ALIGN_PARAGRAPH.LEFT
+    
+    run = p.add_run(texto)
+    run.bold = True
+    run.underline = True
+    run.font.size = Pt(11)
+    run.font.name = 'Arial'
+    
+    set_cell_shading(cell, "E6E6E6")
+
+
+def criar_celula_dados_tabela(cell, texto, is_label=False):
+    """
+    Formata uma célula de dados na tabela principal.
+    
+    Args:
+        cell: Célula da tabela
+        texto: Texto da célula
+        is_label: Se True, formata como rótulo (negrito)
+    """
+    cell.vertical_alignment = WD_ALIGN_VERTICAL.CENTER
+    
+    p = cell.paragraphs[0]
+    p.paragraph_format.left_indent = Cm(0)
+    p.paragraph_format.first_line_indent = Cm(0)
+    p.paragraph_format.space_before = Pt(0)
+    p.paragraph_format.space_after = Pt(0)
+    
+    run = p.add_run(texto)
+    run.bold = is_label
+    run.font.size = Pt(9)
+    run.font.name = 'Arial'
+
+
+# ============================================================================
+# IMAGENS
+# ============================================================================
+
+def inserir_imagem_de_documento(doc, imagem_data, largura_inches=5.5):
+    """
+    Insere uma imagem extraída de outro documento.
+    
+    Args:
+        doc: Documento destino
+        imagem_data: Dict com 'paragraph_element' e 'doc' (documento origem)
+        largura_inches: Largura da imagem
+    
+    Returns:
+        True se inseriu com sucesso, False caso contrário
+    """
+    if not imagem_data:
+        return False
+    
+    try:
+        from io import BytesIO
+        
+        para_elem = imagem_data['paragraph_element']
+        doc_origem = imagem_data['doc']
+        
+        ns_a = '{http://schemas.openxmlformats.org/drawingml/2006/main}'
+        ns_r = '{http://schemas.openxmlformats.org/officeDocument/2006/relationships}'
+        
+        for blip in para_elem.iter(f'{ns_a}blip'):
+            embed_id = blip.get(f'{ns_r}embed')
+            if embed_id:
+                try:
+                    image_part = doc_origem.part.related_parts[embed_id]
+                    image_stream = BytesIO(image_part.blob)
+                    
+                    p = doc.add_paragraph()
+                    p.alignment = WD_ALIGN_PARAGRAPH.CENTER
+                    run = p.add_run()
+                    run.add_picture(image_stream, width=Inches(largura_inches))
+                    return True
+                except Exception as e:
+                    print(f"Erro ao copiar imagem: {e}")
+                    return False
+    except Exception as e:
+        print(f"Erro ao processar imagem: {e}")
+        return False
+    
+    return False
+
+
+def add_image_placeholder(doc):
+    """Adiciona placeholder de imagem."""
+    p = doc.add_paragraph()
+    p.add_run("[IMAGEM]").italic = True
+    p.alignment = WD_ALIGN_PARAGRAPH.CENTER