diff --git a/csv_merge.py b/csv_merge.py index 46d67b6..fa7cc4c 100644 --- a/csv_merge.py +++ b/csv_merge.py @@ -13,9 +13,12 @@ def process(directory): dirname = os.path.basename(subdir) if dirname == '': dirname = os.path.basename(os.path.dirname(subdir)) - mainCSV = pd.read_csv(os.path.join(subdir, f"{dirname}.csv")) - candCSV = pd.read_csv(os.path.join(subdir, "cands","results_a.csv")) - merge(mainCSV, candCSV, os.path.join(subdir, f"{dirname}_merged.csv")) + try: + mainCSV = pd.read_csv(os.path.join(subdir, f"{dirname}.csv")) + candCSV = pd.read_csv(os.path.join(subdir, "cands","results_a.csv")) + merge(mainCSV, candCSV, os.path.join(subdir, f"{dirname}_merged.csv")) + except FileNotFoundError: + logger.error(f"{dirname} has no candidates!") def merge(mainDF:pd.DataFrame, candDF:pd.DataFrame, outname): #add new columns to main dataframe