from tastypie import fields
from tastypie.resources import ModelResource
from tastypie.authentication import BasicAuthentication
from tastypie.authorization import Authorization

from haystack.query import SearchQuerySet

from nodes.models import Node
from nodes.models import Site


class SiteResource(ModelResource):
    class Meta:
        queryset = Site.objects.all()
        always_return_data = True
        authentication = BasicAuthentication()
        authorization = Authorization()

class NodeResource(ModelResource):
    site = fields.ForeignKey(SiteResource, 'site', null=True)

    class Meta:
        queryset = Node.objects.all()
        always_return_data = True
        authentication = BasicAuthentication()
        authorization = Authorization()

    def build_filters(self, filters=None):
        if filters is None:
            filters = {}

        orm_filters = super(NodeResource, self).build_filters(filters)

        if "q" in filters and filters["q"]:
            sqs = SearchQuerySet().auto_query(filters['q'])
            orm_filters = {"pk__in": [ i.pk for i in sqs ]}

        return orm_filters