sigi/sigi/apps/casas/admin.py

from typing import Any
from django.db.models.query import QuerySet
from django.http import HttpRequest
from email_validator import validate_email, EmailNotValidError
from django.db.models import F
from django.contrib import admin
from django.contrib.contenttypes.admin import GenericTabularInline
from django.template.loader import render_to_string
from django.urls import reverse
from django.utils.html import format_html
from django.utils.safestring import mark_safe
from django.utils.translation import gettext as _
from django_weasyprint.views import WeasyTemplateResponse
from djbs import djbs_constants as djbsc
from import_export import resources
from import_export.admin import ExportActionMixin
from import_export.fields import Field
from sigi.apps.casas.forms import OrgaoForm
from sigi.apps.casas.models import Orgao, Funcionario, TipoOrgao
from sigi.apps.casas.filters import (
    GerentesInterlegisFilter,
    ServicoFilter,
)
from sigi.apps.convenios.filters import (
    TipoProjetoFilter,
    ExcluirTipoProjetoFilter,
)
from sigi.apps.contatos.models import Telefone
from sigi.apps.convenios.models import Convenio
from sigi.apps.ocorrencias.models import Ocorrencia
from sigi.apps.parlamentares.models import Parlamentar
from sigi.apps.servicos.models import Servico
from sigi.apps.servicos.filters import ServicoAtivoFilter
from sigi.apps.servidores.models import Servidor
from sigi.apps.utils.mixins import AsciifyQParameter
from sigi.apps.utils.mixins import ReturnMixin


class OrgaoExportResourceContato(resources.ModelResource):
    class Meta:
        model = Orgao
        fields = ("nome", "email")
        export_order = fields
        name = "Exportação para aplicativo Contatos"

    def dehydrate_nome(self, orgao):
        return orgao.nome[:50]

    def export(self, queryset=None, *args, **kwargs):
        if queryset is not None:
            queryset = queryset.exclude(email="").filter(
                email__regex=r"([A-Za-z0-9]+[.\-_])*[A-Za-z0-9]+@[A-Za-z0-9-]+(\.[A-Z|a-z]{2,})+"
            )
        return super().export(queryset, *args, **kwargs)


class OrgaoExportResourseGeral(resources.ModelResource):
    presidente = Field(column_name="presidente")
    telefone = Field(column_name="telefone")
    # servicos_seit = Field(column_name='servicos_seit')
    contato = Field(column_name="contato")
    nome = Field(column_name="nome")

    class Meta:
        model = Orgao
        fields = (
            "municipio__codigo_ibge",
            "cnpj",
            "municipio__codigo_tse",
            "nome",
            "municipio__nome",
            "municipio__uf__sigla",
            "presidente",
            "logradouro",
            "bairro",
            "cep",
            "telefone",
            "pagina_web",
            "email",
            "ult_alt_endereco",
            "contato",
        )
        export_order = fields
        name = "Exportação de uso geral"

    def dehydrate_nome(self, orgao):
        return orgao.nome[:50]

    def dehydrate_presidente(self, orgao):
        return orgao.presidente

    def dehydrate_telefone(self, orgao):
        return orgao.telefone

    # def dehydrate_servicos_seit(self, orgao):
    #     servicos = [s.tipo_servico.nome for s in orgao.servico_set.filter(
    #         data_desativacao__isnull=True)]
    #     return ", ".join(servicos)

    def dehydrate_contato(self, orgao):
        return ", ".join(
            [
                f"{c.cargo if c.cargo else 'Sem cargo'}: {c.nome} ({c.email})"
                for c in orgao.funcionario_set.filter(desativado=False)
            ]
        )


class TelefonesInline(GenericTabularInline):
    model = Telefone
    readonly_fields = ("ult_alteracao",)
    extra = 1


class ParlamentarInline(admin.TabularInline):
    model = Parlamentar
    fields = (
        "get_foto",
        "nome_parlamentar",
        "status_mandato",
        "get_partido",
        "email",
        "redes_sociais",
        "presidente",
    )
    readonly_fields = fields
    extra = 0
    max_num = 0
    show_change_link = True
    can_delete = False

    @mark_safe
    @admin.display(description=_("Foto"))
    def get_foto(self, obj):
        if obj.foto:
            return f'<img class="circle" src="{obj.foto.url}" style="width: 58px; height: 58px;"/>'
        else:
            return (
                '<i class="material-icons medium grey-text">account_circle</i>'
            )

    @admin.display(description=_("Partido"))
    def get_partido(self, obj):
        return obj.partido.sigla


class FuncionarioInline(admin.StackedInline):
    model = Funcionario
    djbs_cols = "1"
    fields = (
        ("desativado", "ult_alteracao"),
        "setor",
        "nome",
        ("cpf", "identidade", "sexo"),
        ("nota", "email"),
        ("cargo", "funcao"),
        ("redes_sociais", "observacoes"),
    )
    readonly_fields = ["ult_alteracao"]
    extra = 0
    show_change_link = True
    can_delete = False
    verbose_name_plural = _("Contatos da Casa")
    ordering = ["setor", "-ult_alteracao"]

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        return qs.exclude(desativado=True)


class ListaFuncionarioInline(admin.TabularInline):
    model = Funcionario
    fields = (
        "get_setor",
        "nome",
        "nota",
        "get_email_link",
        "observacoes",
        "ult_alteracao",
    )
    readonly_fields = fields
    extra = 0
    max_num = 0
    show_change_link = True
    can_delete = False
    verbose_name_plural = _("Lista de contatos")
    ordering = ["setor", "-ult_alteracao"]
    djbs_hide_original = True

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        return qs.exclude(desativado=True)

    @admin.display(description=_("setor"))
    def get_setor(self, func):
        if func.setor == "contato_interlegis":
            return format_html(
                "<span class='green lighten-5 z-depth-1'>{setor}</span>",
                setor=func.get_setor_display(),
            )
        return func.get_setor_display()

    @admin.display(description=_("e-mail"))
    def get_email_link(self, func):
        return format_html(
            "<a href='mailto:{email}' targe='_blank'>{email}</a>",
            email=func.email,
        )


class ConveniosInline(admin.TabularInline):
    model = Convenio
    fields = (
        "num_processo_sf",
        "link_sigad",
        "status_convenio",
        "num_convenio",
        "get_projeto",
        "data_retorno_assinatura",
        "data_termino_vigencia",
        "data_pub_diario",
        "data_sigad",
        "data_solicitacao",
    )
    readonly_fields = fields
    ordering = ("-data_retorno_assinatura",)
    readonly_fields = fields
    extra = 0
    max_num = 0
    show_change_link = True
    can_delete = False

    def get_queryset(self, request: HttpRequest):
        return (
            super()
            .get_queryset(request)
            .order_by(
                F("data_retorno_assinatura").desc(nulls_last=True),
                F("data_termino_vigencia").desc(nulls_last=True),
            )
        )

    @mark_safe
    @admin.display(description=_("Status do convênio"))
    def status_convenio(self, obj):
        if obj.pk is None:
            return None
        status = obj.get_status()
        if status in ["Vencido", "Desistência", "Cancelado", "Extinto"]:
            label = r"bg-danger"
        elif status == "Vigente":
            label = r"bg-success"
        elif status == "Pendente":
            label = r"bg-alert"
        else:
            label = r""
        return f'<span class="{label} p-1 rounded">{status}</span>'

    @admin.display(description=_("Ver no SIGAD"))
    def link_sigad(self, obj):
        if obj.pk is None:
            return ""
        return mark_safe(obj.get_sigad_url(display_type="icone"))

    @admin.display(description=_("Tipo de convênio"))
    def get_projeto(self, obj):
        if obj.pk is None:
            return ""
        return obj.projeto.sigla


class ServicoInline(admin.TabularInline):
    model = Servico
    fields = (
        "get_tipo_servico",
        "get_url",
        "hospedagem_interlegis",
        "data_ativacao",
        "data_desativacao",
        "resultado_verificacao",
        "data_ultimo_uso",
    )
    readonly_fields = fields
    ordering = ("-data_desativacao", "-data_ativacao", "tipo_servico")
    extra = 0
    max_num = 0
    show_change_link = True
    can_delete = False

    @admin.display(description=_("Tipo de serviço"), ordering="tipo_servico")
    def get_tipo_servico(self, obj):
        return obj.tipo_servico.sigla

    @mark_safe
    @admin.display(description="Url do serviço", ordering="url")
    def get_url(self, obj):
        return f"<a href='{obj.url}' target='_blank'>{obj.url}</a>"


class OcorrenciaInline(admin.StackedInline):
    model = Ocorrencia
    fields = (
        "data_criacao",
        "data_modificacao",
        "categoria",
        "tipo_contato",
        "assunto",
        "prioridade",
        "status",
        "ticket",
        "descricao",
        "resolucao",
        "servidor_registro",
    )
    autocomplete_fields = ("categoria", "tipo_contato")
    readonly_fields = ["data_criacao", "data_modificacao", "servidor_registro"]
    ordering = ("-data_modificacao",)
    extra = 0
    show_change_link = True
    can_delete = False

    def has_add_permission(self, request, obj):
        if Servidor.objects.filter(user=request.user).exists():
            return super().has_add_permission(request, obj)
        return False

    def has_change_permission(self, request, obj):
        if Servidor.objects.filter(user=request.user).exists():
            return super().has_change_permission(request, obj)
        return False


@admin.register(TipoOrgao)
class TipoOrgaoAdmin(admin.ModelAdmin):
    list_display = ("sigla", "nome", "legislativo")
    list_filter = ("legislativo",)
    search_fields = ("sigla", "nome")


@admin.register(Funcionario)
class FuncionarioAdmin(ReturnMixin, admin.ModelAdmin):
    fieldsets = (
        (None, {"fields": ("nome", "sexo", "data_nascimento")}),
        (_("Documentos"), {"fields": ("cpf", "identidade")}),
        (_("Contato"), {"fields": ("nota", "email", "redes_sociais")}),
        (
            _("Endereço"),
            {"fields": ("endereco", "municipio", "bairro", "cep")},
        ),
        (
            _("Vínculo"),
            {
                "fields": (
                    "casa_legislativa",
                    "cargo",
                    "funcao",
                    "setor",
                    "tempo_de_servico",
                )
            },
        ),
        (
            _("Validade"),
            {"fields": ("ult_alteracao", "desativado", "observacoes")},
        ),
    )
    readonly_fields = ("ult_alteracao",)
    autocomplete_fields = ("casa_legislativa", "municipio")


@admin.register(Orgao)
class OrgaoAdmin(AsciifyQParameter, ExportActionMixin, admin.ModelAdmin):
    save_on_top = True
    form = OrgaoForm
    resource_classes = [OrgaoExportResourseGeral, OrgaoExportResourceContato]
    inlines = (
        TelefonesInline,
        ParlamentarInline,
        ListaFuncionarioInline,
        FuncionarioInline,
        ConveniosInline,
        ServicoInline,
        OcorrenciaInline,
    )
    list_display = (
        "id",
        "sigla",
        "nome",
        "get_uf",
        "get_gerentes",
        "get_convenios",
        "get_servicos",
    )
    list_display_links = (
        "id",
        "sigla",
        "nome",
    )
    list_filter = (
        "tipo",
        ("gerentes_interlegis", GerentesInterlegisFilter),
        "municipio__uf__nome",
        ("convenio__projeto_id", TipoProjetoFilter),
        ("servico__data_desativacao", ServicoAtivoFilter),
        ("convenio__projeto_id", ExcluirTipoProjetoFilter),
        ServicoFilter,
        "inclusao_digital",
        ("email", admin.EmptyFieldListFilter),
    )
    ordering = ("municipio__uf__nome", "nome")
    fieldsets = (
        (
            _("Identificação"),
            {
                "fields": (
                    "tipo",
                    "nome",
                    "sigla",
                    "cnpj",
                    "data_instalacao",
                    "gerentes_interlegis",
                )
            },
        ),
        (
            _("Endereço"),
            {
                "fields": (
                    "logradouro",
                    "bairro",
                    "municipio",
                    "cep",
                    "ult_alt_endereco",
                ),
            },
        ),
        (
            _("Presença na Internet"),
            {
                "fields": (
                    "inclusao_digital",
                    "data_levantamento",
                    "pesquisador",
                    "pagina_web",
                    "email",
                    "obs_pesquisa",
                )
            },
        ),
        (
            _("Outras informações"),
            {
                "fields": (
                    "telefone_geral",
                    "horario_funcionamento",
                    "observacoes",
                    "foto",
                    "brasao",
                ),
            },
        ),
    )
    autocomplete_fields = ("tipo", "municipio", "pesquisador")
    readonly_fields = ["gerentes_interlegis", "ult_alt_endereco"]
    search_fields = (
        "search_text",
        "sigla",
        "cnpj",
        "bairro",
        "logradouro",
        "cep",
        "municipio__nome",
        "municipio__uf__nome",
        "municipio__codigo_ibge",
        "pagina_web",
        "observacoes",
    )
    reports = [
        "casas_sem_processo",
        "relatorio_simples",
        "relatorio_completo",
        "etiqueta_presidente_25",
        "etiqueta_presidente_39",
        "etiqueta_25",
        "etiqueta_39",
        "etiqueta_parlamentar_25",
        "etiqueta_parlamentar_39",
    ]

    def get_queryset(self, request):
        queryset = super().get_queryset(request)
        return queryset.prefetch_related("gerentes_interlegis", "convenio_set")

    def save_related(self, request, form, formsets, change):
        ocorrencia_formset = next(
            filter(lambda f: f.model == Ocorrencia, formsets), None
        )
        if ocorrencia_formset is not None:
            try:
                servidor = Servidor.objects.get(user=request.user)
            except Servidor.DoesNotExist:
                servidor = None
            if servidor is not None:
                ocorrencia_formset.save(commit=False)
                for obj in ocorrencia_formset.new_objects:
                    if (
                        not hasattr(obj, "servidor_registro")
                        or obj.servidor_registro is None
                    ):
                        obj.servidor_registro = servidor
        return super().save_related(request, form, formsets, change)

    @admin.display(
        description=_("Unidade da Federação"), ordering="municipio__uf__nome"
    )
    def get_uf(self, obj):
        return obj.municipio.uf.nome

    @admin.display(description=_("Gerente Interlegis"))
    def get_gerentes(self, obj):
        return mark_safe(
            "<ul><li>" + "</li><li>".join(obj.lista_gerentes()) + "</ul>"
        )

    @admin.display(description=_("Convênios"))
    def get_convenios(self, obj):
        return mark_safe(
            "<ul>"
            + "".join(
                [
                    f"<li>{c}</li>"
                    for c in obj.convenio_set.order_by(
                        "-data_retorno_assinatura"
                    )
                ]
            )
            + "</ul>"
        )

    @admin.display(description=_("Serviços"))
    def get_servicos(self, obj):
        return mark_safe(
            "<ul>"
            + "".join(
                [
                    f'<li><a href="{s.url}" target="_blank">{s}</a></li>'
                    for s in obj.servico_set.filter(
                        data_desativacao__isnull=True
                    )
                ]
            )
            + "</ul>"
        )

    def lookup_allowed(self, lookup, value):
        return super(OrgaoAdmin, self).lookup_allowed(
            lookup, value
        ) or lookup in [
            "tipo__legislativo__exact",
            "tipo__sigla__exact",
            "tipo__sigla__in",
            "municipio__uf__codigo_ibge__exact",
            "convenio__projeto__id__exact",
        ]

    def casas_sem_processo(self, request):
        context = {
            "casas": self.get_queryset(request)
            .filter(convenio=None)
            .order_by("municipio__uf", "nome"),
            "title": _("Casas sem nenhum processo de convênio"),
        }
        return WeasyTemplateResponse(
            filename="casas_sem_processo.pdf",
            request=request,
            template="casas/casas_sem_convenio_pdf.html",
            context=context,
            content_type="application/pdf",
        )

    casas_sem_processo.title = _("Casas sem nenhum processo de convênio")

    def relatorio_simples(self, request):
        context = {
            "casas": self.get_queryset(request).order_by(
                "municipio__uf", "nome"
            ),
            "title": _("Relatório Simples"),
        }
        return WeasyTemplateResponse(
            filename="relatorio_simples.pdf",
            request=request,
            template="casas/casas_sem_convenio_pdf.html",
            context=context,
            content_type="application/pdf",
        )

    relatorio_simples.title = _("Relatório Simples")

    def relatorio_completo(self, request):
        context = {
            "casas": self.get_queryset(request).order_by(
                "municipio__uf", "nome"
            ),
            "title": _("Relatório completo"),
        }
        return WeasyTemplateResponse(
            filename="relatorio_completo.pdf",
            request=request,
            template="casas/relatorio_completo_pdf.html",
            context=context,
            content_type="application/pdf",
        )

    relatorio_completo.title = _("Relatório completo")

    def get_actions(self, request):
        actions = super(OrgaoAdmin, self).get_actions(request)
        if "delete_selected" in actions:
            del actions["delete_selected"]
        return actions