scoobs
/
cryptools-mirror
mirror of https://github.com/scoobybejesus/cryptools.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Script: checked for duplicate account numbers.

This commit is contained in:
scoobybejesus 2020-11-27 12:57:12 -05:00
parent 29f84a30d3
commit 9a28bfbf64
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
clean_input_csv.py
View File

@ -24,6 +24,7 @@
import csv import csv
import re import re
import os import os
import sys
unedited = "DigiTrnx.csv" # To be replaced with a launch arg, presumably unedited = "DigiTrnx.csv" # To be replaced with a launch arg, presumably
@ -47,6 +48,13 @@ with open(unedited) as fin, open(stage1, 'a') as fout:
wtr.writerow(header3) wtr.writerow(header3)
wtr.writerow(header4) wtr.writerow(header4)
# First, double check there are no account number duplicates
for i, val in enumerate(header):
if val != "":
if header.count(val) > 1:
print("### There is a duplicate account number (" + val +"). Please fix and re-run. ###")
sys.exit()
for row in rdr: for row in rdr:
if row[0] == "" or row[1] == "": if row[0] == "" or row[1] == "":
pass pass
@ -66,12 +74,12 @@ stage2 = "stage2.csv"
## CSV input file. ## CSV input file.
with open(stage1) as fin, open(stage2, 'a') as fout: with open(stage1) as fin, open(stage2, 'a') as fout:
rdr = csv.reader(fin) rdr = csv.reader(fin)
wtr = csv.writer(fout) wtr = csv.writer(fout)
header = next(rdr) header = next(rdr)
header2 = next(rdr) header2 = next(rdr)
header3 = next(rdr) header3 = next(rdr)
header4 = next(rdr) header4 = next(rdr)
colListKept = [] colListKept = []