127 lines
3.0 KiB
Python
127 lines
3.0 KiB
Python
|
# -*- 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)
|