From 5b6dd528366b6792e16099c628066931d8762134 Mon Sep 17 00:00:00 2001 From: Silvio Rhatto Date: Thu, 28 Jan 2021 21:47:42 -0300 Subject: Feat: additional column check --- csv-hasher.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) (limited to 'csv-hasher.py') diff --git a/csv-hasher.py b/csv-hasher.py index 6415c7b..bdd9950 100755 --- a/csv-hasher.py +++ b/csv-hasher.py @@ -92,14 +92,6 @@ class CsvHasher: print('CSV file is too small.') exit (1) - # Start with and empty file - try: - with open(outfile, 'w') as f: - f.truncate(0) - except IOError: - print('Error writing to ' + outfile) - exit(1) - # Holds columns definition columns = None @@ -109,6 +101,19 @@ class CsvHasher: columns = chunk.columns break + # Check for the column + if self.args.colname[0] not in columns: + print('Column not found: ' + self.args.colname[0]) + exit (1) + + # Start with an empty file + try: + with open(outfile, 'w') as f: + f.truncate(0) + except IOError: + print('Error writing to ' + outfile) + exit(1) + # Initialize progress bar progress_bar = tqdm(total=nlines) if self.args.progress else False -- cgit v1.2.3