113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
from ninja import Router, File, Form
|
|
from ninja.files import UploadedFile
|
|
from typing import Optional
|
|
from django.http import HttpRequest, HttpResponse
|
|
from .search_service import (
|
|
buscar_diarios,
|
|
sugestao_termo,
|
|
buscar_diarios_simples,
|
|
)
|
|
from .schemas import BuscaDiariosResponseSchema, SugestaoResponse, DiarioOficialIn, DiarioOficialOut
|
|
from django.shortcuts import render
|
|
|
|
router = Router(tags=["Diários Oficiais"])
|
|
|
|
async def home(request):
|
|
return render(request, 'diarios/index.html')
|
|
|
|
@router.get(
|
|
"/sugestao",
|
|
response=SugestaoResponse,
|
|
summary="Sugestão de correção para termo de busca",
|
|
)
|
|
async def sugestao_busca(request: HttpRequest, q: str) -> SugestaoResponse:
|
|
"""
|
|
Sugere correção para o termo buscado, se necessário.
|
|
|
|
Args:
|
|
request (HttpRequest): Requisição HTTP.
|
|
q (str): Termo original digitado pelo usuário.
|
|
|
|
Returns:
|
|
SugestaoResponse: Termo corrigido.
|
|
"""
|
|
sugestao = await sugestao_termo(q)
|
|
return {"sugestao": sugestao}
|
|
|
|
|
|
@router.get(
|
|
"/busca",
|
|
response=BuscaDiariosResponseSchema,
|
|
summary="Busca simplificada com modos e ordenação",
|
|
)
|
|
async def busca_diarios_oficiais_simples(
|
|
request: HttpRequest,
|
|
q: Optional[str] = None,
|
|
numero_diario: Optional[str] = None,
|
|
data_inicio: Optional[str] = None,
|
|
data_fim: Optional[str] = None,
|
|
tipo: Optional[str] = None,
|
|
ordenar_por: str = "relevancia", # "relevancia", "data_asc", "data_desc"
|
|
modo_busca: str = "exata", # "exata" ou "qualquer"
|
|
page: int = 1,
|
|
page_size: int = 10,
|
|
) -> BuscaDiariosResponseSchema:
|
|
"""
|
|
Busca com modo de correspondência, ordenação e número do diário.
|
|
|
|
Args:
|
|
request (HttpRequest): Requisição HTTP.
|
|
q (Optional[str]): Termo de busca.
|
|
numero_diario (Optional[str]): Número exato do diário (ex: 1234/2024).
|
|
data_inicio (Optional[str]): Data inicial (YYYY-MM-DD).
|
|
data_fim (Optional[str]): Data final (YYYY-MM-DD).
|
|
tipo (Optional[str]): Tipo exato do diário.
|
|
ordenar_por (str): "relevancia", "data_asc" ou "data_desc".
|
|
modo_busca (str): "exata" ou "qualquer".
|
|
page (int): Página atual (mínimo: 1).
|
|
page_size (int): Itens por página (mínimo: 1, máximo: 50).
|
|
|
|
Returns:
|
|
BuscaDiariosResponseSchema: Resultado paginado da busca.
|
|
"""
|
|
page_size = min(max(page_size, 1), 50)
|
|
page = max(page, 1)
|
|
|
|
resultado = await buscar_diarios_simples(
|
|
query=q,
|
|
numero_diario=numero_diario,
|
|
data_inicio=data_inicio,
|
|
data_fim=data_fim,
|
|
tipo_diario=tipo,
|
|
ordenar_por=ordenar_por,
|
|
modo_busca=modo_busca,
|
|
page=page,
|
|
page_size=page_size,
|
|
)
|
|
return resultado
|
|
|
|
@router.post("/", response=DiarioOficialOut, summary="Criar novo diário oficial")
|
|
def criar_diario(
|
|
request: HttpRequest,
|
|
# Usamos Form para os dados normais e File para o upload
|
|
payload: DiarioOficialIn = Form(...),
|
|
arquivo: UploadedFile = File(None)
|
|
):
|
|
"""
|
|
Cria um novo diário oficial.
|
|
|
|
Observações:
|
|
- Aceita tanto upload de arquivo PDF quanto link para o diário
|
|
- Se ambos (arquivo e link) forem fornecidos, o arquivo terá prioridade
|
|
- O arquivo deve ser um PDF válido
|
|
"""
|
|
# Prioriza o arquivo se ambos existirem
|
|
arquivo_final = arquivo if arquivo else payload.arquivo
|
|
|
|
return DiarioOficialService.criar_diario(
|
|
data=payload.data,
|
|
numero=payload.numero,
|
|
tipo_id=payload.tipo_id,
|
|
arquivo=arquivo_final,
|
|
link=payload.link
|
|
) |