from csv_writer import CsvWriter
from sigi.apps.servicos.models import Servico


ARQUIVO_CSV = "/tmp/casas_que_usam_PM_e_SAPL.csv"


def escrever_casas(casas, arquivo):
    with open(arquivo, "wb") as f:
        writer = CsvWriter(f)
        writer.writerow(
            [
                "casa: sigla UF",
                "casa: nome UF",
                "casa: NOME",
            ]
        )
        for casa in casas:
            writer.writerow(
                [
                    casa.municipio.uf.sigla,
                    casa.municipio.uf,
                    casa.nome,
                ]
            )


casas_pm = {
    s.casa_legislativa
    for s in Servico.objects.filter(
        tipo_servico__sigla="PM", data_desativacao__isnull=True
    )
}
casas_sapl = {
    s.casa_legislativa
    for s in Servico.objects.filter(
        tipo_servico__sigla="SAPL", data_desativacao__isnull=True
    )
}
casas_ativas = casas_pm & casas_sapl
escrever_casas(casas_ativas, "/tmp/casas_que_usam_PM_e_SAPL__AMBOS_ATIVOS.csv")

casas_pm = {
    s.casa_legislativa
    for s in Servico.objects.filter(tipo_servico__sigla="PM")
}
casas_sapl = {
    s.casa_legislativa
    for s in Servico.objects.filter(tipo_servico__sigla="SAPL")
}
casas_todas = casas_pm & casas_sapl
escrever_casas(casas_todas, "/tmp/casas_que_usam_PM_e_SAPL__TODOS.csv")


casas = casas_pm.union(casas_sapl)
escrever_casas(casas, "/tmp/casas_que_usam_PM_ou_SAPL.csv")


casas_algum_inativo = casas_todas