api.py

from tastypie import fields
from tastypie.resources import ModelResource
from tastypie.constants import ALL, ALL_WITH_RELATIONS

from guardian.shortcuts import get_objects_for_user

from django.contrib.auth import get_user_model

from geonode.base.models import TopicCategory
from geonode.layers.models import Layer
from geonode.maps.models import Map


class TopicCategoryResource(ModelResource):

    class Meta:
        queryset = TopicCategory.objects.all()
        resource_name = 'categories'
        filtering = {'identifier': ALL,
                    }


class OwnerResource(ModelResource):
    """Owners api, lighter and faster version of the profiles api"""
    class Meta:
        queryset = get_user_model().objects.exclude(username='AnonymousUser')
        resource_name = 'owners'
        allowed_methods = ['get']
        ordering = ['username', 'date_joined']
        excludes = ['is_staff', 'password', 'is_superuser',
                    'is_active', 'last_login']
        filtering = {
            'username': ALL,
        }


class CommonMetaApi:
    allowed_methods = ['get']
    filtering = {'title': ALL,
                 # 'keywords': ALL_WITH_RELATIONS,
                 # 'tkeywords': ALL_WITH_RELATIONS,
                 # 'regions': ALL_WITH_RELATIONS,
                 'category': ALL_WITH_RELATIONS,
                 # 'group': ALL_WITH_RELATIONS,
                 'owner': ALL_WITH_RELATIONS,
                 # 'date': ALL,
                 }
    ordering = ['date', 'title', 'popular_count']


class CommonModelApi(ModelResource):
    category = fields.ToOneField(
        TopicCategoryResource,
        'category',
        null=True,
        full=True)
    owner = fields.ToOneField(OwnerResource, 'owner', full=True)

    def get_object_list(self, request):
        visible_resources = get_objects_for_user(request.user, 'base.view_resourcebase')
        visible_resources_ids = visible_resources.values_list('id', flat=True)
        return super(CommonModelApi, self).get_object_list(request).filter(pk__in=visible_resources_ids)


class LayerResource(CommonModelApi):

    class Meta(CommonMetaApi):
        queryset = Layer.objects.all().order_by('-popular_count')
        resource_name = 'layers'


class MapResource(CommonModelApi):

    detail_url = fields.CharField(null=False)

    class Meta(CommonMetaApi):
        queryset = Map.objects.all().order_by('-popular_count')
        resource_name = 'maps'

    def dehydrate_detail_url(self, bundle):
        if bundle.obj.urlsuffix:
            return '/maps/%s' % bundle.obj.urlsuffix
        else:
            return bundle.obj.detail_url