"""
Formatadores de seções e orientação de página.
"""
from docx import Document
from docx.shared import Cm
from docx.enum.section import WD_ORIENT


def iniciar_secao_paisagem(doc: Document) -> None:
    """
    Adiciona uma nova seção com orientação paisagem.

    Args:
        doc: Documento DOCX
    """
    # Adicionar quebra de seção
    new_section = doc.add_section()

    # Configurar orientação paisagem
    new_section.orientation = WD_ORIENT.LANDSCAPE

    # Trocar largura e altura para paisagem
    new_width = new_section.page_height
    new_height = new_section.page_width
    new_section.page_width = new_width
    new_section.page_height = new_height

    # Ajustar margens para paisagem
    new_section.top_margin = Cm(2)
    new_section.bottom_margin = Cm(2)
    new_section.left_margin = Cm(2)
    new_section.right_margin = Cm(2)


def iniciar_secao_retrato(doc: Document) -> None:
    """
    Adiciona uma nova seção com orientação retrato.

    Args:
        doc: Documento DOCX
    """
    # Adicionar quebra de seção
    new_section = doc.add_section()

    # Configurar orientação retrato
    new_section.orientation = WD_ORIENT.PORTRAIT

    # Trocar largura e altura para retrato (se estava em paisagem)
    if new_section.page_width > new_section.page_height:
        new_width = new_section.page_height
        new_height = new_section.page_width
        new_section.page_width = new_width
        new_section.page_height = new_height

    # Margens padrão para retrato
    new_section.top_margin = Cm(2.5)
    new_section.bottom_margin = Cm(2.5)
    new_section.left_margin = Cm(3)
    new_section.right_margin = Cm(2)