Feat: additional column check

1 files changed, 13 insertions, 8 deletions

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