From: Silvio Rhatto <rhatto@riseup.net>
Date: Fri, 29 Jan 2021 00:47:42 +0000 (-0300)
Subject: Feat: additional column check
X-Git-Url: https://gitweb.fluxo.info/?a=commitdiff_plain;h=5b6dd528366b6792e16099c628066931d8762134;p=csv-hasher.git

Feat: additional column check
---

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