add option for record delimiter
This commit is contained in:
parent
250ca8d31f
commit
6af5067fca
1 changed files with 6 additions and 4 deletions
|
@ -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()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue