embee/pyaccuwage
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add option for record delimiter

Browse source
This commit is contained in:
Mark Riedesel 2019-01-30 14:25:24 -06:00
parent 250ca8d31f
commit 6af5067fca
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pyaccuwage/__init__.py
View file

@ -38,7 +38,7 @@ def test_dump():
record.EmployeeWageRecord(), record.EmployeeWageRecord(),
] ]
out = io.BytesIO() out = io.BytesIO()
dump(records, out) dump(records, out, None)
return out return out
@ -79,14 +79,16 @@ def loads(s):
return load(fp) return load(fp)
def dump(records, fp): def dump(records, fp, delim=None):
for r in records: for r in records:
fp.write(r.output()) fp.write(r.output())
if delim:
fp.write(delim)
def dumps(records): def dumps(records, delim=None):
import io import io
fp = io.BytesIO() fp = io.BytesIO()
dump(records, fp) dump(records, fp, delim=delim)
fp.seek(0) fp.seek(0)
return fp.read() return fp.read()