-
Notifications
You must be signed in to change notification settings - Fork 0
/
scourgify_beta
50 lines (40 loc) · 1.25 KB
/
scourgify_beta
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
import csv
import sys
def check_argvs():
if len(sys.argv) > 3:
print('too many args')
sys.exit(1)
elif len(sys.argv) <3:
print('too few args')
sys.exit(1)
elif sys.argv[2][-4:] != '.csv':
print('not a csv file')
sys.exit(1)
def clean(bad_file:str, cleaned_file:str):
try:
with open(bad_file, 'r') as bf:
reader = csv.DictReader(bf)
data = []
for row in reader:
last,first = row['name'].split(', ')
new_data = {
'first': first,
'last': last,
'house': row['house']
}
data.append(new_data)
with open(cleaned_file, 'w') as cf:
headers = ['first','last','house']
writer = csv.DictWriter(cf, fieldnames=headers)
writer.writeheader()
for row in data:
writer.writerow(row)
print(row)
except FileNotFoundError:
print('file does not exist')
sys.exit(1)
def main():
check_argvs()
clean(sys.argv[1],sys.argv[2])
if __name__ == '__main__':
main()