Changeset 20 for trunk/lib/syck/dumpers.py

Timestamp:

08/12/05 16:15:19 (8 years ago)

Author:

xi

Message:

syck.dump() is implemented.

File:

• trunk/lib/syck/dumpers.py (modified) (4 diffs)

trunk/lib/syck/dumpers.py

@@ -10 +10 @@
     'emit', 'dump', 'emit_documents', 'dump_documents']
 
+INF = 1e300000
+NEGINF = -INF
+NAN = INF/INF
+
+
 class GenericDumper(_syck.Emitter):
 
@@ -16 +21 @@
 
     def _convert(self, object, object_to_node):
-        if id(object) in object_to_node:
+        if id(object) in object_to_node and self.allow_aliases(object):
             return object_to_node[id(object)]
         node = self.represent(object)
@@ -29 +34 @@
                 node.value[self._convert(key, object_to_node)] =    \
                         self._convert(value, object_to_node)
+#        # Workaround against a Syck bug:
+#        if node.kind == 'scalar' and node.style not in ['1quote', '2quote'] \
+#                and node.value and node.value[-1] in [' ', '\t']:
+#            node.style = '2quote'
         return node
 
@@ -39 +48 @@
             return _syck.Scalar(str(object), tag="tag:yaml.org,2002:str")
 
+    def allow_aliases(self, object):
+        return True
+
 class Dumper(GenericDumper):
 
+    def __init__(self, *args, **kwds):
+        super(Dumper, self).__init__(*args, **kwds)
+
     def represent(self, object):
+        for object_type in type(object).__mro__:
+            if object_type.__module__ == '__builtin__':
+                name = object_type.__name__
+            else:
+                name = '%s.%s' % (object_type.__module__, object_type.__name__)
+            method = 'represent_%s' % name.replace('.', '_')
+            if hasattr(self, method):
+                return getattr(self, method)(object)
         return super(Dumper, self).represent(object)
+
+    def represent_object(self, object):
+        return _syck.Scalar(repr(object), tag="tag:yaml.org,2002:str")
+
+    def represent_NoneType(self, object):
+        return _syck.Scalar('~', tag="tag:yaml.org,2002:null")
+
+    def represent_bool(self, object):
+        return _syck.Scalar(repr(object), tag="tag:yaml.org,2002:bool")
+
+    def represent_str(self, object):
+        return _syck.Scalar(str(object), tag="tag:yaml.org,2002:str")
+
+    def represent_list(self, object):
+        return _syck.Seq(object[:], tag="tag:yaml.org,2002:seq")
+
+    def represent_dict(self, object):
+        return _syck.Map(object.copy(), tag="tag:yaml.org,2002:map")
+
+    def represent_int(self, object):
+        return _syck.Scalar(repr(object), tag="tag:yaml.org,2002:int")
+
+    def represent_int(self, object):
+        return _syck.Scalar(repr(object), tag="tag:yaml.org,2002:int")
+
+    def represent_float(self, object):
+        value = repr(object)
+        if value == repr(INF):
+            value = '.inf'
+        elif value == repr(NEGINF):
+            value = '-.inf'
+        elif value == repr(NAN):
+            value = '.nan'
+        return _syck.Scalar(value, tag="tag:yaml.org,2002:float")
+
+    def represent_sets_Set(self, object):
+        return _syck.Seq(list(object), tag="tag:yaml.org,2002:set")
+
+    def represent_datetime_datetime(self, object):
+        return _syck.Scalar(object.isoformat(), tag="tag:yaml.org,2002:timestamp")
+
+    def allow_aliases(self, object):
+        if object is None or type(object) in [int, bool, float]:
+            return False
+        if type(object) is str and (not object or object.isalnum()):
+            return False
+        return True
 
 def emit(node, output=None, **parameters):