Add CSV deserialization

jpmorganchase · Nov 1, 2023 · 8c7a21c · 8c7a21c
1 parent 2839036
commit 8c7a21c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/py_adapter/plugin/_csv.py b/src/py_adapter/plugin/_csv.py
@@ -39,3 +39,18 @@ def serialize(obj: py_adapter.Basic, stream: BinaryIO) -> BinaryIO:
     stream.write(text_stream.read().encode("utf-8"))
     stream.flush()
     return stream
+
+
+@py_adapter.plugin.hook
+def deserialize(stream: BinaryIO) -> py_adapter.Basic:
+    """
+    Deserialize CSV data as an object of basic Python types
+
+    :param stream: File-like object to deserialize
+    """
+    import csv
+
+    text_stream = io.StringIO(stream.read().decode("utf-8"))
+    csv_reader = csv.DictReader(text_stream)
+    obj = next(csv_reader)
+    return obj
diff --git a/tests/test_csv.py b/tests/test_csv.py
@@ -40,3 +40,5 @@ def test_serialize_1_record(simple_ship):
     data = py_adapter.serialize(simple_ship, format="CSV")
     expected_lines = [b"name,build_on", b"Elvira,1970-12-31"]
     assert data.splitlines() == expected_lines
+    obj_out = py_adapter.deserialize(data, SimpleShip, format="CSV")
+    assert obj_out == simple_ship