summaryrefslogtreecommitdiff
path: root/ishtar_common/rest.py
blob: 9354a943d2e0dfbfc5713b2168cc189080be214d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from rest_framework import authentication, permissions
from rest_framework.response import Response
from rest_framework.views import APIView

from ishtar_common.models import ApiSearchModel


class IpModelPermission(permissions.BasePermission):
    def has_permission(self, request, view):
        if not request.user or not getattr(request.user, "apiuser", None):
            return False
        ip_addr = request.META['REMOTE_ADDR']
        q = ApiSearchModel.objects.filter(
            user=request.user.apiuser,
            user__ip=ip_addr,
            content_type__app_label=view.model._meta.app_label,
            content_type__model=view.model._meta.model_name)
        return bool(q.count())


class SearchAPIView(APIView):
    model = None
    authentication_classes = (authentication.TokenAuthentication,)
    permission_classes = (permissions.IsAuthenticated, IpModelPermission)

    def __init__(self, **kwargs):
        assert self.model is not None
        super(SearchAPIView, self).__init__(**kwargs)

    def get(self, request, format=None):
        return Response({})