|
@@ -1,13 +1,16 @@
|
1
|
|
-from argparse import ArgumentParser
|
|
1
|
+from argparse import ArgumentParser, FileType
|
2
|
2
|
from sys import stdout
|
3
|
3
|
import pandas as p
|
4
|
4
|
|
5
|
5
|
def main():
|
6
|
6
|
parser = ArgumentParser(description='Transform a "pickle" to a csv file')
|
7
|
|
- parser.add_argument("-i", "--input", dest="input", help = "input pickle path", metavar="PATH", required = True)
|
8
|
|
- parser.add_argument("-o", "--output", dest="output", help = "output csv path", metavar="PATH")
|
9
|
|
- parser.add_argument("-r", "--row-names", dest = "rownames", help = "include row names in csv; if ommitted prints to stdout", action = "store_true")
|
10
|
|
- parser.add_argument("-v", "--verbose", dest = "verbose", action ="store_true")
|
|
7
|
+ parser.add_argument(dest="input", help = "input pickle path", type = FileType('rb'))
|
|
8
|
+ parser.add_argument(dest="output", nargs="?", help = "output csv path", type = FileType('w'))
|
|
9
|
+ parser.add_argument("-r", "--row-names", dest = "rownames", help = "include row names in csv; "
|
|
10
|
+ "needed to faithfully reproduce some dataframes. "
|
|
11
|
+ "Use -v to check information about dataframe", action = "store_true")
|
|
12
|
+ parser.add_argument("-v", "--verbose", dest = "verbose", action ="store_true",
|
|
13
|
+ help="print information about dataframe and process")
|
11
|
14
|
args = parser.parse_args()
|
12
|
15
|
|
13
|
16
|
if args.output is None:
|
|
@@ -24,6 +27,9 @@ def main():
|
24
|
27
|
|
25
|
28
|
ptc.to_csv(args.output, index = args.rownames)
|
26
|
29
|
|
|
30
|
+ args.input.close()
|
|
31
|
+ args.output.close()
|
|
32
|
+
|
27
|
33
|
|
28
|
34
|
if __name__ == "__main__":
|
29
|
35
|
main()
|