diff --git a/snipts/api.py b/snipts/api.py index 15585c8..3380322 100644 --- a/snipts/api.py +++ b/snipts/api.py @@ -193,8 +193,9 @@ class PrivateSniptResource(ModelResource): user=request.user, **kwargs) def obj_update(self, bundle, request=None, **kwargs): + bundle.data['user'] = request.user - if type(bundle.data['tags']) == unicode: + if type(bundle.data['tags']) in (str, unicode): bundle.data['tags_list'] = bundle.data['tags'] else: bundle.data['tags_list'] = '' @@ -249,6 +250,7 @@ class PrivateSniptResource(ModelResource): tags = bundle.data.get('tags_list', []) if tags != '': bundle.obj.tags.set(*parse_tags(tags)) + class PrivateFavoriteResource(ModelResource): user = fields.ForeignKey(PrivateUserResource, 'user', full=True) snipt = fields.ForeignKey(PrivateSniptResource, 'snipt', full=False)