arno/isso
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix IOError when piping wynaut-export to head(1)

Browse Source
This commit is contained in:
Martin Zimmermann 2013-12-25 11:46:55 +01:00
parent d81a6f6b5c
commit 67dcf43ac6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
wynaut/export.py
View File

@ -49,14 +49,18 @@ def csv(db, threads=True, comments=False):
fp.seek(0) fp.seek(0)
for line in fp: for line in fp:
sys.stdout.write(line.encode("utf-8")) try:
sys.stdout.write(line.encode("utf-8"))
sys.stdout.flush()
except IOError: # head(1) reads from stdout then closes it.
break
def main(): def main():
parser = get_parser("export to various formats") parser = get_parser("export to various formats")
parser.add_argument("-t", "--to", dest="type", choices=["csv"], parser.add_argument("-t", "--to", dest="type", choices=["csv"],
help="export format") help="export format", required=True)
group = parser.add_mutually_exclusive_group() group = parser.add_mutually_exclusive_group()
group.add_argument("--threads", action="store_true", group.add_argument("--threads", action="store_true",