Untitled

from tastypie import fields, resources
from tastypie.api import Api
from models import *
from datetime import date


api = Api(api_name='v1')


class CinemaInfoResource(resources.ModelResource):
    class Meta:
        allowed_methods = ('get',)
        resource_name = 'cinemainfo'
        include_resource_uri = False
        queryset = CinemaInfo.objects.all()
        fields = (
            'obs',
        )
        filtering = {
            'cinema': resources.ALL_WITH_RELATIONS,
        }

api.register(CinemaInfoResource())


class CinemaResource(resources.ModelResource):
    info = fields.ToManyField('claquete.api.CinemaInfoResource',
        'cinemainfo_set', null=True, full=True)

    class Meta:
        allowed_methods = ('get',)
        resource_name = 'cinema'
        queryset = Cinema.objects.all()
        filtering = {
            'codigo': resources.ALL,
            'nome': resources.ALL,
            'cidade': resources.ALL,
            'uf': resources.ALL,
        }

    def build_filters(self, filters=None):
        built_filters = super(CinemaResource, self).build_filters(filters or {})
        built_filters.update(filter_by_cinema_distance(filters))
        return built_filters

api.register(CinemaResource())