Add a module to serialize a set of tabular data as a CSV file.

diff --git a/TODO b/TODO
index 7b293b04adaacd131793ecef454150f1694a0c95..3147467b2beb3a26ffef14bcf20dccabce1d24a0 100644 (file)
--- a/TODO
+++ b/TODO
@@ -12,9 +12,6 @@ Ideas / TODO:
   ser muy simple y genérico y en caso de agregar funcionalidad no sea necesario
   modificarlo.
 
-* Que las respuestas puedan ser listas, y que el encargado de "serializar" para
-  pasarlo por la red sea el daemon o una capa intermedia.
-
 Estas cosas quedan sujetas a necesitada y a definición del protocolo.
 Para mí lo ideal es que el protocolo de red sea igual que la consola del
 usuario, porque después de todo no va a ser más que eso, mandar comanditos.

diff --git a/serializer.py b/serializer.py
new file mode 100644 (file)
index 0000000..e76030b
--- /dev/null
+++ b/serializer.py
@@ -0,0 +1,85 @@
+# vim: set encoding=utf-8 et sw=4 sts=4 :
+
+import ucsv
+import seqtools
+
+r"UTF-8 encoded CSV serializer."
+
+def serialize(obj, output=None):
+    r"""serialize(obj[, output]) -> None/unicode string
+
+    Serialize the object obj to a UTF-8 encoded CSV string. If output
+    is not None, it's used as a file object to store the string. If it's
+    None, the string is returned.
+
+    obj is expected to be a sequence of sequences, i.e. a list of rows.
+    """
+    stringio = False
+    if output is None:
+        stringio = True
+        try:
+            from cStringIO import StringIO
+        except ImportError:
+            from StringIO import StringIO
+        output = StringIO()
+    ucsv.writer(output).writerows(seqtools.as_table(obj))
+    if stringio:
+        return output.getvalue()
+
+
+if __name__ == '__main__':
+
+    from seqtools import Sequence
+
+    class Host(Sequence):
+        r"""Host(name, ip, mac) -> Host instance :: Class representing a host.
+
+        name - Host name, should be a fully qualified name, but no checks are done.
+        ip - IP assigned to the hostname.
+        mac - MAC address to associate to the hostname.
+        """
+
+        def __init__(self, name, ip, mac):
+            r"Initialize Host object, see class documentation for details."
+            self.name = name
+            self.ip = ip
+            self.mac = mac
+
+        def as_tuple(self):
+            return (self.name, self.ip, self.mac)
+
+        def __unicode__(self):
+            return u'no anda'
+
+    print serialize(1)
+
+    print serialize("lala")
+
+    print serialize(u"lala")
+
+    print serialize([1, 2])
+
+    print serialize(["lala", "lala"])
+
+    print serialize([u"lala", u"lala"])
+
+    h = Host('name', 'ip', 'mac')
+    print serialize(h)
+
+    print serialize(dict(a=1, b=2))
+
+    print serialize([[1, 2, 3], [7, 4, 2]])
+
+    print serialize([["adfj", "jdfhk"], ["alskdjal", "1uas"]])
+
+    print serialize([[u"adfj", u"jdfhk"], [u"alskdjal", u"1uas"]])
+
+    print serialize([h, h])
+
+    import sys
+    print 'stdout:'
+    serialize([h, h], sys.stdout)
+    print
+
+    for i in h: print i
+