base.py

import pandas as pd
import os

#TODO: set error file location from cmd
log_errors_path = "MELD_errors.csv"

class bcolors: #for printing coloured text
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

def raise_(ex):
	raise ex



def log_invalid_code(codes, mask, code_type=None, file_path=None, cause=None):
	print("ERROR WITH CODES", file_path, codes[~mask])
	
	errors = pd.DataFrame([])
	errors["CODE"] = codes[~mask].astype(str)
	errors["CODE_TYPE"] = code_type
	errors["SOURCE"] = file_path
	errors["CAUSE"] = cause
	
	#append to error log csv
	if os.path.exists(log_errors_path):
		print("FILE EXISTS")
		df_error = pd.read_csv(log_errors_path)
		df_error = pd.concat([df_error, errors])
		df_error.to_csv(log_errors_path, index=False)
	else:
		print("FILE NOT EXIST")
		df_error = errors
		df_error.to_csv(log_errors_path, index=False)
		
	return codes[mask]