class StockData(QtCore.QObject):
def __init__(self):
QtCore.QObject.__init__(self)
self.myDM = QicsDataModelDefault()
self.connect(self.myDM, QtCore.SIGNAL("modelChanged(QicsRegion)"),
self.dataValuesChanged)
self.loadDT()
def dataModel(self):
return self.myDM
def updateStocks(self):
for i in range(self.myDM.lastRow() + 1):
self.updateStock(i)
def dataValuesChanged(self, reg):
row = reg.startCell().row()
col = reg.startCell().column()
if col == day_zero + cur_idx:
yesterday = self.getFloatValue(row, day_one + cur_idx)
today = self.getFloatValue(row, day_zero + cur_idx)
self.emit(PYSIGNAL("stockChanged"), (row, (today > yesterday)))
def loadDT(self):
f = QtCore.QFile(fname)
f.open(QtCore.QIODevice.ReadOnly)
stream = QtCore.QTextStream(f)
self.myDM.readASCII(stream)
f.close()
def updateStock(self, row):
cur_price = self.getFloatValue(row, day_zero + cur_idx)
if cur_price < 0:
# yesterday's close
cur_price = self.getFloatValue(row, day_one + cur_idx)
cur_price = self.newprice(cur_price, 0.02, self.posneg())
fdata = QicsDataFloat(cur_price)
self.myDM.setItem(row, day_zero + cur_idx, fdata)
# Check high.
high = self.getFloatValue(row, day_zero + high_idx)
if high < 0 or cur_price > high:
self.myDM.setItem(row, day_zero + high_idx, fdata)
# Check low.
low = self.getFloatValue(row, day_zero + low_idx)
if low < 0 or cur_price < low:
self.myDM.setItem(row, day_zero + low_idx, fdata)
def getFloatValue(self, row, col):
itm = self.myDM.item(row, col)
if itm:
if itm.type() == QicsDataItem_Float:
return itm.data()
elif itm.type() == QicsDataItem_Double:
return itm.data()
elif itm.type() == QicsDataItem_Int:
return float(itm.data())
return -1.0
def newprice(self, old_price, percent_change=0.02, updown=1):
new_price = old_price + (self.randf() * percent_change * old_price *
updown)
new_price = (float(self.roundf(new_price * 100))) / 100
return new_price
def roundf(self, f):
base = int(math.floor(f))
if (f - base) < 0.5:
return base
else:
return base + 1
def randf(self):
return random.random()
def posneg(self):
if self.roundf(self.randf()) < 0.5:
return -1
else:
return 1
class Consultant(QtGui.QMainWindow):
"""
Consultant - The primary class. Subclassed from QMainWindow, this class
contains both the table and the data model, as well as all the methods to
implement the UI behavior.
"""
def __init__(self, parent, name):
QtGui.QMainWindow.__init__(self, parent)
# create the data model
self.dataModel = QicsDataModelDefault(0, NUM_COLUMNS)
# create the table, using the data model we created above
self.table = QicsTable(self.dataModel, self)
self.setCentralWidget(self.table)
# customize the column header to make things look pretty
self.table.columnHeaderRef().setAlignment(QtCore.Qt.AlignCenter)
self.table.columnHeaderRef().cellRef(0,0).setLabel("First Name")
self.table.columnHeaderRef().cellRef(0,1).setLabel("Last Name")
self.table.columnHeaderRef().cellRef(0,2).setLabel("S.S. Number")
self.table.columnHeaderRef().cellRef(0,3).setLabel("Date of Birth")
self.table.columnHeaderRef().cellRef(0,4).setLabel("Office")
self.table.columnHeaderRef().cellRef(0,5).setLabel("Sec. Clearance")
self.table.columnHeaderRef().cellRef(0,6).setLabel("Billable Rate")
# Add a validator to the SSN to insure a consistant display behaviour
# and error checking.
self.ssnValidator = SSNValidator()
self.table.columnRef(2).setValidator(self.ssnValidator)
# Format the combo box for the office column
self.officeComboDisplay = QicsComboCellDisplay()
self.officeComboDisplay.addItem("Boston")
self.officeComboDisplay.addItem("Cambridge")
self.officeComboDisplay.addItem("Helsinki")
self.officeComboDisplay.setAddValueToList(False)
self.table.columnRef(4).setDisplayer(self.officeComboDisplay)
# Security clearance check box
self.columnHeaderCheckDisp = SecCheckBox()
self.table.columnRef(5).setDisplayer(self.columnHeaderCheckDisp)
# Format the billable rate column with a dollar sign, decimal point,
# and the appropriate cents digits.
self.brFormatter = QicsDataItemSprintfFormatter()
self.brFormatter.addFormatString(QicsDataItem_Float, "$%.2f")
self.brFormatter.addFormatString(QicsDataItem_Int, "$%d.00")
self.table.columnRef(6).setFormatter(self.brFormatter)
self.table.columnRef(6).setValidator(QtGui.QDoubleValidator(self.table))
# Add a title widget to the top of the table
label = QtGui.QLabel("Consultants", self.table)
label.setAlignment(QtCore.Qt.AlignCenter)
self.table.setTopTitleWidget(label)
# Set column widths
for i in range(NUM_COLUMNS):
self.table.columnRef(i).setWidthInChars(7)
self.table.setVisibleColumns(NUM_COLUMNS)
# Menu Bar
menubar = self.menuBar()
self.filemenu = QtGui.QMenu("File", self)
self._Open = QtGui.QAction("Open...", self)
self.filemenu.addAction(self._Open)
self.connect(self._Open, QtCore.SIGNAL("triggered()"), self.fileOpen)
self._Save = QtGui.QAction("Save...", self)
self.filemenu.addAction(self._Save)
self.connect(self._Save, QtCore.SIGNAL("triggered()"), self.fileSave)
self._Exit = QtGui.QAction("Quit", self)
self.filemenu.addAction(self._Exit)
self.connect(self._Exit, QtCore.SIGNAL("triggered()"), QtGui.qApp, QtCore.SLOT("quit()"))
menubar.addMenu(self.filemenu)
self.toolsmenu = QtGui.QMenu("Tools", self)
self._AddCs = QtGui.QAction("Add consultant", self)
self._DelCs = QtGui.QAction("Delete consultant", self)
self._SortA = QtGui.QAction("Sort Ascending", self)
self._SortD = QtGui.QAction("Sort Descending", self)
self.connect(self._AddCs, QtCore.SIGNAL("triggered()"), self.addRow)
self.connect(self._DelCs, QtCore.SIGNAL("triggered()"), self.deleteRow)
self.connect(self._SortA, QtCore.SIGNAL("triggered()"), self.sortAscending)
self.connect(self._SortD, QtCore.SIGNAL("triggered()"), self.sortDescending)
self.toolsmenu.addAction(self._AddCs)
self.toolsmenu.addAction(self._DelCs)
self.toolsmenu.addAction(self._SortA)
self.toolsmenu.addAction(self._SortD)
menubar.addMenu(self.toolsmenu)
# Connection for autosort
QtCore.QObject.connect(self.table.columnHeaderRef(),
QtCore.SIGNAL("doubleClicked(int, int, int, const QPoint &)"),
self.autoSort)
# Reset autoSort sanity vars
self._autoSortCol = -1
self._autoSortOrder = Qics.Descending
def addRow(self):
self.table.addRows(1)
def deleteRow(self):
self.table.deleteRow(self.table.currentCell().rowIndex())
def sortAscending(self):
self.table.sortRows([self.table.currentCell().columnIndex()], Qics.Ascending)
def sortDescending(self):
self.table.sortRows([self.table.currentCell().columnIndex()], Qics.Descending)
def fileOpen(self):
inFileName = QtGui.QFileDialog.getOpenFileName(self, "Choose a file",
"./",
"CSV files (*.csv)")
#,
#"open file dialog")
if inFileName.isEmpty():
return;
inFile = QtCore.QFile(inFileName)
error = 0
if inFile.open(QtCore.QIODevice.ReadOnly):
inStream = QtCore.QTextStream(inFile)
self.table.setIgnoreModelSizeChanges(True)
self.dataModel.readASCII(inStream, ',', 0, 0, True)
inFile.close()
self.table.setIgnoreModelSizeChanges(False)
else:
error = 1
if error:
fileBox = QtGui.QMessageBox()
fileBox.setCaption("File Operation")
fileBox.setText(QtCore.QString("Could not read file: %1").arg(inFileName))
fileBox.setButtonText(0, "OK")
fileBox.exec_()
def fileSave(self):
outFileName = QtGui.QFileDialog.getSaveFileName(self,
"save file dialog"
"Choose a file",
"./",
"CSV files (*, csv)")
if outFileName.isEmpty():
return
outFile = QtCore.QFile(outFileName)
error = 0
if outFile.open(QtCore.QIODevice.WriteOnly):
outStream = QtCore.QTextStream(outFile)
self.dataModel.writeASCII(outStream, ',')
#outStream << "\n"
outFile.close()
else:
error = 1
fileBox = QtGui.QMessageBox()
fileBox.setWindowTitle("File Operation")
if error < 1:
fileBox.setText(QtCore.QString("Saved Table Data to %1").arg(outFileName))
else:
fileBox.setText(QtCore.QString("Error Accessing %1").arg(outFileName))
fileBox.setButtonText(0, "OK")
fileBox.exec_()
def autoSort(self, row, col, button, pos):
if self._autoSortCol == col:
if self._autoSortOrder == Qics.Ascending:
self._autoSortOrder = Qics.Descending
else:
self._autoSortOrder = Qics.Ascending
else:
self._autoSortCol = col
self._autoSortOrder = Qics.Ascending
self.table.sortRows(col, self._autoSortOrder)
class Spreadsheet(QtGui.QMainWindow):
def __init__(self, p=None, name=None):
"""
Construct the Spreadsheet object. Creates a QMainWindow with a menu
and toolbar, a QicsTable, and connects the appropriate slots to load
and manipulate data.
"""
QtGui.QMainWindow.__init__(self)
self._currency = QicsDataItemCurrencyFormatter()
#self.setCaption("Excellent Spreadsheet Demo")
# Pulldown Menu
# File
file = QtGui.QMenu("&File", self)
# Formating menu File with connect to Python function
## New
self._New = QtGui.QAction("&New", self)
#self._New.setShortcut(QtGui.QKeySequence("Ctrl+N"))
## Shortcut keys
self._New.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_N
file.addAction(self._New)
## Connect to self.fileNew Python func
self.connect(self._New, QtCore.SIGNAL("triggered()"), self.fileNew)
## Open
self._Open = QtGui.QAction("&Open", self)
## Shortcut keys
self._Open.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_O
file.addAction(self._Open)
## Connect to self.openFile Python func
self.connect(self._Open, QtCore.SIGNAL("triggered()"), self.openFile)
## Save
self._Save = QtGui.QAction("&Save", self)
## Shortcut keys
self._Save.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_S
file.addAction(self._Save)
## Connect to self.saveFile Python func
self.connect(self._Save, QtCore.SIGNAL("triggered()"), self.saveFile)
## Save As
self._SaveAs = QtGui.QAction("S&ave As", self)
## Shortcut keys
#self._Save.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_S
file.addAction(self._SaveAs)
## Connect to self.saveFile Python func
self.connect(self._SaveAs, QtCore.SIGNAL("triggered()"), self.saveFileAs)
## Exit
self._Exit = QtGui.QAction("E&xit", self)
## Shortcut keys
self._Save.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_X
file.addAction(self._Exit)
## Connect to self.saveFile Python func
self.connect(self._Exit,QtCore.SIGNAL("triggered()"), Qt.qApp, QtCore.SLOT("closeAllWindows()"))
# Show File menu in menubar
self.menuBar().addMenu(file)
# Edit
edit = QtGui.QMenu("&Edit", self)
self._Copy = QtGui.QAction("&Copy", self)
## Shortcut keys
self._Copy.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_C
edit.addAction(self._Copy)
## Connect to self.fileNew Python func
self.connect(self._Copy, QtCore.SIGNAL("triggered()"), self.copy)
# Paste
self._Paste = QtGui.QAction("&Paste", self)
## Shortcut keys
self._Paste.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_C
edit.addAction(self._Paste)
## Connect to self.fileNew Python func
self.connect(self._Paste, QtCore.SIGNAL("triggered()"), self.paste)
self.menuBar().addMenu(edit)
## View
view = QtGui.QMenu("View", self)
self._View = QtGui.QAction("&View", self)
## Shortcut keys
self._View.shortcut = QtCore.Qt.CTRL+QtCore.Qt.Key_V
view.addAction(self._View)
#self.connect(self._View, QtCore.SIGNAL("triggered()"), self.view)
self.menuBar().addMenu(view)
## Insert
insert = QtGui.QMenu("Insert", self)
self._InsertRow = QtGui.QAction("&Row", self)
## Shortcut keys
insert.addAction(self._InsertRow)
self.connect(self._InsertRow, QtCore.SIGNAL("triggered()"), self.insertRow)
self._InsertCol = QtGui.QAction("Co&lumn", self)
## Shortcut keys
insert.addAction(self._InsertCol)
self.connect(self._InsertCol, QtCore.SIGNAL("triggered()"), self.insertColumn)
self.menuBar().addMenu(insert)
## Format
format = QtGui.QMenu("Format", self)
self.menuBar().addMenu(format)
rowFormat = QtGui.QMenu("Row", self)
format.addMenu(rowFormat)
format.addSeparator()
self._Foreground = QtGui.QAction("Fore&ground...", self)
format.addAction(self._Foreground)
self.connect(self._Foreground, QtCore.SIGNAL("triggered()"), self.formatForegroundColor)
self._Background = QtGui.QAction("&Background...", self)
format.addAction(self._Background)
self.connect(self._Background, QtCore.SIGNAL("triggered()"), self.formatBackgroundColor)
self._Font = QtGui.QAction("&Font...", self)
format.addAction(self._Font)
self.connect(self._Font, QtCore.SIGNAL("triggered()"), self.formatFont)
# Tools
tools = QtGui.QMenu("&Tools", self)
self.menuBar().addMenu(tools)
self.showQSAAction = QtGui.QAction(self.tr("Script Editor"), self)
self.connect(self.showQSAAction, QtCore.SIGNAL("triggered()"), self.showQSAWorkbench)
tools.addAction(self.showQSAAction)
self.scriptTestAction = QtGui.QAction(self.tr("Script Test"), self)
self.connect(self.scriptTestAction, QtCore.SIGNAL("triggered()"), self.testScripts)
tools.addAction(self.scriptTestAction)
self.saveScriptAction = QtGui.QAction(self.tr("Save Script"), self)
self.connect(self.saveScriptAction, QtCore.SIGNAL("triggered()"), self.saveScripts)
tools.addAction(self.saveScriptAction)
# Data
data = QtGui.QMenu("&Data", self)
self.menuBar().addMenu(data)
self.sortAscAction = QtGui.QAction(self.tr("&Ascendingt"), self)
self.connect(self.sortAscAction, QtCore.SIGNAL("triggered()"), self.sortAscending)
data.addAction(self.sortAscAction)
self.sortDescAction = QtGui.QAction(self.tr("&Descendingt"), self)
self.connect(self.sortDescAction, QtCore.SIGNAL("triggered()"), self.sortDescending)
data.addAction(self.sortDescAction)
# Spans
spans = QtGui.QMenu("Spans", self)
self.menuBar().addMenu(spans)
self.addSpanAction = QtGui.QAction(self.tr("Add Span"), self)
self.connect(self.addSpanAction, QtCore.SIGNAL("triggered()"), self.addSpan)
spans.addAction(self.addSpanAction)
self.removeSpanAction = QtGui.QAction(self.tr("Remove Span"), self)
self.connect(self.removeSpanAction, QtCore.SIGNAL("triggered()"), self.removeSpan)
spans.addAction(self.removeSpanAction)
# Toolbars
# File operation toolbar
fileTools = QtGui.QToolBar("File tools", self)
#QtGui.QToolButton(_getIconSet("filenew.xpm"), "New", QtCore.QString, self.fileNew, fileTools)
self.fileNewButton = QtGui.QToolButton()
self.fileNewAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/filenew.xpm")), self.tr("&New"), self)
self.connect(self.fileNewAct, QtCore.SIGNAL("triggered()"), self.fileNew)
fileTools.addAction(self.fileNewAct)
self.fileOpenAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/fileopen.xpm")), self.tr("&Open"), self)
self.connect(self.fileOpenAct, QtCore.SIGNAL("triggered()"), self.openFile)
fileTools.addAction(self.fileOpenAct)
self.fileSaveAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/filesave.xpm")), self.tr("&Save"), self)
self.connect(self.fileSaveAct, QtCore.SIGNAL("triggered()"), self.saveFile)
fileTools.addAction(self.fileSaveAct)
fileTools.addSeparator()
# Edit operations
self.cutAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/editcut.xpm")), self.tr("&Cut"), self)
self.connect(self.cutAct, QtCore.SIGNAL("triggered()"), self.cut)
fileTools.addAction(self.cutAct)
self.copyAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/editcopy.xpm")), self.tr("&Copy"), self)
self.connect(self.copyAct, QtCore.SIGNAL("triggered()"), self.copy)
fileTools.addAction(self.copyAct)
self.pasteAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/editpaste.xpm")), self.tr("&Paste"), self)
self.connect(self.pasteAct, QtCore.SIGNAL("triggered()"), self.paste)
fileTools.addAction(self.pasteAct)
fileTools.addSeparator()
# Sort
self.sortAscendingAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/sortasc.xpm")), self.tr("Sort Ascending"), self)
self.connect(self.sortAscendingAct, QtCore.SIGNAL("triggered()"), self.sortAscending)
fileTools.addAction(self.sortAscendingAct)
self.sortDescendingAct = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/sortdesc.xpm")), self.tr("Sort Descending"), self)
self.connect(self.sortDescendingAct, QtCore.SIGNAL("triggered()"), self.sortDescending)
fileTools.addAction(self.sortDescendingAct)
fileTools.addSeparator()
self.addToolBar(fileTools)
# Font name and point size
fontTools = QtGui.QToolBar("Font tools", self)
self.fontFamilyCombo = QtGui.QComboBox(fontTools)
self.fontSizeCombo = QtGui.QComboBox(fontTools)
fontTools.addWidget(self.fontFamilyCombo)
fontTools.addWidget(self.fontSizeCombo)
self.connect(self.fontFamilyCombo, QtCore.SIGNAL("activated(int)"), self.selectFont)
self.connect(self.fontSizeCombo, QtCore.SIGNAL("activated(int)"), self.selectFont)
fdb = QtGui.QFontDatabase()
families = fdb.families()
self.fontFamilyCombo.insertItems(0, families)
# Font format operations
self.boldAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textbold.xpm")), self.tr("Bold"), self)
self.connect(self.boldAction, QtCore.SIGNAL("triggered()"), self.textBold)
fontTools.addAction(self.boldAction)
self.boldAction.setCheckable(True)
self.italicAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textitalic.xpm")), self.tr("Italic"), self)
self.connect(self.italicAction, QtCore.SIGNAL("triggered()"), self.textItalic)
fontTools.addAction(self.italicAction)
self.italicAction.setCheckable(True)
self.underlineAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textunder.xpm")), self.tr("Underline"), self)
self.connect(self.underlineAction, QtCore.SIGNAL("triggered()"), self.textUnderline)
fontTools.addAction(self.underlineAction)
self.underlineAction.setCheckable(True)
fontTools.addSeparator()
# Text Alignment
self.textLeftAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textleft.xpm")), self.tr("Align Left"), self)
self.connect(self.textLeftAction, QtCore.SIGNAL("triggered()"), self.textAlignLeft)
fontTools.addAction(self.textLeftAction)
self.textLeftAction.setCheckable(True)
self.textCenterAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textcenter.xpm")), self.tr("Align Center"), self)
self.connect(self.textCenterAction, QtCore.SIGNAL("triggered()"), self.textAlignCenter)
fontTools.addAction(self.textCenterAction)
self.textCenterAction.setCheckable(True)
self.textRightAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/textright.xpm")),"Align Right", self)
self.connect(self.textRightAction, QtCore.SIGNAL("triggered()"), self.textAlignRight)
fontTools.addAction(self.textRightAction)
self.textRightAction.setCheckable(True)
fontTools.addSeparator()
# Formatters
currencyTB = QtGui.QToolButton(fontTools)
currencyTB.setText("$")
self.connect(currencyTB, QtCore.SIGNAL("clicked()"), self.setCurrencyFormatter)
fontTools.addWidget(currencyTB)
fontTools.addSeparator()
# Colors
self.formatForegroundColorAction = QtGui.QAction(QtGui.QIcon(QtGui.QPixmap("images/fontcolor.xpm")), self.tr("Font Color"), self)
self.connect(self.formatForegroundColorAction, QtCore.SIGNAL("triggered()"), self.formatForegroundColor)
fontTools.addAction(self.formatForegroundColorAction)
self.addToolBar(fontTools)
# Table
vbox = QtGui.QWidget(self)
vboxLayout = QtGui.QVBoxLayout();
cellrow = QtGui.QWidget(vbox)
cellrowLayout = QtGui.QHBoxLayout()
cellrowLayout.setMargin(2)
cellrowLayout.addWidget(QtGui.QLabel("Cell value = "))
self.cellLineEdit = QtGui.QLineEdit()
cellrowLayout.addWidget(self.cellLineEdit)
cellrow.setLayout(cellrowLayout)
self.connect(self.cellLineEdit, QtCore.SIGNAL("returnPressed()"), self.cellLineUpdate)
lastRow = 10
lastCol = 50
# Create the data model
self.dm = QicsDataModelDefault(lastRow, lastCol)
# Create the table, using the data model we created above
self.table = QicsTable(self.dm)
# self.table.setViewport(Qics.QicsRegion(0, 0, 50, 50))
vboxLayout.addWidget(cellrow)
vboxLayout.addWidget(self.table)
vbox.setLayout(vboxLayout)
self.connect(self.table, QtCore.SIGNAL("selectionListChanged(bool)"), self.updateSelection)
# Set some visual resources on the table
self.table.rowHeader().setAlignment(QtCore.Qt.AlignHCenter)
self.table.columnHeader().setAlignment(QtCore.Qt.AlignHCenter)
self.table.rowHeader().column(0).setWidthInChars(3)
self.table.mainGridRef().setBackgroundColor(QtCore.Qt.white)
# Set row sorting mode to non-distractive sord
self.table.show()
self.setFontCombos(self.table.font())
self.setCentralWidget(vbox)
self.statusBar().showMessage("Ready", -1)
self.fileName = QtCore.QString("")
def loadFile(self, fname):
"""
Load a Comma Delimited File into the data table.
"""
# Create the stream from the file, read into data model
file = QtCore.QFile(fname)
if file.open(QtCore.QIODevice.ReadOnly):
stream = QtCore.QTextStream(file)
self.dm.readASCII(stream, ',')
def writeFile(self, fname):
"""
Save a Comma Delimited File from the data table.
"""
# Create the stream from the file, read into data model
file = QtCore.QFile(fname)
if file.open(QtCore.QIODevice.WriteOnly):
stream = QtCore.QTextStream(file)
self.dm.writeASCII(stream, ',', 0, 0, 10, 10)
def getSelectedRegion(self):
"""
Query the table, find the start and end row/column of the currently
selected region, if any.
"""
selection = QicsSelection()
list = self.table.selectionList()
# Get selected boundry
if list is not None:
if len(list) > 0:
selection = list[0]
self.startSelectedRow = selection.anchorRow()
self.startSelectedColumn = selection.anchorColumn()
else:
# Reset flags
self.startSelectedRow = -1
self.startSelectedColumn = -1
row = selection.endRow()
if row > self.dm.lastRow():
row = self.dm.lastRow() + 1
else:
row += 1
col = selection.endColumn()
if col > self.dm.lastColumn():
col = self.dm.lastColumn() + 1
else:
col += 1
self.endSelectedRow = row
self.endSelectedColumn = col
def setFontCombos(self, font):
"""
Given a font, set the fontFamilyCombo to show it, set the fontSizeCombo
to contain valid sizes for that font, and then show the correct font
size.
"""
# Set font family
self.fontFamilyCombo.setEditText(font.family())
fdb = QtGui.QFontDatabase()
sizes = fdb.pointSizes(font.family())
# Clear the sizes combo, fill it with valid sizes based in family
self.fontSizeCombo.clear()
if len(sizes) > 1:
for pts in sizes:
self.fontSizeCombo.insertItem(pts - 6, QtCore.QString.number(pts))
self.fontSizeCombo.setEditText(QtCore.QString.number(font.pointSize()))
def setCellValue(self, row, col, value):
"""
Set cell[row][col] = value.
"""
self.dm.setItem(row, col, value)
def getCellValue(self, row, col):
"""
Return the value of cell[row][col].
"""
return self.dm.itemString(row, col)
def sortAscending(self):
"""
Sort the table by the selected column in ascending order.
"""
self.getSelectedRegion()
self.table.sortRows([self.startSelectedColumn], Qics.Ascending)
def sortDescending(self):
"""
Sort the table by the selected column in descending order.
"""
self.getSelectedRegion()
self.table.sortRows([self.startSelectedColumn], Qics.Descending)
def copy(self):
"""
Calls the table's copy operation to place data into the QClipboard
owned by qApp.
"""
self.table.copy()
def cut(self):
"""
Calls the table's cut operation to place data into the QClipboard owned
by qApp and remove it from the visible table area.
"""
self.table.cut()
def paste(self):
"""
Calls the table's paste operation to retrieve cells from the QClipboard
owned by qApp and paste them at the current cell.
"""
self.table.paste()
def fileNew(self):
"""
Clear out all the data currently displayed in the table.
"""
print "To be implemented"
def openFile(self):
"""
Bring up a QFileDialog to select a file, then load it.
"""
fname = QtGui.QFileDialog.getOpenFileName()
if not fname.isEmpty():
self.fileName = fname
self.loadFile(self.fileName)
def saveFile(self):
"""
Save data to the current file name.
"""
if not self.fileName.isEmpty():
self.writeFile(self.fileName)
def saveFileAs(self):
"""
Bring up a QFileDialog to select a file, then save to it.
"""
fname = QtGui.QFileDialog.getSaveFileName()
if not fname.isEmpty():
self.fileName = fname
self.writeFile(self.fileName)
def insertRow(self):
"""
Add a row to the table before the currently selected row.
"""
self.getSelectedRegion()
if self.startSelectedRow > -1:
self.table.insertRow(self.startSelectedRow)
def insertColumn(self):
"""
Add a column to the table before the currently selected column.
"""
self.getSelectedRegion()
if self.startSelectedColumn > -1:
self.table.insertColumn(self.startSelectedColumn)
def formatFont(self):
"""
Bring up the font dialog, select a font, apply it to the selected
region.
"""
# Select a font
font, ok = QtGui.QFontDialog.getFont(QtGui.QFont("Helvetica", 10), self)
if not ok:
return
# If cells are selected, set their display font
self.getSelectedRegion()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setFont(font)
def formatBackgroundColor(self):
"""
Bring up the color dialog, select a color, apply it to the selected
region as the background color.
"""
color = QtGui.QColorDialog.getColor()
if not color.isValid():
return
# If cells are selected, set their display font
self.getSelectedRegion()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setBackgroundColor(color)
def formatForegroundColor(self):
"""
Bring up the color dialog, select a color, apply it to the selected
region as the foreground color.
"""
color = QtGui.QColorDialog.getColor()
if not color.isValid():
return
# If cells are selected, set their display font
self.getSelectedRegion()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setForegroundColor(color)
def selectFont(self, combo):
family = self.fontFamilyCombo.currentText()
size, ok = self.fontSizeCombo.currentText().toInt()
# If cells are selected, set their display font
self.getSelectedRegion()
if self.startSelectedRow > -1:
fdb = QtGui.QFontDatabase()
currentFont = self.table.cellRef(self.startSelectedRow, self.startSelectedColumn).font()
font = fdb.font(family, fdb.styleString(currentFont), size)
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setFont(font)
def textAlignLeft(self):
"""
Set the alignment value of the displayed cell to AlignLeft.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setAlignment(QtCore.Qt.AlignLeft)
self.textRightAction.setChecked(False)
self.textCenterAction.setChecked(False)
def textAlignRight(self):
"""
Set the alignment value of the displayed cell to AlignRight.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setAlignment(QtCore.Qt.AlignRight)
self.textLeftAction.setChecked(False)
self.textCenterAction.setChecked(False)
def textAlignCenter(self):
"""
Set the alignment value of the displayed cell to AlignHCenter.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setAlignment(QtCore.Qt.AlignHCenter)
self.textLeftAction.setChecked(False)
self.textRightAction.setChecked(False)
def textBold(self):
"""
Toggle the bold-ness state of whatever cells are selected.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
on = self.boldAction.isChecked()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
f = self.table.cellRef(i, j).font()
f.setBold(on)
self.table.cellRef(i, j).setFont(f)
def textItalic(self):
"""
Toggle the italic-ness state of whatever cells are selected.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
on = self.italicAction.isChecked()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
f = self.table.cellRef(i, j).font()
f.setItalic(on)
self.table.cellRef(i, j).setFont(f)
def textUnderline(self):
"""
Toggle the underline-ed-ness state of whatever cells are selected.
"""
if self.startSelectedRow < 0 and self.startSelectedColumn < 0:
return
on = self.underlineAction.isChecked()
if self.startSelectedRow > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
f = self.table.cellRef(i, j).font()
f.setUnderline(on)
self.table.cellRef(i, j).setFont(f)
def setCurrencyFormatter(self):
"""
Set a formatter on the selected cells do display the values in dollar
value ($ XX.YY).
"""
self.getSelectedRegion()
if self.startSelectedRow > -1 and self.startSelectedColumn > -1:
for i in range(self.startSelectedRow, self.endSelectedRow):
for j in range(self.startSelectedColumn, self.endSelectedColumn):
self.table.cellRef(i, j).setFormatter(self._currency)
def cellLineUpdate(self):
"""
When a value is entered in the cellLineEdit and a cell is selected in
the table, store the data element as a new data item.
"""
self.getSelectedRegion()
if self.startSelectedRow > -1 and self.startSelectedColumn > -1:
# OK, we got a selected cell and some data. Time to put it into
# the data model
# temp var because they are easier to type
row = self.startSelectedRow
col = self.startSelectedColumn
# Try to convert to int
i, ok = self.cellLineEdit.text().toInt()
if ok:
self.dm.setItem(row, col, QicsDataInt(i))
return
# Try to convert to float
f, ok = self.cellLineEdit.text().toFloat()
if ok:
self.dm.setItem(row, col, QicsDataFloat(f))
return
self.dm.setItem(row, col, QicsDataString(self.cellLineEdit.text()))
def updateSelection(self, in_progress):
"""
A cell was selected, so update the state of the toolbar pieces.
"""
if in_progress:
return
self.getSelectedRegion()
if self.startSelectedRow > -1 and self.startSelectedColumn > -1:
font = self.table.cellRef(self.startSelectedRow, self.startSelectedColumn).font()
self.setFontCombos(font)
if font.bold():
self.boldAction.setChecked(True)
else:
# Is on, turn off (not bold)
self.boldAction.setChecked(False)
if font.italic():
self.italicAction.setChecked(True)
else:
# Is on, turn off (not bold)
self.italicAction.setChecked(False)
if font.underline():
self.underlineAction.setChecked(True)
else:
# Is on, turn off (not bold)
self.underlineAction.setChecked(False)
alignment = self.table.cellRef(self.startSelectedRow, self.startSelectedColumn).alignment()
if alignment == QtCore.Qt.AlignLeft:
self.textLeftAction.setChecked(True)
self.textRightAction.setChecked(False)
self.textCenterAction.setChecked(False)
elif alignment == QtCore.Qt.AlignRight:
self.textLeftAction.setChecked(False)
self.textRightAction.setChecked(True)
self.textCenterAction.setChecked(False)
elif alignment == QtCore.Qt.AlignCenter:
self.textLeftAction.setChecked(False)
self.textRightAction.setChecked(False)
self.textCenterAction.setChecked(True)
else:
self.textLeftAction.setChecked(False)
self.textRightAction.setChecked(False)
self.textCenterAction.setChecked(False)
if self.dm.item(self.startSelectedRow, self.startSelectedColumn) is None:
self.cellLineEdit.setText("")
return
cell = self.table.cell(self.startSelectedRow, self.startSelectedColumn, True)
if cell.dataValue():
cellval = cell.dataValue().string()
self.cellLineEdit.setText(cellval)
def showQSAWorkbench(self):
QtGui.QMessageBox.warning(None, "No QSA support!",
"Sorry, no QSA support compiled in",
QtGui.QMessageBox.Ok, QtGui.QMessageBox.NoButton)
def testScripts(self):
QtGui.QMessageBox.warning(None, "No QSA support!",
"Sorry, no QSA support compiled in",
QtGui.QMessageBox.Ok, QtGui.QMessageBox.NoButton)
def saveScripts(self):
QtGui.QMessageBox.warning(None, "No QSA support!",
"Sorry, no QSA support compiled in",
QtGui.QMessageBox.Ok, QtGui.QMessageBox.NoButton)
def addSpan(self):
"""
Adds span to selected cells
"""
slist = self.table.selectionList()
if slist:
[self.table.addCellSpan(QicsSpan(elem.topRow(), elem.leftColumn(), elem.numRows(), elem.numColumns())) for elem in slist]
def removeSpan(self):
"""
Remove span from selected cells
"""
slist = self.table.selectionList()
if slist:
[self.table.removeCellSpan(elem.topRow(), elem.leftColumn()) for elem in slist]