Add JSON to CSV parser

CHANGELOG.md

@@ -1,6 +1,12 @@
 CHANGELOG
 =========
 
+0.5.2
+-----
+
+ - Add json_to_csv converter
+
+
 0.5.1
 -----
 

py_partiql_parser/__init__.py

@@ -1,7 +1,7 @@
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 
 
 from ._internal.parser import DynamoDBStatementParser, S3SelectParser  # noqa
 from ._internal.json_parser import SelectEncoder  # noqa
-from ._internal.csv_converter import csv_to_json  # noqa
+from ._internal.csv_converter import csv_to_json, json_to_csv  # noqa
 from ._internal.utils import MissingVariable, QueryMetadata  # noqa

py_partiql_parser/_internal/csv_converter.py

@@ -1,5 +1,5 @@
 import json
-from typing import List
+from typing import Any, Dict, List
 
 
 def csv_to_json(input: str, headers_included: bool = False) -> str:
@@ -18,3 +18,12 @@ def csv_to_json(input: str, headers_included: bool = False) -> str:
     if output.endswith("\n"):
         output = output.rstrip("\n")
     return output
+
+
+def json_to_csv(input: List[Dict[str, Any]], field_delimiter, record_delimiter) -> str:
+    result = ""
+    for row in input:
+        result += (
+            field_delimiter.join([f"{v}" for v in row.values()]) + record_delimiter
+        )
+    return result

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "py-partiql-parser"
-version = "0.5.1"
+version = "0.5.2"
 description = "Pure Python PartiQL Parser"
 readme = "README.md"
 keywords = ["pypartiql", "parser"]

tests/test_csv_converter.py

@@ -1,4 +1,4 @@
-from py_partiql_parser._internal.csv_converter import csv_to_json
+from py_partiql_parser import csv_to_json, json_to_csv
 import json
 
 
@@ -37,3 +37,18 @@ def test_csv_to_json_with_headers() -> None:
 
     line2 = json.loads(lines[2])
     assert line2["City"] == "Seattle"
+
+
+def test_json_to_csv() -> None:
+    json_result = csv_to_json(input_csv)
+    lines = json_result.split("\n")
+
+    csv_result = json_to_csv(
+        [json.loads(lines[0])], field_delimiter=",", record_delimiter="\n"
+    )
+    assert csv_result == "Sam,(949) 555-6701,Irvine,Solutions Architect\n"
+
+    csv_result = json_to_csv(
+        [json.loads(lines[4])], field_delimiter=",", record_delimiter="\n"
+    )
+    assert csv_result == "Sean,(949) 555-6705,Chicago,Developer\n"