# -*- coding: utf-8 -*-


class ColumnList:

    # Default columns (immutable)
    COL_REFERENCE = 'References'
    COL_DESCRIPTION = 'Description'
    COL_VALUE = 'Value'
    COL_FP = 'Footprint'
    COL_FP_LIB = 'Footprint Lib'
    COL_PART = 'Part'
    COL_PART_LIB = 'Part Lib'
    COL_DATASHEET = 'Datasheet'

    # Default columns for groups
    COL_GRP_QUANTITY = 'Quantity Per PCB'
    COL_GRP_TOTAL_COST = 'Total Cost'
    COL_GRP_BUILD_QUANTITY = 'Build Quantity'

    # Generated columns
    _COLUMNS_GEN = [
        COL_GRP_QUANTITY,
        COL_GRP_BUILD_QUANTITY,
    ]

    # Default columns
    _COLUMNS_DEFAULT = [
        COL_DESCRIPTION,
        COL_PART,
        COL_PART_LIB,
        COL_REFERENCE,
        COL_VALUE,
        COL_FP,
        COL_FP_LIB,
        COL_GRP_QUANTITY,
        COL_GRP_BUILD_QUANTITY,
        COL_DATASHEET
    ]

    # Default columns
    # These columns are 'immutable'
    _COLUMNS_PROTECTED = [
        COL_REFERENCE,
        COL_GRP_QUANTITY,
        COL_VALUE,
        COL_PART,
        COL_PART_LIB,
        COL_DESCRIPTION,
        COL_DATASHEET,
        COL_FP,
        COL_FP_LIB
    ]

    def __str__(self):
        return " ".join(map(str, self.columns))

    def __repr__(self):
        return self.__str__()

    def __init__(self, cols=_COLUMNS_DEFAULT):

        self.columns = []

        # Make a copy of the supplied columns
        for col in cols:
            self.AddColumn(col)

    def _hasColumn(self, col):
        # Col can either be <str> or <Column>
        return col.lower() in [c.lower() for c in self.columns]

    """
    Remove a column from the list. Specify either the heading or the index
    """
    def RemoveColumn(self, col):
        if type(col) is str:
            self.RemoveColumnByName(col)
        elif type(col) is int and col >= 0 and col < len(self.columns):
            self.RemoveColumnByName(self.columns[col])

    def RemoveColumnByName(self, name):

        # First check if this is in an immutable colum
        if name in self._COLUMNS_PROTECTED:
            return

        # Column does not exist, return
        if name not in self.columns:
            return

        try:
            index = self.columns.index(name)
            del self.columns[index]
        except ValueError:
            return

    # Add a new column (if it doesn't already exist!)
    def AddColumn(self, col, index=None):

        # Already exists?
        if self._hasColumn(col):
            return

        if type(index) is not int or index < 0 or index >= len(self.columns):
            self.columns.append(col)

        # Otherwise, splice the new column in
        else:
            self.columns = self.columns[0:index] + [col] + self.columns[index:]


if __name__ == '__main__':
    c = ColumnList()

    c.AddColumn("Test1")
    c.AddColumn("Test1")
    c.AddColumn("Test2")
    c.AddColumn("Test3")
    c.AddColumn("Test4")
    c.AddColumn("Test2")

    c.RemoveColumn("Test2")
    c.RemoveColumn("Part")
    c.RemoveColumn(2)
    c.RemoveColumn(5)