mirror of
https://github.com/pinry/pinry.git
synced 2025-11-13 08:35:41 +01:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from tastypie.resources import ModelResource
|
|
from tastypie import fields
|
|
from tastypie.authorization import DjangoAuthorization
|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
from pinry.pins.models import Pin
|
|
|
|
|
|
class UserResource(ModelResource):
|
|
class Meta:
|
|
queryset = User.objects.all()
|
|
resource_name = 'user'
|
|
excludes = ['password', 'is_superuser', 'first_name',
|
|
'last_name', 'is_active', 'is_staff', 'last_login', 'date_joined']
|
|
include_resource_uri = False
|
|
|
|
|
|
class PinResource(ModelResource):
|
|
tags = fields.ListField()
|
|
submitter = fields.ForeignKey(UserResource, 'submitter', full=True)
|
|
|
|
class Meta:
|
|
queryset = Pin.objects.all()
|
|
resource_name = 'pin'
|
|
include_resource_uri = False
|
|
filtering = {
|
|
'published': ['gt'],
|
|
'submitter': ['exact']
|
|
}
|
|
authorization = DjangoAuthorization()
|
|
|
|
def build_filters(self, filters=None):
|
|
if filters is None:
|
|
filters = {}
|
|
|
|
orm_filters = super(PinResource, self).build_filters(filters)
|
|
|
|
if 'tag' in filters:
|
|
orm_filters['tags__name__in'] = filters['tag'].split(',')
|
|
|
|
return orm_filters
|
|
|
|
def dehydrate_tags(self, bundle):
|
|
return map(str, bundle.obj.tags.all())
|
|
|
|
def save_m2m(self, bundle):
|
|
tags = bundle.data.get('tags', [])
|
|
bundle.obj.tags.set(*tags)
|
|
return super(PinResource, self).save_m2m(bundle)
|