-
Notifications
You must be signed in to change notification settings - Fork 0
/
transpose.py
74 lines (73 loc) · 2.73 KB
/
transpose.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import itertools
import sys
import os
import re
from csv import reader, writer
if sys.argv[1] == '--help': #activiate help mode
#print ("-----Help Mode-----\n")
print("Usage: transpose.(py|exe) FILE DELIMITER_FLAG [CUSTOM_DELIMITER]")
# standard argument <required argument> [optional option] curly braces {default values} parenthesis (miscellaneous info)
# syntax taken from: https://stackoverflow.com/questions/21503865/how-to-denote-that-a-command-line-argument-is-optional-when-printing-usage
print("")
print("Transpose a given .csv/.dsv file's X and Y axis")
print("")
print("Supported Switches/Arguments.")
print(" -c, --comma\t comma delimiter \",\"")
print(" -p, --pipe\t pipe delimiter \"|\"")
print(" -t, --tab\t tab delimiter \"\t\"")
print(" -h, --hash\t comma delimiter \"#\"")
print(" -r, --caret\t caret (hat) delimiter \"^\"")
print(" -x, --custom\t custom delimiter, follow with [CUSTOM_DELIMITER]")
print(" --help\t display this help message")
print("")
print("Usage Examples:")
print(" comma seperated file called example.csv:")
print("\ttranspose example.csv -c")
print("")
print(" hash seperated file called hashes.hsv:")
print("\ttranspose hashes.hsv -h")
print("")
print(" custom hat-and-pipe seperated extract.mi:")
print("\ttranspose extract.mi -x ^|")
print("")
print("")
print("Adam Bonner, 2020, https://github.com/adambonneruk/transpose-dsv")
print("")
else:
#print("-----Transpose Mode-----\n")
if len(sys.argv) > 2: #3rd+ argument present
inputFile = sys.argv[1]
outputFile, outputFileExtension = os.path.splitext(inputFile)
outputFile = outputFile+"_transposed"+outputFileExtension
if re.search("^(-x|--custom)$", sys.argv[2]):
#print("-----Custom Mode-----\n")
delimiter = sys.argv[3]
elif re.search("^(-c|--comma)$", sys.argv[2]):
#print("-----Comma Mode-----\n")
delimiter = ","
elif re.search("^(-p|--pipe)$", sys.argv[2]):
#print("-----Pipe Mode-----\n")
delimiter = "|"
elif re.search("^(-t|--tab)$", sys.argv[2]):
#print("-----Tab Mode-----\n")
delimiter = "\t"
elif re.search("^(-h|--hash)$", sys.argv[2]):
#print("-----Hash Mode-----\n")
delimiter = "#"
elif re.search("^(-r|--caret)$", sys.argv[2]):
#print("-----Caret Mode-----\n")
delimiter = "^"
else:
print("-----Delim Error-----\n")
delimiter = "ERROR"
#print('Input File: "' + str(inputFile) + '"')
#print('Output File: "' + str(outputFile) + '"')
#print('Delimeter: "' + str(delimiter) + '"')
if delimiter != "" and delimiter != "ERROR":
with open(inputFile) as f:
with open(outputFile, 'w', newline="\n", encoding="utf-8") as fw:
writer(fw, delimiter=delimiter).writerows(zip(*reader(f, delimiter=delimiter)))
else:
print("error")
else:
print("too few arguments")