from enum import Enum

# Possible modes v = vector length of (x, y) / a = absolut values, reduce spread / c = cartesian, empirical values
class Mode(Enum):
    V = "v"
    A = "a"
    C = "c"

# Display pandas dataframe description, head and info
class Information(Enum):
    DISABLED = False
    ENABLED = True

# Display countplot, heatmap and scatterplot graphs
class Graph(Enum):
    DISABLED = False
    ENABLED = True

# Class Arguments for accepted values 
class Arguments:

    # Constructor. Default params: "filepath", v, False, False
    def __init__(self, file_path, mode, information, graph):
        self.file_path = file_path
        self.mode = mode
        self.information = information
        self.graph = graph

    # Filepath getter
    def get_file_path(self):
        return self.file_path

    # Filepath setter
    def set_file_path(self, value):
        self.file_path = value

    # Mode getter
    def get_mode(self):
        return self.mode.value

    # Mode setter
    def set_mode(self, value):
        self.mode = Mode(value)
        
    # Information getter
    def get_information(self):
        return self.information.value

    # Information setter
    def set_information(self, value):
        self.information = Information(value)

    # Graph getter 
    def get_graph(self):
        return self.graph.value

    # GRaph setter
    def set_graph(self, value):
        self.graph = Graph(value)