import copy
import unittest

from docx import Document
from docx.opc.constants import RELATIONSHIP_TYPE as RT
from docx.oxml import parse_xml
from docx.oxml.ns import qn

from document.evolutivo import construction_bdi_notes, construction_groups
from test_evo_adjustments import body
from test_evolutivo_documento import fixture, resolved


LABELS = ['Administração central', 'Despesas financeiras', 'Seguros e garantias', 'Riscos', 'Tributos', 'Remuneração']


def composition(values):
    return [{'descricao': label, 'percentual': value} for label, value in zip(LABELS, values)]


def notes(doc):
    root = parse_xml(doc.part.part_related_by(RT.FOOTNOTES).blob)
    return ' '.join(''.join(n.itertext(qn('w:t'))) for n in root if int(n.get(qn('w:id'))) > 0)


class BdiSectionsTests(unittest.TestCase):
    def data(self):
        data = fixture((2021, 2022))
        for year in data['anos']:
            c = year['construcao']['construcoes'][0]
            c.update(bdiComposicaoAdotada=composition([4, 1, 1, 2, 7, 10]), bdiJustificativa='Composição aditiva sobre o custo direto.')
        return data

    def render(self, data):
        imported = resolved(data)
        before = copy.deepcopy(imported)
        doc = Document()
        construction_bdi_notes(doc, construction_groups(imported['avaliacoes']))
        self.assertEqual(imported, before)
        return doc

    def test_general_chapter_only_introduction_and_each_construction_has_total(self):
        data = self.data()
        for year in data['anos']:
            second = copy.deepcopy(year['construcao']['construcoes'][0])
            second.update(id='second', seq=2, nome='Galpão B', bdi=20,
                          bdiComposicaoAdotada=composition([3, 1, 1, 2, 6, 7]))
            year['construcao']['construcoes'].append(second)
        doc = body(resolved(data))
        paragraphs = [p.text for p in doc.paragraphs]
        start = paragraphs.index('3.4.1.3 BDI e justificativa')
        end = paragraphs.index('3.4.1.3.1 Construção 1 — Galpão A')
        self.assertEqual(len(paragraphs[start + 1:end]), 2)
        self.assertTrue(paragraphs[start + 1].startswith('BDI (%):'))
        self.assertTrue(paragraphs[start + 2].startswith('O BDI informado em percentual'))
        self.assertIn('3.4.1.3.2 Construção 2 — Galpão B', paragraphs)
        tables = [t for t in doc.tables if t.cell(0, 0).text == 'Componente de BDI adotado']
        self.assertEqual(len(tables), 2)
        for table, total in zip(tables, ['25,00%', '20,00%']):
            self.assertEqual([c.text for c in table.rows[-1].cells], ['Total do BDI', total])
            self.assertEqual(len(table.rows), 8)
            self.assertEqual(len(table.columns), 2)
            self.assertTrue(all(run.bold for c in table.rows[-1].cells for p in c.paragraphs for run in p.runs))
        self.assertFalse(any('Aplicável a Construção' in p for p in paragraphs[start:paragraphs.index('3.4.1.4 Depreciação e conservação')]))

    def test_year_changes_compare_components_and_totals_without_duplicate_subchapters(self):
        data = self.data()
        c = data['anos'][1]['construcao']['construcoes'][0]
        c.update(bdi=20, bdiComposicaoAdotada=composition([3, 1, 1, 2, 6, 7]), bdiJustificativa='Justificativa do segundo exercício.')
        doc = self.render(data)
        self.assertEqual(len(doc.tables), 1)
        table = doc.tables[0]
        self.assertEqual([c.text for c in table.rows[0].cells], ['Parâmetro / valor', 'IPTU 2022', 'IPTU 2023'])
        self.assertEqual([c.text for c in table.rows[1].cells], ['Administração central', '4,00%', '3,00%'])
        self.assertEqual([c.text for c in table.rows[-1].cells], ['Total do BDI', '25,00%', '20,00%'])
        text = '\n'.join(p.text for p in doc.paragraphs)
        self.assertIn('Justificativa do BDI (IPTU 2022): Composição aditiva', text)
        self.assertIn('Justificativa do BDI (IPTU 2023): Justificativa do segundo exercício.', text)

    def test_divergent_sum_does_not_replace_saved_bdi(self):
        data = self.data()
        for year in data['anos']:
            year['construcao']['construcoes'][0]['bdiComposicaoAdotada'][0]['percentual'] = 5
        doc = self.render(data)
        self.assertEqual([c.text for c in doc.tables[0].rows[-2].cells], ['Soma dos componentes', '26,00%'])
        self.assertEqual([c.text for c in doc.tables[0].rows[-1].cells], ['Total do BDI', '25,00%'])
        self.assertIn('difere do percentual global registrado', notes(doc))

    def test_missing_composition_and_private_reason_are_not_invented_or_exposed(self):
        data = self.data()
        for year in data['anos']:
            c = year['construcao']['construcoes'][0]
            c.pop('bdiComposicaoAdotada')
            c.update(bdi=0, bdiJustificativa='JUSTIFICATIVA INTERNA', bdiJustificativaInterna=True)
        doc = self.render(data)
        self.assertEqual(len(doc.tables[0].rows), 2)
        self.assertEqual([c.text for c in doc.tables[0].rows[-1].cells], ['Total do BDI', '0,00%'])
        self.assertNotIn('JUSTIFICATIVA INTERNA', '\n'.join(p.text for p in doc.paragraphs) + notes(doc))
        self.assertIn('sem discriminar sua composição', notes(doc))


if __name__ == '__main__':
    unittest.main()