def to_records(cls):
columns = Column.count()
rows = Row.count()
return [[str(Cell.array[(row, column)]._data)
if (row, column) in Cell.array else ""
for column in range(columns)] for row in range(rows)]
def columnCount(self, index=None, *args, **kwargs):
columns_count = Column.count()
# logger.debug("columns_count={}, CellModel.Columns={}"
# .format(columns_count, CellModel.Columns))
# if CellModel.Columns - COL_OFFSET <= columns_count:
# logger.debug("CellModel.Columns changed.")
# CellModel.Columns = columns_count + COL_OFFSET
# logger.debug("index.row={}, index.col={}".format(index.row(),
# index.column()))
# self.columnChange()
# CellModel.Columns += 1
return CellModel.Columns
def execute(self):
query = Query(select=self.input.toPlainText())
query.execute()
if query.error == Query.NoError:
var_name = self.target.text().strip()
column = None
new_column = False
if var_name:
logger.debug("var_name={}".format(var_name))
column = Column.get_by_name(var_name)
if not column:
column = Column(Column.count(), name=var_name)
new_column = True
logger.debug("new_column={}".format(new_column))
for row, data in query.result.items():
if new_column:
Cell(row, column.id, data=data)
else:
cell = Cell.get(row, column.id)
cell.data = data
def update_columns_list(self):
begin = ColList.Model.createIndex(0, 0)
end = ColList.Model.createIndex(0, Column.count())
ColList.Model.dataChanged.emit(begin, end)
def rowCount(self, QModelIndex_parent=None, *args, **kwargs):
return Column.count()
