diff options
Diffstat (limited to 'coip/apps/opensocial/serializer.py')
| -rw-r--r-- | coip/apps/opensocial/serializer.py | 130 |
1 files changed, 72 insertions, 58 deletions
diff --git a/coip/apps/opensocial/serializer.py b/coip/apps/opensocial/serializer.py index 14ac2df..df0dcc3 100644 --- a/coip/apps/opensocial/serializer.py +++ b/coip/apps/opensocial/serializer.py @@ -4,87 +4,101 @@ Created on Jun 19, 2011 @author: leifj ''' from tastypie.serializers import Serializer, get_type_string -from lxml.etree import Element from tastypie.bundle import Bundle -from tastypie.fields import ApiField, ToOneField, ToManyField +from django.utils import simplejson from django.utils.encoding import force_unicode +import string import logging from pprint import pformat +from django.core.exceptions import ImproperlyConfigured +from django.core.serializers import json + +try: + import lxml + from lxml.etree import parse as parse_xml + from lxml.etree import Element, tostring +except ImportError: + lxml = None class OpenSocialSerializer(Serializer): def __init__(self, formats=None, content_types=None, datetime_formatting=None): super(OpenSocialSerializer,self).__init__(formats,content_types,datetime_formatting) - def to_etree(self, data, options=None, name=None, depth=0): + def name_data(self,data): + name = 'unknown' + + if hasattr(data,"resource_name"): + name = data.resource_name() + + if isinstance(data,Bundle): + name = self.name_data(data.obj) + elif hasattr(data,"__class__"): + name = string.lower(data.__class__.__name__) - #logging.debug("--------") - #logging.debug(name) - #logging.debug(depth) - #logging.debug(pformat(data)) + return name + + def to_json(self, data, options=None): + """ + Given some Python data, produces JSON output. + """ + options = options or {} + data = self.to_simple(data, options) + return simplejson.dumps(data['response'], cls=json.DjangoJSONEncoder, sort_keys=True) + + def to_xml(self, data, options=None): + """ + Given some Python data, produces XML output. + """ + options = options or {} + + if lxml is None: + raise ImproperlyConfigured("Usage of the XML aspects requires lxml.") + + etree = self.to_etree(data, options) + etree.set("xmlns","http://ns.opensocial.org/2008/opensocial") + + return tostring(etree, xml_declaration=True, encoding='utf-8') + + def to_etree(self, data, options=None, name=None, depth=0, parent=None): """ Given some data, converts that data to an ``etree.Element`` suitable for use in the XML output. """ - + + if parent is not None: + logging.debug("+++++++++++ %d %s %s" % (depth,name,tostring(parent))) + else: + logging.debug("+++++++++++ %d %s <no parent>" % (depth,name)) + logging.debug(pformat(data)) + + if parent is None: + parent = Element(name or self.name_data(data)) + if isinstance(data, (list, tuple)): - element = Element(name or 'objects') - if name: - element = Element(name) - #element.set('type', 'list') - else: - element = Element('objects') for item in data: - element.append(self.to_etree(item, options, depth=depth+1)) + element = Element(name or 'item') + self.to_etree(item, options, depth=depth+1,parent=element) + parent.append(element) elif isinstance(data, dict): - if depth == 0: - element = Element(name or 'response',attrib={'xmlns': "http://ns.opensocial.org/2008/opensocial"} ) + if len(data) == 1 and depth == 0: + for (key,value) in data.iteritems(): + parent = self.to_etree(value, options, name=key, depth=depth+1,parent=None) else: - element = Element(name or 'object') - element.set('type', 'hash') - - if data.has_key('objects'): - if len(data['objects']) == 1: - return self.to_etree(data['objects'][0], options, name=None, depth=depth) - else: - for v in data['objects']: - element.append(self.to_etree(v, options, name='entry', depth=depth+1)) - else: - for (key, value) in data.iteritems(): - keyname = key - if keyname == 'user': - keyname = 'person' - element.append(self.to_etree(value, options, name=keyname, depth=depth+1)) - elif isinstance(data, Bundle): - element = Element(name or 'object') + for (key,value) in data.iteritems(): + parent.append(self.to_etree(value, options, name=key, depth=depth+1,parent=parent)) + elif isinstance(data, Bundle): + element = Element(self.name_data(data)) for field_name, field_object in data.data.items(): - keyname = field_name - if keyname == 'user': - keyname = 'person' - element.append(self.to_etree(field_object, options, name=keyname, depth=depth+1)) - elif isinstance(data, ApiField): - if isinstance(data, ToOneField): - if data.full: - return self.to_etree(data.fk_resource, options, name, depth+1) - else: - return self.to_etree(data.value, options, name, depth+1) - elif isinstance(data, ToManyField): - if data.full: - element = Element(name or 'objects') - for bundle in data.m2m_bundles: - element.append(self.to_etree(bundle, options, bundle.resource_name, depth+1)) - else: - element = Element(name or 'objects') - for value in data.value: - element.append(self.to_etree(value, options, name, depth=depth+1)) - else: - return self.to_etree(data.value, options, name) + self.to_etree(field_object, options, name=field_name, depth=depth+1, parent=element) + parent.append(element) else: element = Element(name or 'value') simple_data = self.to_simple(data, options) data_type = get_type_string(simple_data) - if data_type != 'string': - element.set('type', get_type_string(simple_data)) + #if data_type != 'string': + # element.set('type', get_type_string(simple_data)) if data_type != 'null': element.text = force_unicode(simple_data) - return element
\ No newline at end of file + parent.append(element) + return parent
\ No newline at end of file |