54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from tastypie.resources import ModelResource
|
|
from django.contrib.auth.models import User
|
|
from snipts.models import Comment, Snipt
|
|
from tastypie import fields
|
|
from taggit.models import Tag
|
|
|
|
class PublicUserResource(ModelResource):
|
|
class Meta:
|
|
queryset = User.objects.all()
|
|
resource_name = 'user'
|
|
fields = ['username',]
|
|
|
|
def dehydrate(self, bundle):
|
|
bundle.data['url'] = 'https://snipt.net/%s/' % bundle.obj.username
|
|
return bundle
|
|
|
|
class PublicCommentSniptResource(ModelResource):
|
|
class Meta:
|
|
queryset = Snipt.objects.filter(public=True).order_by('-created')
|
|
resource_name = 'snipt'
|
|
fields = ['id',]
|
|
|
|
class PublicTagResource(ModelResource):
|
|
class Meta:
|
|
queryset = Tag.objects.all()
|
|
resource_name = 'tag'
|
|
fields = ['name', '']
|
|
|
|
class PublicCommentResource(ModelResource):
|
|
user = fields.ForeignKey(PublicUserResource, 'user')
|
|
snipt = fields.ForeignKey(PublicCommentSniptResource, 'snipt')
|
|
|
|
class Meta:
|
|
queryset = Comment.objects.all()
|
|
resource_name = 'comment'
|
|
fields = ['user', 'snipt', 'comment', 'created', 'modified',]
|
|
|
|
class PublicSniptResource(ModelResource):
|
|
user = fields.ForeignKey(PublicUserResource, 'user')
|
|
comments = fields.ToManyField(PublicCommentResource, 'comment_set',related_name='comment')
|
|
tags = fields.ToManyField(PublicTagResource, 'tags', related_name='tag')
|
|
|
|
class Meta:
|
|
queryset = Snipt.objects.filter(public=True).order_by('-created')
|
|
resource_name = 'snipt'
|
|
fields = ['user', 'title', 'slug', 'tags', 'lexer', 'code', 'stylized'
|
|
'created', 'modified',]
|
|
|
|
def dehydrate(self, bundle):
|
|
bundle.data['url'] = bundle.obj.get_absolute_url()
|
|
bundle.data['user_username'] = bundle.obj.user.username
|
|
bundle.data['user_url'] = 'https://snipt.net/%s/' % bundle.obj.user.username
|
|
return bundle
|