Skip to content
Snippets Groups Projects
Select Git revision
  • 55d359582e053a4e058ed8c857a3839aef5cedb5
  • master default protected
  • pipeline
  • dev
  • quality
  • docs
  • issue-65
  • issue-30
  • issue-19
  • issue-28
  • 0.1.1
  • 0.1.0
12 results

models.py

Blame
  • base.py 1.00 KiB
    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]