parent
af12f80a26
commit
c2db336f74
10 changed files with 160 additions and 23 deletions
@ -0,0 +1,17 @@ |
||||
from rest_framework_filters import FilterSet, AllLookupsFilter, RelatedFilter |
||||
from .models import Location |
||||
|
||||
|
||||
class LocationFilterSet(FilterSet): |
||||
id = AllLookupsFilter() |
||||
children = RelatedFilter('common.filters.LocationFilterSet') |
||||
level = AllLookupsFilter() |
||||
lft = AllLookupsFilter() |
||||
name = AllLookupsFilter() |
||||
parent = RelatedFilter('common.filters.LocationFilterSet') |
||||
rght = AllLookupsFilter() |
||||
tree_id = AllLookupsFilter() |
||||
type = AllLookupsFilter() |
||||
|
||||
class Meta: |
||||
model = Location |
||||
@ -0,0 +1,20 @@ |
||||
from rest_framework.serializers import ModelSerializer |
||||
|
||||
from .models import Location |
||||
|
||||
|
||||
class LocationSerializer(ModelSerializer): |
||||
class Meta: |
||||
model = Location |
||||
|
||||
fields = ( |
||||
'id', |
||||
'name', |
||||
'children', |
||||
'parent', |
||||
'type', |
||||
'lft', |
||||
'rght', |
||||
'level', |
||||
'tree_id', |
||||
) |
||||
@ -1,14 +1,16 @@ |
||||
from django.http import HttpResponseForbidden |
||||
from django.contrib import messages |
||||
|
||||
|
||||
class CheckForUserMixin(object): |
||||
|
||||
def dispatch(self, request, *args, **kwargs): |
||||
# if request.user.is_authenticated(): |
||||
# pk = kwargs.get('pk') |
||||
# if pk: |
||||
# if request.user.pk != int(pk): |
||||
# return HttpResponseForbidden('403 Forbidden') |
||||
# else: |
||||
# return HttpResponseForbidden('403 Forbidden') |
||||
messages.info(request,str(request.user)) |
||||
if request.user.is_authenticated(): |
||||
pk = kwargs.get('pk') |
||||
if pk: |
||||
if request.user.pk != int(pk): |
||||
return HttpResponseForbidden('403 Forbidden') |
||||
else: |
||||
return HttpResponseForbidden('403 Forbidden') |
||||
return super().dispatch(request, *args, **kwargs) |
||||
|
||||
Loading…
Reference in new issue