mirror of https://github.com/interlegis/sapl.git
Leandro Roberto
6 years ago
3 changed files with 348 additions and 295 deletions
@ -0,0 +1,326 @@ |
|||
|
|||
import logging |
|||
|
|||
from django import apps |
|||
from django.conf import settings |
|||
from django.contrib.contenttypes.models import ContentType |
|||
from django.db.models import Q |
|||
from django.db.models.fields.files import FileField |
|||
from django.http import Http404 |
|||
from django.utils.decorators import classonlymethod |
|||
from django.utils.text import capfirst |
|||
from django.utils.translation import ugettext_lazy as _ |
|||
import django_filters |
|||
from django_filters.rest_framework.backends import DjangoFilterBackend |
|||
from django_filters.rest_framework.filterset import FilterSet |
|||
from django_filters.utils import resolve_field |
|||
from rest_framework import serializers as rest_serializers |
|||
from rest_framework.decorators import list_route, detail_route,\ |
|||
permission_classes |
|||
from rest_framework.generics import ListAPIView |
|||
from rest_framework.mixins import ListModelMixin, RetrieveModelMixin |
|||
from rest_framework.permissions import (IsAuthenticated, |
|||
IsAuthenticatedOrReadOnly, AllowAny) |
|||
from rest_framework.response import Response |
|||
from rest_framework.viewsets import GenericViewSet, ModelViewSet |
|||
|
|||
from sapl.api.forms import (AutorChoiceFilterSet, AutoresPossiveisFilterSet, |
|||
AutorSearchForFieldFilterSet) |
|||
from sapl.api.permissions import SaplModelPermissions |
|||
from sapl.api.serializers import (AutorChoiceSerializer, AutorSerializer, |
|||
ChoiceSerializer, |
|||
ModelChoiceSerializer, |
|||
SessaoPlenariaOldSerializer, |
|||
MateriaLegislativaOldSerializer) |
|||
from sapl.base.models import TipoAutor, Autor |
|||
from sapl.comissoes.models import Comissao |
|||
from sapl.materia.models import MateriaLegislativa, Proposicao |
|||
from sapl.parlamentares.models import Parlamentar |
|||
from sapl.sessao.models import SessaoPlenaria |
|||
from sapl.utils import SaplGenericRelation |
|||
|
|||
|
|||
class ModelChoiceView(ListAPIView): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
|
|||
""" |
|||
|
|||
# FIXME aplicar permissão correta de usuário |
|||
permission_classes = (IsAuthenticated,) |
|||
serializer_class = ModelChoiceSerializer |
|||
|
|||
def get(self, request, *args, **kwargs): |
|||
self.model = ContentType.objects.get_for_id( |
|||
self.kwargs['content_type']).model_class() |
|||
|
|||
pagination = request.GET.get('pagination', '') |
|||
|
|||
if pagination == 'False': |
|||
self.pagination_class = None |
|||
|
|||
return ListAPIView.get(self, request, *args, **kwargs) |
|||
|
|||
def get_queryset(self): |
|||
return self.model.objects.all() |
|||
|
|||
|
|||
class AutorListView(ListAPIView): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
|
|||
Listagem de Autores com filtro para autores já cadastrados |
|||
e/ou possíveis autores. |
|||
|
|||
- tr - tipo do resultado |
|||
Prepera Lista de Autores para 2 cenários distintos |
|||
|
|||
- default = 1 |
|||
|
|||
= 1 -> para (value, text) usados geralmente |
|||
em combobox, radiobox, checkbox, etc com pesquisa básica |
|||
de Autores feita pelo django-filter |
|||
-> processo usado nas pesquisas, o mais usado. |
|||
|
|||
|
|||
= 3 -> Devolve instancias da classe Autor filtradas pelo |
|||
django-filter |
|||
|
|||
- tipo - chave primária do Tipo de Autor a ser filtrado |
|||
|
|||
- q - busca textual no nome do Autor ou em fields_search |
|||
declarados no field SaplGenericRelation das GenericFks |
|||
A busca textual acontece via django-filter com a |
|||
variável `tr` igual 1 ou 3. Em caso contrário, |
|||
o django-filter é desativado e a busca é feita |
|||
no model do ContentType associado ao tipo. |
|||
|
|||
- q_0 / q_1 - q_0 é opcional e quando usado, faz o código ignorar "q"... |
|||
|
|||
q_0 -> campos lookup a serem filtrados em qualquer Model |
|||
que implemente SaplGenericRelation |
|||
q_1 -> o valor que será pesquisado no lookup de q_0 |
|||
|
|||
q_0 e q_1 podem ser separados por ","... isso dará a |
|||
possibilidade de filtrar mais de um campo. |
|||
|
|||
|
|||
http://localhost:8000 |
|||
/api/autor?tr=1&q_0=parlamentar_set__ativo&q_1=False |
|||
/api/autor?tr=1&q_0=parlamentar_set__ativo&q_1=True |
|||
/api/autor?tr=3&q_0=parlamentar_set__ativo&q_1=False |
|||
/api/autor?tr=3&q_0=parlamentar_set__ativo&q_1=True |
|||
|
|||
http://localhost:8000 |
|||
/api/autor?tr=1 |
|||
&q_0=parlamentar_set__nome_parlamentar__icontains, |
|||
parlamentar_set__ativo |
|||
&q_1=Carvalho,False |
|||
/api/autor?tr=1 |
|||
&q_0=parlamentar_set__nome_parlamentar__icontains, |
|||
parlamentar_set__ativo |
|||
&q_1=Carvalho,True |
|||
/api/autor?tr=3 |
|||
&q_0=parlamentar_set__nome_parlamentar__icontains, |
|||
parlamentar_set__ativo |
|||
&q_1=Carvalho,False |
|||
/api/autor?tr=3 |
|||
&q_0=parlamentar_set__nome_parlamentar__icontains, |
|||
parlamentar_set__ativo |
|||
&q_1=Carvalho,True |
|||
|
|||
|
|||
não importa o campo que vc passe de qualquer dos Models |
|||
ligados... é possível ver que models são esses, |
|||
na ocasião do commit deste texto, executando: |
|||
In [6]: from sapl.utils import models_with_gr_for_model |
|||
|
|||
In [7]: models_with_gr_for_model(Autor) |
|||
Out[7]: |
|||
[sapl.parlamentares.models.Parlamentar, |
|||
sapl.parlamentares.models.Frente, |
|||
sapl.comissoes.models.Comissao, |
|||
sapl.materia.models.Orgao, |
|||
sapl.sessao.models.Bancada, |
|||
sapl.sessao.models.Bloco] |
|||
|
|||
qualquer atributo destes models podem ser passados |
|||
para busca |
|||
""" |
|||
logger = logging.getLogger(__name__) |
|||
|
|||
TR_AUTOR_CHOICE_SERIALIZER = 1 |
|||
TR_AUTOR_SERIALIZER = 3 |
|||
|
|||
permission_classes = (IsAuthenticatedOrReadOnly,) |
|||
queryset = Autor.objects.all() |
|||
model = Autor |
|||
|
|||
filter_class = AutorChoiceFilterSet |
|||
filter_backends = (DjangoFilterBackend, ) |
|||
serializer_class = AutorChoiceSerializer |
|||
|
|||
@property |
|||
def tr(self): |
|||
username = self.request.user.username |
|||
try: |
|||
tr = int(self.request.GET.get |
|||
('tr', AutorListView.TR_AUTOR_CHOICE_SERIALIZER)) |
|||
|
|||
if tr not in (AutorListView.TR_AUTOR_CHOICE_SERIALIZER, |
|||
AutorListView.TR_AUTOR_SERIALIZER): |
|||
return AutorListView.TR_AUTOR_CHOICE_SERIALIZER |
|||
except Exception as e: |
|||
self.logger.error('user=' + username + '. ' + str(e)) |
|||
return AutorListView.TR_AUTOR_CHOICE_SERIALIZER |
|||
return tr |
|||
|
|||
def get(self, request, *args, **kwargs): |
|||
if self.tr == AutorListView.TR_AUTOR_SERIALIZER: |
|||
self.serializer_class = AutorSerializer |
|||
self.permission_classes = (IsAuthenticated,) |
|||
|
|||
if self.filter_class and 'q_0' in request.GET: |
|||
self.filter_class = AutorSearchForFieldFilterSet |
|||
|
|||
return ListAPIView.get(self, request, *args, **kwargs) |
|||
|
|||
|
|||
class AutoresProvaveisListView(ListAPIView): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
""" |
|||
|
|||
logger = logging.getLogger(__name__) |
|||
|
|||
permission_classes = (IsAuthenticatedOrReadOnly,) |
|||
queryset = Autor.objects.all() |
|||
model = Autor |
|||
|
|||
filter_class = None |
|||
filter_backends = [] |
|||
serializer_class = ChoiceSerializer |
|||
|
|||
def get_queryset(self): |
|||
|
|||
params = {'content_type__isnull': False} |
|||
username = self.request.user.username |
|||
tipo = '' |
|||
try: |
|||
tipo = int(self.request.GET.get('tipo', '')) |
|||
if tipo: |
|||
params['id'] = tipo |
|||
except Exception as e: |
|||
self.logger.error('user= ' + username + '. ' + str(e)) |
|||
pass |
|||
|
|||
tipos = TipoAutor.objects.filter(**params) |
|||
|
|||
if not tipos.exists() and tipo: |
|||
raise Http404() |
|||
|
|||
r = [] |
|||
for tipo in tipos: |
|||
q = self.request.GET.get('q', '').strip() |
|||
|
|||
model_class = tipo.content_type.model_class() |
|||
|
|||
fields = list(filter( |
|||
lambda field: isinstance(field, SaplGenericRelation) and |
|||
field.related_model == Autor, |
|||
model_class._meta.get_fields(include_hidden=True))) |
|||
|
|||
""" |
|||
fields - é um array de SaplGenericRelation que deve possuir o |
|||
atributo fields_search. Verifique na documentação da classe |
|||
a estrutura de fields_search. |
|||
""" |
|||
|
|||
assert len(fields) >= 1, (_( |
|||
'Não foi encontrado em %(model)s um atributo do tipo ' |
|||
'SaplGenericRelation que use o model %(model_autor)s') % { |
|||
'model': model_class._meta.verbose_name, |
|||
'model_autor': Autor._meta.verbose_name}) |
|||
|
|||
qs = model_class.objects.all() |
|||
|
|||
q_filter = Q() |
|||
if q: |
|||
for item in fields: |
|||
if item.related_model != Autor: |
|||
continue |
|||
q_fs = Q() |
|||
for field in item.fields_search: |
|||
q_fs = q_fs | Q(**{'%s%s' % ( |
|||
field[0], |
|||
field[1]): q}) |
|||
q_filter = q_filter & q_fs |
|||
|
|||
qs = qs.filter(q_filter).distinct( |
|||
fields[0].fields_search[0][0]).order_by( |
|||
fields[0].fields_search[0][0]) |
|||
else: |
|||
qs = qs.order_by(fields[0].fields_search[0][0]) |
|||
|
|||
qs = qs.values_list( |
|||
'id', fields[0].fields_search[0][0]) |
|||
r += list(qs) |
|||
|
|||
if tipos.count() > 1: |
|||
r.sort(key=lambda x: x[1].upper()) |
|||
return r |
|||
|
|||
|
|||
class AutoresPossiveisListView(ListAPIView): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
""" |
|||
|
|||
permission_classes = (IsAuthenticatedOrReadOnly,) |
|||
queryset = Autor.objects.all() |
|||
model = Autor |
|||
|
|||
pagination_class = None |
|||
|
|||
filter_class = AutoresPossiveisFilterSet |
|||
serializer_class = AutorChoiceSerializer |
|||
|
|||
|
|||
class MateriaLegislativaViewSet(ListModelMixin, |
|||
RetrieveModelMixin, |
|||
GenericViewSet): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
""" |
|||
|
|||
permission_classes = (IsAuthenticated,) |
|||
serializer_class = MateriaLegislativaOldSerializer |
|||
queryset = MateriaLegislativa.objects.all() |
|||
filter_backends = (DjangoFilterBackend,) |
|||
filter_fields = ('numero', 'ano', 'tipo', ) |
|||
|
|||
|
|||
class SessaoPlenariaViewSet(ListModelMixin, |
|||
RetrieveModelMixin, |
|||
GenericViewSet): |
|||
""" |
|||
Deprecated |
|||
|
|||
TODO Migrar para customização na api automática |
|||
""" |
|||
|
|||
permission_classes = (AllowAny,) |
|||
serializer_class = SessaoPlenariaOldSerializer |
|||
queryset = SessaoPlenaria.objects.all() |
|||
filter_backends = (DjangoFilterBackend,) |
|||
filter_fields = ('data_inicio', 'data_fim', 'interativa') |
|||
Loading…
Reference in new issue