138 lines
5.0 KiB
Python
Executable File
138 lines
5.0 KiB
Python
Executable File
import logging
|
|
|
|
from tastypie import fields
|
|
from tastypie.constants import ALL, ALL_WITH_RELATIONS
|
|
from tastypie.resources import ModelResource, Resource, Bundle
|
|
from tastypie.utils import trailing_slash
|
|
from tastypie.authentication import ApiKeyAuthentication, Authentication
|
|
|
|
from django.conf.urls.defaults import url
|
|
from django.contrib import auth
|
|
from django.contrib.auth.models import User, AnonymousUser
|
|
from django.db.models import Q
|
|
|
|
from regluit.core import models
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class UserResource(ModelResource):
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = User.objects.all()
|
|
resource_name = 'user'
|
|
fields = ['username', 'first_name', 'last_name']
|
|
|
|
class EditionResource(ModelResource):
|
|
work = fields.ForeignKey('regluit.api.resources.WorkResource', 'work')
|
|
identifiers = fields.ToManyField('regluit.api.resources.IdentifierResource', 'identifiers')
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Edition.objects.all()
|
|
resource_name = 'edition'
|
|
filtering = {
|
|
"isbn_13": ALL, "identifiers": ALL_WITH_RELATIONS,
|
|
}
|
|
|
|
def build_filters(self, filters = None):
|
|
if filters is None:
|
|
filters = {}
|
|
for filter_expr, value in filters.items():
|
|
if filter_expr.startswith('isbn_13'):
|
|
filters['identifiers__type'] = 'isbn'
|
|
if len(filter_expr)>7:
|
|
filters['identifiers__value'+filter_expr[7:]] = value
|
|
else:
|
|
filters['identifiers__value'] = value
|
|
del filters[ filter_expr ]
|
|
return super(EditionResource, self).build_filters(filters)
|
|
|
|
class IdentifierResource(ModelResource):
|
|
work = fields.ForeignKey('regluit.api.resources.WorkResource', 'work')
|
|
edition = fields.ForeignKey('regluit.api.resources.EditionResource', 'edition')
|
|
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Identifier.objects.all()
|
|
resource_name = 'identifier'
|
|
filtering = {
|
|
"value": ALL, "type" : ALL,
|
|
}
|
|
|
|
|
|
|
|
class WorkResource(ModelResource):
|
|
editions = fields.ToManyField(EditionResource, 'editions')
|
|
identifiers = fields.ToManyField(IdentifierResource, 'identifiers')
|
|
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Work.objects.all()
|
|
resource_name = 'work'
|
|
filtering = {'editions': ALL_WITH_RELATIONS, 'id': ALL, 'identifiers': ALL_WITH_RELATIONS}
|
|
ordering = ['num_wishes']
|
|
|
|
class CampaignResource(ModelResource):
|
|
work = fields.ToOneField(WorkResource, 'work')
|
|
|
|
def alter_list_data_to_serialize(self, request, data):
|
|
"""
|
|
annotate the list of campaigns with information from the logged in
|
|
user. note: this isn't the user identified by the api username/api_key
|
|
it's the the user that client might be logged into unglue.it as.
|
|
"""
|
|
u = auth.get_user(request)
|
|
if isinstance(u, User):
|
|
data['meta']['logged_in_username'] = u.username
|
|
wishlist_work_ids = [w.id for w in u.wishlist.works.all()]
|
|
else:
|
|
data['meta']['logged_in_username'] = None
|
|
wishlist_work_ids = []
|
|
|
|
for o in data['objects']:
|
|
o.data['in_wishlist'] = o.obj.work.id in wishlist_work_ids
|
|
# there's probably a better place up the chain (where the Campaign objects are directly available) to grab the status
|
|
c = models.Campaign.objects.get(id=o.data["id"])
|
|
o.data['status'] = c.status
|
|
o.data['current_total'] = c.current_total
|
|
|
|
# TODO: add pledging information
|
|
return data
|
|
|
|
def alter_detail_data_to_serialize(self, request, obj):
|
|
c = models.Campaign.objects.get(id=obj.data["id"])
|
|
obj.data['status'] = c.status
|
|
obj.data['current_total'] = c.current_total
|
|
return obj
|
|
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Campaign.objects.all()
|
|
resource_name = 'campaign'
|
|
excludes = ['amazon_receiver', 'paypal_receiver']
|
|
filtering = {
|
|
"work": ALL_WITH_RELATIONS,
|
|
}
|
|
|
|
class AuthorResource(ModelResource):
|
|
works = fields.ToManyField(WorkResource, 'works')
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Author.objects.all()
|
|
resource_name = 'author'
|
|
|
|
class SubjectResource(ModelResource):
|
|
works = fields.ToManyField(WorkResource, 'works')
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Subject.objects.all()
|
|
resource_name = 'subject'
|
|
|
|
class WishlistResource(ModelResource):
|
|
user = fields.ToOneField(UserResource, 'user')
|
|
works = fields.ToManyField(WorkResource, 'works')
|
|
|
|
class Meta:
|
|
authentication = ApiKeyAuthentication()
|
|
queryset = models.Wishlist.objects.all()
|
|
resource_name = 'wishlist'
|