# Note that `DisplayCodec` is deliberately omitted from the documentation,

# as it is considered an implementation detail.

# It may move into a utility function in the future.

from __future__ import unicode_literals

from coreapi.codecs.base import BaseCodec

from coreapi.document import Document, Link, Array, Object, Error, Field

def _to_repr(node):

if isinstance(node, Document):

content = ', '.join([

'%s: %s' % (repr(key), _to_repr(value))

for key, value in node.items()

])

return 'Document(url=%s, title=%s, content={%s})' % (

repr(node.url), repr(node.title), content

)

elif isinstance(node, Error):

content = ', '.join([

'%s: %s' % (repr(key), _to_repr(value))

for key, value in node.items()

])

return 'Error(title=%s, content={%s})' % (

repr(node.title), content

)

elif isinstance(node, Object):

return '{%s}' % ', '.join([

'%s: %s' % (repr(key), _to_repr(value))

for key, value in node.items()

])

elif isinstance(node, Array):

return '[%s]' % ', '.join([

_to_repr(value) for value in node

])

elif isinstance(node, Link):

args = "url=%s" % repr(node.url)

if node.action:

args += ", action=%s" % repr(node.action)

if node.encoding:

args += ", encoding=%s" % repr(node.encoding)

if node.transform:

args += ", transform=%s" % repr(node.transform)

if node.description:

args += ", description=%s" % repr(node.description)

if node.fields:

fields_repr = ', '.join(_to_repr(item) for item in node.fields)

args += ", fields=[%s]" % fields_repr

return "Link(%s)" % args

elif isinstance(node, Field):

args = repr(node.name)

if not node.required and not node.location:

return args

if node.required:

args += ', required=True'

if node.location:

args += ', location=%s' % repr(node.location)

if node.schema:

args += ', schema=%s' % repr(node.schema)

return 'Field(%s)' % args

return repr(node)

class PythonCodec(BaseCodec):

"""

A Python representation of a Document, for use with '__repr__'.

"""

media_type = 'text/python'

def encode(self, document, **options):

# Object and Array only have the class name wrapper if they

# are the outermost element.

if isinstance(document, Object):

return 'Object(%s)' % _to_repr(document)

elif isinstance(document, Array):

return 'Array(%s)' % _to_repr(document)

return _to_repr(document)