Changeset 120:e6e7a471146a

Timestamp:

Sep 2, 2013, 1:31:19 PM (7 years ago)

Author:

Stefano Rivera <stefano@…>

Branch:

default

rebase_source:

df5e75e21bb3a94ce8890b07c732263bfe57c4d0

Message:

Support numeric types

Location:

nagslang

Files:

• tests/test_yamlish.py (modified) (1 diff)
• yamlish.py (modified) (4 diffs)

nagslang/tests/test_yamlish.py

@@ -87 +87 @@
         self.roundtrip({'foo': [True, False, None]})
 
+    def test_numeric(self):
+        self.roundtrip({'foo': [1, 2.0, -1, -2.5]})
+
 
 class TestFromPyYAML(TestRoundTrip):

nagslang/yamlish.py

@@ -5 +5 @@
  * lists, dicts,
  * single line strings,
+ * ints, floats,
  * True, False, and None
 '''
@@ -33 +34 @@
 
     def _dump(self, data, indent=0):
-        for type_ in (list, dict, basestring):
+        for type_ in (list, dict, basestring, int, float):
             if isinstance(data, type_):
                 f = getattr(self, '_dump_%s' % type_.__name__)
@@ -67 +68 @@
             data = "'%s'" % data
         return [' ' * indent + data]
+
+    def _dump_int(self, data, indent):
+        return ['%s%i' % (' ' * indent, data)]
+
+    def _dump_float(self, data, indent):
+        return ['%s%f' % (' ' * indent, data)]
 
     def _dump_literal(self, data, indent):
@@ -180 +187 @@
         if value == 'null':
             return None
+        for type_ in (int, float):
+            try:
+                return type_(value)
+            except ValueError:
+                pass
         return value