From 75504e2c9539e977a393134b16da96449d15eda5 Mon Sep 17 00:00:00 2001
From: iquasere <maildosequeira@gmail.com>
Date: Wed, 3 Jan 2024 08:54:41 +0000
Subject: [PATCH] Now I get it

---
 keggcharter.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/keggcharter.py b/keggcharter.py
index 26fb6bb..b0c0426 100644
--- a/keggcharter.py
+++ b/keggcharter.py
@@ -183,8 +183,8 @@ def read_input_file(args: argparse.Namespace) -> pd.DataFrame:
         "cog_column": (r"^COG\d{4}$", 'COGXXXX')
     }
     for col in ["kegg_column", "ko_column", "ec_column", "cog_column"]:
-        if getattr(args, col) is not None:      # the "if getattr(args, col)" does not work here, returns True
-            if result[col].apply(bad_value, patterns[col][0]).sum() > 0:
+        if getattr(args, col):
+            if result[getattr(args, col)].apply(bad_value, patterns[col][0]).sum() > 0:
                 sys.exit(f"Invalid format for '{getattr(args, col)}' column. Use only '{patterns[col][1]}' format, "
                          f"optionally separated by commas.")
     return result