add option for record delimiter

This commit is contained in:
Mark Riedesel 2019-01-30 14:25:24 -06:00
parent 250ca8d31f
commit 6af5067fca

View file

@ -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()