mirror of
https://github.com/pinry/pinry.git
synced 2025-11-14 00:55:43 +01:00
e2a38f8a10
There has been some refactoring going on in the pinry.pins.models module. The upload_to code has been refactored into its own function, images have been moved to their own models - otherwise the number of fields in the Pin model would skyrocket. Also ModelManagers have been written to move image fetching and generating outside of models.
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
from tastypie.resources import ModelResource
|
|
from tastypie import fields
|
|
from tastypie.authorization import DjangoAuthorization
|
|
|
|
from pinry.core.models import User
|
|
|
|
from pinry.pins.models import Pin
|
|
|
|
|
|
class UserResource(ModelResource):
|
|
gravatar = fields.CharField()
|
|
|
|
def dehydrate_gravatar(self, bundle):
|
|
return bundle.obj.gravatar
|
|
|
|
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):
|
|
images = fields.DictField()
|
|
tags = fields.ListField()
|
|
submitter = fields.ForeignKey(UserResource, 'submitter', full=True)
|
|
|
|
def dehydrate_images(self, bundle):
|
|
images = {}
|
|
for type in ['standard', 'thumbnail', 'original']:
|
|
image_obj = getattr(bundle.obj, type, None)
|
|
images[type] = {'url': image_obj.image.url, 'width': image_obj.width, 'height': image_obj.height}
|
|
return images
|
|
|
|
class Meta:
|
|
queryset = Pin.objects.all()
|
|
resource_name = 'pin'
|
|
include_resource_uri = False
|
|
filtering = {
|
|
'published': ['gt'],
|
|
'submitter': ['exact']
|
|
}
|
|
fields = ['submitter', 'tags', 'published', 'description', 'url']
|
|
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)
|