django-tastypie: связывание ModelResource с ресурсом - PullRequest
Новая
       19

django-tastypie: связывание ModelResource с ресурсом

2 голосов
/ 06 октября 2011

Я сейчас пробую django-tastypie для разработки RESTful API. У меня проблема: 

# the RevisionObject retrieve commits info through pysvn
# This Resource is fully functionnal (RevisionObject code is not here)
class RevisionResource(Resource):
    id = fields.CharField(attribute='revision')
    description = fields.CharField(attribute='message')
    author = fields.CharField(attribute='author')
    changed_path = fields.ListField(attribute='changed_paths')

    class Meta:
        object_class = RevisionObject
        allowed_methods = ['get']
        resource_name = 'revision'

class RevisionToApplyResource(ModelResource):
    #### here's the problem
    revision = fields.ToManyField(RevisionResource, 'revision')
    ####
    class Meta:
        queryset = RevisionToApply.objects.all()

В моем models.py у меня есть: 

class RevisionToApply(models.Model):
    patch = models.ForeignKey(PatchRequest)
    revision = models.PositiveIntegerField()
    applied = models.BooleanField(default = False)

Моя проблема в том, что модели RevisionToApply (для django) используют для ревизии int.

Как я могу сказать tastypie использовать поле RevisionToApplyResource в качестве указателя на RevisionResource? Если ToXxxxField предназначен только для связи с моделями django, то какой идеальный момент для вставки ResourceObject?

спасибо. 

class NoForeignKeyToOneField(ToOneField):
    def dehydrate(self, bundle):
        try:
            obj_key = getattr(bundle.obj, self.attribute)
            foreign_obj = self.to_class().obj_get(pk=obj_key)
        except ObjectDoesNotExist:
            foreign_obj= None

        if not foreign_obj:
            if not self.null:
                raise ApiFieldError("The model '%r' has an empty attribute"
                   "'%s' and doesn't allow null value." % (bundle.obj,
                   self.attribute))
            return None

        self.fk_resource = self.get_related_resource(foreign_obj)
        fk_bundle = Bundle(obj=foreign_obj, request=bundle.request)
        return self.dehydrate_related(fk_bundle, self.fk_resource)

1 Ответ

2 голосов
/ 07 октября 2011

Вот как бы я это сделал.Взглянув на то, как работает класс ToOneField, вы заметите, что пара методов гидрат / дегидрат заботится о получении и настройке фактического связанного экземпляра.Подклассифицируя ToOneField и переопределяя эти два метода, вы можете получить преимущество автоматизированной обработки ресурсов Tastypie без фактического внешнего ключа.

(я имею в виду ToOneField вместо ToManyField, потому что в вашеммодель, данная RevisionToApply может указывать только на одну ревизию, кажется.)

Это будет выглядеть примерно так: 

class NoForeignKeyToOneField(ToOneField):

    def dehydrate(self, bundle):
        # Look up the related object manually
        try:
            obj_key = getattr(bundle.obj, self.attribute)
            ###
            # Get the revision object here. If you want to make it generic,
            # maybe pass a callable on __init__ that can be invoked here
            ###
            foreign_obj = revision_object
        except ObjectDoesNotExist:
            foreign_obj = None
        # The rest remains the same
        if not foreign_obj:
            if not self.null:
                raise ApiFieldError("The model '%r' has an empty attribute '%s' and doesn't allow a null value." % (bundle.obj, self.attribute))
            return None

        self.fk_resource = self.get_related_resource(foreign_obj)
        fk_bundle = Bundle(obj=foreign_obj, request=bundle.request)
        return self.dehydrate_related(fk_bundle, self.fk_resource)

    def hydrate(self, bundle):
        value = super(NoForeignKeyToOneField, self).hydrate(bundle)

        if value is None:
            return value

        # Here, don't return the full resource, only the primary key
        related_resource = self.build_related_resource(value, request=bundle.request)
        return related_resource.pk

А затем используйте этот тип поля в вашем ресурсе, а нечем основной ToOneField.Я не проверял это, но я верю, что подход здравый, простой, и он сделает работу.

...