|
@@ -1,8 +1,10 @@
|
1
|
1
|
from argparse import ArgumentParser, FileType
|
2
|
2
|
from sys import stdout
|
|
3
|
+from signal import signal, SIGPIPE, SIG_DFL
|
3
|
4
|
import pandas as p
|
4
|
5
|
|
5
|
6
|
def main():
|
|
7
|
+ signal(SIGPIPE, SIG_DFL) # Prevent broken pipe errors when piping to less or head
|
6
|
8
|
parser = ArgumentParser(description='Transform a "pickle" to a csv file')
|
7
|
9
|
parser.add_argument(dest="input", help = "input pickle path", type = FileType('rb'))
|
8
|
10
|
parser.add_argument(dest="output", nargs="?", help = "output csv path", type = FileType('w'))
|
|
@@ -27,6 +29,7 @@ def main():
|
27
|
29
|
|
28
|
30
|
ptc.to_csv(args.output, index = args.rownames)
|
29
|
31
|
|
|
32
|
+
|
30
|
33
|
args.input.close()
|
31
|
34
|
args.output.close()
|
32
|
35
|
|