From 6af5067fcaa161489a0f9cfbd7b876c511bdacc3 Mon Sep 17 00:00:00 2001 From: Mark Riedesel Date: Wed, 30 Jan 2019 14:25:24 -0600 Subject: [PATCH] add option for record delimiter --- pyaccuwage/__init__.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/pyaccuwage/__init__.py b/pyaccuwage/__init__.py index a6775be..abb380f 100644 --- a/pyaccuwage/__init__.py +++ b/pyaccuwage/__init__.py @@ -38,7 +38,7 @@ def test_dump(): record.EmployeeWageRecord(), ] out = io.BytesIO() - dump(records, out) + dump(records, out, None) return out @@ -79,14 +79,16 @@ def loads(s): return load(fp) -def dump(records, fp): +def dump(records, fp, delim=None): for r in records: fp.write(r.output()) + if delim: + fp.write(delim) -def dumps(records): +def dumps(records, delim=None): import io fp = io.BytesIO() - dump(records, fp) + dump(records, fp, delim=delim) fp.seek(0) return fp.read()