def on_copy(self):
select_group_dialog = QDialog(self)
select_group_dialog.setWindowTitle(self.tr("Choose source group"))
layout = QVBoxLayout(select_group_dialog)
select_group_dialog.setLayout(layout)
groups_list_view = QTableView(self)
layout.addWidget(groups_list_view)
groups_list_view.setModel(self.ds_model)
groups_list_view.setColumnHidden(DSManagerModel.COLUMN_VISIBILITY, True)
groups_list_view.horizontalHeader().setResizeMode(DSManagerModel.COLUMN_GROUP_DS, QHeaderView.Stretch)
groups_list_view.setSelectionMode(QTableView.NoSelection)
groups_list_view.setAlternatingRowColors(True)
groups_list_view.setShowGrid(False)
groups_list_view.verticalHeader().setResizeMode(QHeaderView.ResizeToContents)
groups_list_view.verticalHeader().hide()
groups_list_view.clicked.connect(
lambda index: select_group_dialog.accept() \
if self.ds_model.isGroup(index) and \
index.column() == DSManagerModel.COLUMN_GROUP_DS \
else None
)
if select_group_dialog.exec_() == QDialog.Accepted:
group_info = self.ds_model.data(groups_list_view.currentIndex(), Qt.UserRole)
group_info.id += "_copy"
edit_dialog = GroupEditDialog()
edit_dialog.setWindowTitle(self.tr('Create group from existing'))
edit_dialog.fill_group_info(group_info)
if edit_dialog.exec_() == QDialog.Accepted:
self.feel_list()
self.ds_model.resetModel()
class OrderDlg(QDialog):
def __init__(self, parent=None):
super(OrderDlg, self).__init__(parent)
self.create_widgets()
self.layout_widgets()
self.setMinimumWidth(850)
self.setWindowTitle(u'Список Order Item')
def create_widgets(self):
self.model = QtSql.QSqlTableModel(self)
self.model.setTable("OrderItem")
self.model.setEditStrategy(QtSql.QSqlTableModel.OnManualSubmit)
self.model.setSort(ID, Qt.AscendingOrder)
self.model.setHeaderData(ID, Qt.Horizontal, QVariant("Order Item"))
self.model.setHeaderData(ORDER, Qt.Horizontal,
QVariant("Order"))
self.model.setHeaderData(PRODUCT, Qt.Horizontal,
QVariant("Product"))
self.model.select()
self.view = QTableView()
self.view.setModel(self.model)
self.view.resizeColumnsToContents()
def layout_widgets(self):
layout = QVBoxLayout()
layout.addWidget(self.view)
self.setLayout(layout)
class DataFrameWidget(QWidget):
''' a simple widget for using DataFrames in a gui '''
def __init__(self, name='DataFrameTable1', parent=None):
super(DataFrameWidget, self).__init__(parent)
self.name = name
self.dataModel = DataFrameModel()
self.dataModel.setDataFrame(DataFrame())
self.dataTable = QTableView()
self.dataTable.setSelectionBehavior(QAbstractItemView.SelectRows)
self.dataTable.setSortingEnabled(True)
self.dataTable.setModel(self.dataModel)
self.dataModel.signalUpdate()
#self.dataTable.setFont(QFont("Courier New", 8))
layout = QVBoxLayout()
layout.addWidget(self.dataTable)
self.setLayout(layout)
def setFormat(self, fmt):
""" set non-default string formatting for a column """
for colName, f in fmt.iteritems():
self.dataModel.columnFormat[colName] = f
def fitColumns(self):
self.dataTable.horizontalHeader().setResizeMode(QHeaderView.Stretch)
def setDataFrame(self, df):
self.dataModel.setDataFrame(df)
def resizeColumnsToContents(self):
self.dataTable.resizeColumnsToContents()
class OrderDlg(QDialog):
def __init__(self, parent=None):
super(OrderDlg, self).__init__(parent)
self.create_widgets()
self.layout_widgets()
self.setMinimumWidth(850)
self.setWindowTitle(u'Список Order Item')
def create_widgets(self):
self.model = QtSql.QSqlTableModel(self)
self.model.setTable("OrderItem")
self.model.setEditStrategy(QtSql.QSqlTableModel.OnManualSubmit)
self.model.setSort(ID, Qt.AscendingOrder)
self.model.setHeaderData(ID, Qt.Horizontal, QVariant("Order Item"))
self.model.setHeaderData(ORDER, Qt.Horizontal, QVariant("Order"))
self.model.setHeaderData(PRODUCT, Qt.Horizontal, QVariant("Product"))
self.model.select()
self.view = QTableView()
self.view.setModel(self.model)
self.view.resizeColumnsToContents()
def layout_widgets(self):
layout = QVBoxLayout()
layout.addWidget(self.view)
self.setLayout(layout)
class ConfSettingsWidget ( QWidget ):
def __init__ ( self, conf, parent=None ):
QWidget.__init__( self, parent )
self._rowHeight = 20
self._view = QTableView()
self._view.setShowGrid ( False )
self._view.setAlternatingRowColors( True )
self._view.setSelectionBehavior ( QAbstractItemView.SelectRows )
#self._view.setSortingEnabled ( True )
#self._view.installEventFilter ( self )
horizontalHeader = self._view.horizontalHeader ()
horizontalHeader.setStretchLastSection ( True )
horizontalHeader.setMinimumSectionSize ( 150 )
horizontalHeader.setResizeMode ( QHeaderView.ResizeToContents )
horizontalHeader.setDefaultSectionSize ( 150 )
verticalHeader = self._view.verticalHeader ();
verticalHeader.setVisible ( False );
verticalHeader.setDefaultSectionSize ( self._rowHeight );
self._baseModel = ConfSettingsModel( conf )
self._view.setModel( self._baseModel );
self._view.horizontalHeader().setStretchLastSection( True );
self._view.resizeColumnToContents( 0 );
peanoDataLayout = QGridLayout();
peanoDataLayout.addWidget( self._view, 0, 0, 1, 1 );
self.setLayout ( peanoDataLayout );
return
class MatlForm(QDialog):
def __init__(self):
super(MatlForm, self).__init__()
if 1 == 0:
self.matlModel= QSqlRelationalTableModel(self)
self.matlModel.setTable('matl')
self.matlModel.setHeaderData(0, Qt.Horizontal,QVariant('name'))
self.matlModel.select()
self.matlView= QTableView()
self.matlView.setModel(self.matlModel)
self.matlView.setItemDelegate(MatlDelegate(self))
self.matlView.setSelectionMode(QTableView.SingleSelection)
self.matlView.setSelectionBehavior(QTableView.SelectRows)
self.matlView.resizeColumnsToContents()
self.matlView.layout()
matlViewLabel= QLabel("<font color=green size=72><b> Material Editor </b></font>")
if 1 == 0:
dataLayout= QVBoxLayout()
dataLayout.addWidget(self.matlViewLabel)
dataLayout.addWidget(self.matlView)
self.setLayout(dataLayout)
self.matlPopup.show()
self.matlPropsHere.show()
self.show()
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
uic.loadUi("mainwindow.ui", self)
model = QStandardItemModel(3, 2, self)
model.setItem(0, 0, QStandardItem(_fromUtf8("xiaoming")))
model.setItem(0, 1, QStandardItem(_fromUtf8("0")))
model.setItem(1, 0, QStandardItem(_fromUtf8("xiaogang")))
model.setItem(1, 1, QStandardItem(_fromUtf8("5")))
model.setItem(2, 0, QStandardItem(_fromUtf8("xiaohong")))
model.setItem(2, 1, QStandardItem(_fromUtf8("0")))
model.setItem(3, 0, QStandardItem(_fromUtf8("赵六")))
model.setItem(3, 1, QStandardItem(_fromUtf8("8")))
self.mapper = QDataWidgetMapper(self)
# 设置模型
self.mapper.setModel(model)
# 设置窗口部件和模型中的列的映射
self.mapper.addMapping(self.lineEdit, 0)
self.mapper.addMapping(self.lineEdit_2, 1)
# 显示模型中的第一行
self.mapper.toFirst()
#----------------------------------------------------------
tableview = QTableView()
tableview.setModel(model)
tableview.show()
class dlg_csv_error(QDialog):
def __init__(self,parent):
QDialog.__init__(self,parent=parent)
self.initGui
def initGui(self):
self.VBL_main = QVBoxLayout()
self.LB_caution = QLabel()
self.VBL_main.addWidget(self.LB_caution)
self.model_content = QStandardItemModel()
self.TB_content = QTableView()
self.TB_content.setModel(self.model_content)
self.VBL_main.addWidget(self.TB_content)
self.DBB_main = QDialogButtonBox()
self.DBB_main.setStandardButtons(QDialogButtonBox.Close)
self.VBL_main.addWidget(self.DBB_main)
self.setLayout(self.VBL_main)
self.DBB_main.rejected.connect(self.close)
def showPeakTable(self):
"""
TODO: write little function to check if sample is None or good
or write exception with good code
"""
if not self.view.sampleTableView.selectedIndexes(
): #not self.acTree.selectedIndexes():
s, b = QInputDialog.getItem(
self.view, "Select one sample", "Select one sample :",
[spl.shortName() for spl in self.model])
if not b:
return
sample = self.model.sample(str(s), fullNameEntry=False)
else:
idx = self.view.sampleTableView.selectedIndexes()[0]
sample = self.model.sample(idx.data().toString(),
fullNameEntry=False)
if sample is None:
print("sample not found...")
return
if not sample.rawPeaks:
self.view.showErrorMessage(
"No peaks found",
"This sample does not have peaks, please do peak picking before"
)
view = QTableView()
view.horizontalHeader().setStretchLastSection(True)
view.setSortingEnabled(True)
model = MSDialogController.getSampleModel(sample, flags='peak')
view.setModel(model)
self.view.addMdiSubWindow(
view, " ".join(["PeakList of",
str(sample.shortName())]))
def showClusterTable(self):
if not self.view.sampleTableView.selectedIndexes(
): #not self.acTree.selectedIndexes():
s, b = QInputDialog.getItem(
self.view, "Select one sample", "Select one sample :",
[spl.shortName() for spl in self.model])
if not b:
return
sample = self.model.sample(str(s), fullNameEntry=False)
else:
idx = self.view.sampleTableView.selectedIndexes()[0]
sample = self.model.sample(idx.data().toString(),
fullNameEntry=False)
if sample is None:
print("sample not found...")
return
if not sample.mappedPeaks:
self.view.showErrorMessage(
"No peaks found",
"This sample does not have peaks, please do peak picking before"
)
view = QTableView()
view.setSortingEnabled(True)
view.horizontalHeader().setStretchLastSection(True)
view.setModel(
MSDialogController.getSampleModel(sample, flags='cluster'))
self.view.addMdiSubWindow(view,
"ClusterList of%s" % str(sample.shortName()))
def showPeakTable(self):
"""
TODO: write little function to check if sample is None or good
or write exception with good code
"""
if not self.view.sampleTableView.selectedIndexes():#not self.acTree.selectedIndexes():
s, b = QInputDialog.getItem(self.view, "Select one sample", "Select one sample :",
[spl.shortName() for spl in self.model])
if not b:
return
sample = self.model.sample(str(s), fullNameEntry=False)
else:
idx = self.view.sampleTableView.selectedIndexes()[0]
sample = self.model.sample(idx.data().toString(), fullNameEntry=False)
if sample is None:
print ("sample not found...")
return
if not sample.rawPeaks:
self.view.showErrorMessage("No peaks found",
"This sample does not have peaks, please do peak picking before")
view=QTableView()
view.horizontalHeader().setStretchLastSection(True)
view.setSortingEnabled(True)
model=MSDialogController.getSampleModel(sample, flags='peak')
view.setModel(model)
self.view.addMdiSubWindow(view, " ".join(["PeakList of", str(sample.shortName())]))
def createTable(self):
# create the view
tv = QTableView()
self.param = self.createParam()
# set the delegate to allow editing table entries
self.td = SpinBoxDelegate(self.param.traits.xslot)
tv.setItemDelegate(self.td)
tm = ParamTableModel(self.createParam(), self)
tv.setModel(tm)
# set the minimum size
tv.setMinimumSize(400, 300)
# hide grid
tv.setShowGrid(False)
# set the font
font = QFont("Courier New", 8)
tv.setFont(font)
# hide horizontal header
hh = tv.horizontalHeader()
hh.setVisible(False)
# set vertical header properties
vh = tv.verticalHeader()
#vh.setStretchLastSection(True)
# set column width to fit contents
tv.resizeColumnsToContents()
return tv
class DetailedProgressDialog(QDialog):
def __init__(self, parent, states):
super(DetailedProgressDialog, self).__init__(parent)
self.setWindowTitle("Realization Progress")
layout = QGridLayout(self)
self.detailed_progress_widget = DetailedProgress(states, self)
self.overview_label = QLabel("Realizations")
self.single_view = QTableView()
self.single_view.setModel(SingleProgressModel(self.single_view))
self.single_view_label = QLabel("Realization details")
self.detailed_progress_widget.clicked.connect(self.show_selection)
layout.addWidget(self.single_view_label, 0, 0)
layout.addWidget(self.overview_label, 0, 1)
layout.addWidget(self.single_view, 1, 0)
layout.addWidget(self.detailed_progress_widget, 1, 1)
self.detailed_progress_widget.show()
self.setLayout(layout)
self.layout().setColumnStretch(0, 1)
self.layout().setColumnStretch(1, 2)
self.progress = None
self.selected_realization = None
self.resize(parent.width(), parent.height())
def set_progress(self, progress, iteration):
self.progress = progress
self.detailed_progress_widget.set_progress(progress, iteration)
self.overview_label.setText("Realizations for iteration {}".format(iteration))
self.update_single_view()
self.update()
def show_selection(self, iens):
if not self.progress:
return
self.selected_realization = iens
self.single_view_label.setText("Realization id: {}".format(iens))
self.update_single_view()
def update_single_view(self):
if not self.single_view.isVisible() or not self.selected_realization in self.progress:
return
model_data = []
jobs = self.progress[self.selected_realization]
headers = []
for job in jobs:
data = job.dump_data()
row = [str(data[key]) for key in data]
model_data.append(row)
headers = data.keys()
self.single_view.model().update_data(headers, model_data)
self.single_view.resizeColumnsToContents()
self.single_view.model().modelReset.emit()
class UpgradeWidget(QWidget):
def init_datasource(self):
self.data_source = DataSource()
def init_view(self):
self.table = QTableView(self)
self.table.setMinimumSize(600, 600)
manager = core.packagemanager.get_package_manager()
package = manager.get_package("edu.cmu.sv.components", "1.0.0")
modules = package.descriptors
apis = []
self.modules_to_upgrade = []
for module in modules.values():
api = self.data_source.api_by_id(module.name)
if ("1.0" != module.version):
apis.append(api)
self.modules_to_upgrade.append(module)
table_model = QStandardItemModel(len(apis), 4)
row = 0
for api in apis:
table_model.setData(table_model.index(row, 0), QVariant(api['id']))
table_model.setData(table_model.index(row, 1), QVariant(api['category']))
table_model.setData(table_model.index(row, 2), QVariant(api['version']))
table_model.setData(table_model.index(row, 3), QVariant(api['description']))
row += 1
table_model.setHeaderData(0, Qt.Horizontal, QVariant("Link"))
table_model.setHeaderData(1, Qt.Horizontal, QVariant("Category"))
table_model.setHeaderData(2, Qt.Horizontal, QVariant("Version"))
table_model.setHeaderData(3, Qt.Horizontal, QVariant("Description"))
self.table.setModel(table_model)
btn = QPushButton(self)
btn.clicked.connect(self.upgrade_api)
btn.setText("Upgrade")
btn.move(600, 500)
def upgrade_api(self):
model = self.table.selectionModel()
indexes = model.selectedIndexes()
for index in indexes:
module = self.modules_to_upgrade[index.row()]
now = datetime.datetime.now()
name = "%s %d-%d-%d" % (module.name, now.year, now.month, now.day)
manager = core.packagemanager.get_package_manager()
package = manager.get_package("edu.cmu.sv.components", "1.0.0")
new_module = vistrails_module.new_module(Module, name)
core.modules.module_registry.set_current_package(package)
new_descriptor = core.modules.module_registry.get_module_registry().add_module(new_module)
core.modules.module_registry.get_module_registry().update_module(module, new_descriptor)
# core.modules.module_registry.get_module_registry().delete_module(module.identifier, module.name)
self.hide()
return
def __init__(self, parent=None):
super(TableWidget, self).__init__(parent)
# Create a simple model for storing data.
model = CustomTableModel()
# Create the table view and add the model to it.
tableView = QTableView()
tableView.setModel(model)
tableView.horizontalHeader().setResizeMode(QHeaderView.Stretch)
tableView.verticalHeader().setResizeMode(QHeaderView.Stretch)
chart = QChart()
chart.setAnimationOptions(QChart.AllAnimations)
# Series 1.
series = QLineSeries()
series.setName("Line 1")
mapper = QVXYModelMapper(self)
mapper.setXColumn(0)
mapper.setYColumn(1)
mapper.setSeries(series)
mapper.setModel(model)
chart.addSeries(series)
# Get the color of the series and use it for showing the mapped area.
seriesColorHex = '#' + hex(series.pen().color().rgb()).upper()[-6:]
model.addMapping(seriesColorHex, QRect(0, 0, 2, model.rowCount()))
# Series 2.
series = QLineSeries()
series.setName("Line 2")
mapper = QVXYModelMapper(self)
mapper.setXColumn(2)
mapper.setYColumn(3)
mapper.setSeries(series)
mapper.setModel(model)
chart.addSeries(series)
# Get the color of the series and use it for showing the mapped area.
seriesColorHex = '#' + hex(series.pen().color().rgb()).upper()[-6:]
model.addMapping(seriesColorHex, QRect(2, 0, 2, model.rowCount()))
chart.createDefaultAxes()
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
chartView.setMinimumSize(640, 480)
# Create the main layout.
mainLayout = QGridLayout()
mainLayout.addWidget(tableView, 1, 0)
mainLayout.addWidget(chartView, 1, 1)
mainLayout.setColumnStretch(1, 1)
mainLayout.setColumnStretch(0, 0)
self.setLayout(mainLayout)
class DetailWindow(myWindow):
listToParaAndRdnr = {}
def __init__(self, parent = None):
super(DetailWindow, self).__init__(parent, layoutCls = QHBoxLayout, fixed = False)
self._listWidget = QListWidget(self)
self._paraAndRdnr = QTableView(self)
self.layout().addWidget(self._listWidget)
self.layout().addWidget(self._paraAndRdnr)
self._listWidget.clicked.connect(self.showParaAndRdnr)
def showParaAndRdnr(self, listItem):
paraAndRdnr = self.listToParaAndRdnr.get(listItem.row())
model = QStandardItemModel(self)
c = 0
for para, rdnrs in paraAndRdnr.iteritems():
l = self._createListItem(para, rdnrs)
model.insertRow(c, l)
c += 1
model.setHeaderData(0, Qt.Horizontal, "Paragraph")
model.setHeaderData(1, Qt.Horizontal, "Randnummern")
self._paraAndRdnr.setModel(model)
def _createListItem(self, para, rdnrs):
rdnrsAsStr = toStr(collect([item for sublist in rdnrs for item in sublist]))
item1 = QStandardItem(para.decode("utf-8"))
item1.setToolTip(para.decode("utf-8"))
item2 = QStandardItem(rdnrsAsStr)
item2.setToolTip(rdnrsAsStr)
return [item1, item2]
def showDetails(self, content, windowTitle = None):
if windowTitle: self.setWindowTitle(windowTitle)
details = self._map_details(content)
self._listWidget.clear()
i = 0
for fileName, paraAndRdnrs in details.iteritems():
self._listWidget.addItem(fileName)
self.listToParaAndRdnr[i] = paraAndRdnrs
i += 1
def _map_details(self, data):
def nestedMap(data):
res = {}
idx = len(data[0])-1
for d in data:
old = res.get(d[idx], [])
old.append(d[:idx])
res[d[idx]] = old
return res
res = nestedMap(data)
for k, v in res.iteritems():
res[k] = nestedMap(v)
return res
def attb_table(self):
from PyQt4.QtGui import QApplication, QTableView
from qgis.gui import QgsAttributeTableModel
QMessageBox.about(self.view, 'teste', str(self.view.currentLayer()))
cache = QgsVectorLayerCache(self.view.currentLayer(), 10000)
model = QgsAttributeTableModel(cache)
model.loadLayer()
table = QTableView()
table.setModel(model)
table.show()
def __init__(self, parent=None):
super(TableWidget, self).__init__(parent)
# Create a simple model for storing data.
model = CustomTableModel()
# Create the table view and add the model to it.
tableView = QTableView()
tableView.setModel(model)
tableView.setMinimumWidth(300)
tableView.horizontalHeader().setResizeMode(QHeaderView.Stretch)
tableView.verticalHeader().setResizeMode(QHeaderView.Stretch)
chart = QChart()
chart.setAnimationOptions(QChart.AllAnimations)
# Series 1.
series = QBarSeries()
first = 3
count = 5
mapper = QVBarModelMapper(self)
mapper.setFirstBarSetColumn(1)
mapper.setLastBarSetColumn(4)
mapper.setFirstRow(first)
mapper.setRowCount(count)
mapper.setSeries(series)
mapper.setModel(model)
chart.addSeries(series)
# Get the color of the series and use it for showing the mapped area.
for i, barset in enumerate(series.barSets()):
seriesColorHex = '#' + hex(
barset.brush().color().rgb()).upper()[-6:]
model.addMapping(seriesColorHex,
QRect(1 + i, first, 1, barset.count()))
categories = ["April", "May", "June", "July", "August"]
axis = QBarCategoriesAxis(chart)
axis.append(categories)
chart.createDefaultAxes()
chart.setAxisX(axis, series)
chartView = QChartView(chart)
chartView.setRenderHint(QPainter.Antialiasing)
chartView.setMinimumSize(640, 480)
# Create the main layout.
mainLayout = QGridLayout()
mainLayout.addWidget(tableView, 1, 0)
mainLayout.addWidget(chartView, 1, 1)
mainLayout.setColumnStretch(1, 1)
mainLayout.setColumnStretch(0, 0)
self.setLayout(mainLayout)
class MyWindow(QWidget):
def __init__(self, *args):
QWidget.__init__(self, *args)
self.tablemodel = MyTableModel(my_array, self)
self.tableview = QTableView()
self.tableview.setModel(self.tablemodel)
layout = QVBoxLayout(self)
layout.addWidget(self.tableview)
self.setLayout(layout)
def __init__(self, parent=None):
super(Form, self).__init__(parent)
self.resize(640,480)
self.setWindowTitle('Model test')
model = SqliteTableModel(con,'tbl_symbols')
table = QTableView()
table.setModel(model)
lay = QVBoxLayout()
lay.addWidget(table)
self.setLayout(lay)
def __init__(self, parent=None):
super(Form, self).__init__(parent)
self.resize(640, 480)
self.setWindowTitle('Model test')
model = SqliteTableModel(con, 'tbl_symbols')
table = QTableView()
table.setModel(model)
lay = QVBoxLayout()
lay.addWidget(table)
self.setLayout(lay)
def createTable(self, header, tabledata):
"""creates a table"""
gridlayout2 = QGridLayout(self)
gridlayout2.setObjectName("gridlayout2")
tableView = QTableView(self)
tableView.setObjectName("tableView")
gridlayout2.addWidget(tableView, 0, 0, 1, 1)
self.__popUp = QMenu(tableView)
self.__copyAction = QAction(self.tr("Copy data"), tableView)
self.connect(self.__copyAction, SIGNAL("triggered()"), self.copy)
self.__popUp.addAction(self.__copyAction)
tableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.connect(tableView, SIGNAL('customContextMenuRequested(QPoint)'),
self.popUpMenu)
# set the table model
tm = TableModel(tabledata, header, self)
tableView.setModel(tm)
tableView.setAlternatingRowColors(True)
# set the minimum size
self.setMinimumSize(400, 300)
# hide grid
tableView.setShowGrid(True)
# set the font
#font = QFont("Courier New", 12)
#self.tableView.setFont(font)
# hide vertical header
vh = tableView.verticalHeader()
vh.setVisible(True)
# set horizontal header properties
hh = tableView.horizontalHeader()
hh.setStretchLastSection(True)
# set column width to fit contents
tableView.resizeColumnsToContents()
tableView.setSortingEnabled(True)
tableView.sortByColumn(0, Qt.AscendingOrder)
# set row height
nrows = len(tabledata)
for row in xrange(nrows):
tableView.setRowHeight(row, 18)
def setModel(self, model):
QTableView.setModel(self._table, model)
self._table.setSelectionModel(model._selectionModel)
if model.rowCount() > 0:
self.setCurrentIndex(self.PAGE_LISTVIEW)
else:
self.setCurrentIndex(self.PAGE_EMPTY)
model.rowsInserted.connect(self._onRowsChanged)
model.rowsRemoved.connect(self._onRowsChanged)
self._setListViewLook()
class MainWindow(QMainWindow) :
def __init__(self):
super(MainWindow, self).__init__(None)
self.model = QSqlQueryModel(self)
self.model.setQuery("select * from student")
self.model.setHeaderData(0, Qt.Horizontal, self.tr("学号"))
self.model.setHeaderData(1, Qt.Horizontal, self.tr("姓名"))
self.model.setHeaderData(2, Qt.Horizontal, self.tr("课程"))
self.view = QTableView(self)
self.view.setModel(self.model)
self.setCentralWidget(self.view)
def setModel(self, model):
QTableView.setModel(self, model)
# The model needs to be set to set header stuff
hheader = self.horizontalHeader()
hheader.setHighlightSections(False)
hheader.setStretchLastSection(False)
hheader.resizeSection(0, 100)
hheader.setResizeMode(0, QHeaderView.Fixed)
hheader.setResizeMode(1, QHeaderView.Stretch)
hheader.setResizeMode(2, QHeaderView.Stretch)
vheader = self.verticalHeader()
vheader.setVisible(False)
vheader.setDefaultSectionSize(18)
def setModel(self, model):
QTableView.setModel(self, model)
# The model needs to be set to set header stuff
hheader = self.horizontalHeader()
hheader.setHighlightSections(False)
hheader.setStretchLastSection(False)
hheader.resizeSection(0, 100)
hheader.setResizeMode(0, QHeaderView.Fixed)
hheader.setResizeMode(1, QHeaderView.Stretch)
hheader.setResizeMode(2, QHeaderView.Stretch)
vheader = self.verticalHeader()
vheader.setVisible(False)
vheader.setDefaultSectionSize(18)
class PositionWindow(QMainWindow):
WA_Maemo5StackedWindow = 127
WA_Maemo5PortraitOrientation = 128
WA_Maemo5LandscapeOrientation = 129
'''
@param transactionModel - TransactionModel
'''
def __init__(self, parent, exchange, ticker, transactionModel):
QMainWindow.__init__(self, parent)
self.prop = MaeMoneyProperties.instance()
self.setupUi(exchange, ticker, transactionModel)
def setupUi(self, exchange, ticker, transactionModel):
self.setWindowTitle("%s:%s" %(exchange, ticker))
widget = QWidget(self)
self.setCentralWidget(widget)
self.layout = QVBoxLayout()
widget.setLayout(self.layout)
self.transactTableView = QTableView(self)
self.transactTableView.setSelectionBehavior(QAbstractItemView.SelectRows)
header = self.transactTableView.horizontalHeader()
header.setResizeMode(QHeaderView.ResizeToContents)
self.transactTableView.setHorizontalHeader(header)
self.transactTableView.setModel(transactionModel)
self.transactTableView.resizeRowsToContents()
self.transactTableView.resizeColumnsToContents()
header.setStretchLastSection(True)
self.layout.addWidget(self.transactTableView)
self.setOrientation()
self.setAttributeAndCatch(self.WA_Maemo5StackedWindow, True)
def setAttributeAndCatch(self, attribute, trueFalse):
try:
self.setAttribute(attribute, trueFalse)
except AttributeError:
qDebug("Can't set attribute %d" %(attribute))
def setOrientation(self):
if self.prop.isPortraitMode():
self.setAttributeAndCatch(self.WA_Maemo5PortraitOrientation, True)
self.setAttributeAndCatch(self.WA_Maemo5LandscapeOrientation, False)
else:
self.setAttributeAndCatch(self.WA_Maemo5LandscapeOrientation, True)
self.setAttributeAndCatch(self.WA_Maemo5PortraitOrientation, False)
def setModel(self, model):
QTableView.setModel(self._table, model)
self._table.setSelectionModel(model._selectionModel)
if model.rowCount() > 0:
self.setCurrentIndex(self.PAGE_LISTVIEW)
else:
self.setCurrentIndex(self.PAGE_EMPTY)
model.rowsInserted.connect(self._onRowsChanged)
model.rowsRemoved.connect(self._onRowsChanged)
self.model=model
self._setListViewLook()
class QtViewTable(QtControl, ProxyViewTable):
""" A Qt implementation of an Enaml ProxyTable.
"""
#: A reference to the widget created by the proxy.
widget = Typed(QTableView)
#--------------------------------------------------------------------------
# Initialization API
#--------------------------------------------------------------------------
def create_widget(self):
""" Create the underlying QTableView widget.
"""
self.widget = QTableView(self.parent_widget())
self.widget.setAttribute(Qt.WA_StaticContents, True)
def init_widget(self):
""" Initialize the underlying widget.
"""
super(QtViewTable, self).init_widget()
d = self.declaration
self.set_table_model(d.table_model)
self.set_orientation(d.orientation)
#--------------------------------------------------------------------------
# ProxyViewTable API
#--------------------------------------------------------------------------
def set_table_model(self, model):
""" Set the table model for the underlying control.
"""
if model is None:
self.widget.setModel(None)
else:
self.widget.setModel(QItemModelWrapper(model))
def set_orientation(self, orientation):
""" Set the orientation for the widget.
"""
widget = self.widget
if orientation == 'horizontal':
widget.verticalHeader().setVisible(False)
widget.horizontalHeader().setVisible(True)
else:
widget.verticalHeader().setVisible(True)
widget.horizontalHeader().setVisible(False)
class Window(QWidget):
def __init__(self, parent = None):
QWidget.__init__(self, parent)
self.table = QTableView()
self.imageTable = QTableView()
delegate = PixelDelegate(self)
self.imageTable.setItemDelegate(delegate)
self.imageTable.horizontalHeader().hide()
self.imageTable.verticalHeader().hide()
self.imageTable.setShowGrid(False)
self.imageCombo = QComboBox()
self.imageCombo.addItem("Dream", QVariant(":/Pictures/dream.png"))
self.imageCombo.addItem("Teapot", QVariant(":/Pictures/teapot.png"))
gridCheckBox = QCheckBox(self.tr("Show grid:"))
gridCheckBox.setCheckState(Qt.Unchecked)
self.connect(self.imageCombo, SIGNAL("currentIndexChanged(int)"),
self.setModel)
self.connect(gridCheckBox, SIGNAL("toggled(bool)"),
self.imageTable, SLOT("setShowGrid(bool)"))
self.imageCombo.setCurrentIndex(1)
layout = QGridLayout()
layout.addWidget(self.imageTable, 0, 0, 1, 2)
layout.addWidget(self.table, 0, 2, 1, 2)
layout.addWidget(gridCheckBox, 1, 0)
layout.addWidget(self.imageCombo, 1, 1)
self.setLayout(layout)
def setModel(self, row):
image = QImage(self.imageCombo.itemData(row).toString())
model = ImageModel(image, self)
self.table.setModel(model)
self.imageTable.setModel(model)
for row in range(model.rowCount(QModelIndex())):
self.imageTable.resizeRowToContents(row)
for column in range(model.columnCount(QModelIndex())):
self.imageTable.resizeColumnToContents(column)
class DataFrameWidget(QWidget):
''' a simple widget for using DataFrames in a gui '''
def __init__(self, dataFrame, parent=None):
super(DataFrameWidget, self).__init__(parent)
self.dataModel = DataFrameModel()
self.dataModel.setDataFrame(dataFrame)
self.dataTable = QTableView()
self.dataTable.setModel(self.dataModel)
self.dataModel.signalUpdate()
layout = QVBoxLayout()
layout.addWidget(self.dataTable)
self.setLayout(layout)
def resizeColumnsToContents(self):
self.dataTable.resizeColumnsToContents()
class DataFrameWidget(QWidget):
''' a simple widget for using DataFrames in a gui '''
def __init__(self, dataFrame, parent=None):
super(DataFrameWidget, self).__init__(parent)
self.dataModel = DataFrameModel()
self.dataModel.setDataFrame(dataFrame)
self.dataTable = QTableView()
self.dataTable.setModel(self.dataModel)
self.dataModel.signalUpdate()
layout = QVBoxLayout()
layout.addWidget(self.dataTable)
self.setLayout(layout)
def resizeColumnsToContents(self):
self.dataTable.resizeColumnsToContents()
class Oldorder(QtGui.QDialog):
def __init__(self, parent=None):
super(Oldorder,self).__init__(parent)
self.setGeometry(150,150,960,630)
self.setWindowTitle('Old Order')
self.setWindowIcon(QtGui.QIcon('logo1.png'))
self.btn1=QtGui.QPushButton('Retrieve Data',self)
self.btn1.setGeometry(10,10,200,50)
self.btn1.clicked.connect(self.data)
self.orno=QtGui.QLineEdit('',self)
self.orno.setGeometry(220,10,300,30)
self.find=QtGui.QPushButton('FIND',self)
self.find.setGeometry(530,10,100,30)
self.find.clicked.connect(self.act1)
def data(self):
self.db=QSqlDatabase.addDatabase("QSQLITE")
self.db.setDatabaseName("test.db")
self.db.open()
query=QtSql.QSqlQuery("select * from COMPANY2")
self.projectModel=QtSql.QSqlTableModel()
self.projectModel.setQuery(query)
self.projectModel.select()
self.projectView = QTableView(self)
self.projectView.setGeometry(10,60,940,560)
self.projectView.setModel(self.projectModel)
self.projectView.show()
self.db.close()
def act1(self):
orna=self.orno.text()
self.db=QSqlDatabase.addDatabase("QSQLITE")
self.db.setDatabaseName("test.db")
self.db.open()
query=QtSql.QSqlQuery("select * from COMPANY2 WHERE ORDER_NO =+orna+")
self.projectMode2=QtSql.QSqlTableModel()
self.projectMode2.setQuery(query)
self.projectMode2.select()
self.projectView1 = QTableView(self)
self.projectView1.setGeometry(10,60,940,560)
self.projectView1.setModel(self.projectMode2)
self.projectView1.show()
self.db.close()
class MainWindow(QMainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
uic.loadUi("mainwindow.ui", self)
model = QStandardItemModel(7, 4, self)
for row in range(7):
for column in range(4):
item = QStandardItem(QString("%1").arg(row * 4 + column))
model.setItem(row, column, item)
self.tableView = QTableView()
self.tableView.setModel(model)
self.setCentralWidget(self.tableView)
# 获取视图的项目选择模型
selectionModel = self.tableView.selectionModel()
# 定义左上角和右下角的索引，然后使用这两个索引创建选择
topLeft = model.index(1, 1, QModelIndex())
bottomRight = model.index(5, 2, QModelIndex())
selection = QItemSelection(topLeft, bottomRight)
# 使用指定的选择模式来选择项目
selectionModel.select(selection, QItemSelectionModel.Select)
self.mainToolBar.addAction(_fromUtf8("当前项目"), self.getCurrentItemData)
self.mainToolBar.addAction(_fromUtf8("切换选择"), self.toggleSelection)
# 输出当前项目的内容
def getCurrentItemData(self, ):
print "当前项目的内容：", \
self.tableView.selectionModel().currentIndex().data().toString()
# 切换选择的项目
def toggleSelection(self):
topLeft = self.tableView.model().index(0, 0, QModelIndex())
bottomRight = self.tableView.model().index(
self.tableView.model().rowCount(QModelIndex()) - 1,
self.tableView.model().columnCount(QModelIndex()) - 1, QModelIndex())
curSelection = QItemSelection(topLeft, bottomRight)
self.tableView.selectionModel().select(curSelection, QItemSelectionModel.Toggle)
def __init__(self, *args):
QWidget.__init__(self, *args)
self.tablemodel = tablemodel = SqliteTableModel(
[], self)
tableview = QTableView()
tableview.setModel(tablemodel)
dbFileMatch = re.compile(r'.*\.db$')
files = [f for f in os.listdir('.') if os.path.isfile(f)
and dbFileMatch.match(f)]
dbList = self.createComboBoxWithStrings(files)
QObject.connect(dbList,
SIGNAL("currentIndexChanged(QString)"),
self.onDBListChanged)
self.queryList = queryList = self.createComboBoxWithStrings(
SELECT_QUERIES)
self.execListPushButton = QPushButton('Exec')
QObject.connect(self.execListPushButton,
SIGNAL("clicked()"),
self.onExecListPushButtonClicked)
self.queryEdit = QLineEdit()
self.execEditPushButton = QPushButton('Exec')
QObject.connect(self.execEditPushButton,
SIGNAL("clicked()"),
self.onExecEditPushButtonClicked)
layout = QVBoxLayout(self)
layout.addWidget(dbList)
hLayout = QHBoxLayout()
hLayout.addWidget(queryList)
hLayout.addWidget(self.execListPushButton)
layout.addLayout(hLayout)
hLayout = QHBoxLayout()
hLayout.addWidget(self.queryEdit)
hLayout.addWidget(self.execEditPushButton)
layout.addLayout(hLayout)
layout.addWidget(tableview)
self.setLayout(layout)
class DataFrameWidget(QWidget):
''' a simple widget for using DataFrames in a gui '''
def __init__(self,name='DataFrameTable1', parent=None):
super(DataFrameWidget,self).__init__(parent)
self.name = name
self.dataModel = DataFrameModel()
self.dataModel.setDataFrame(DataFrame())
self.dataTable = QTableView()
#self.dataTable.setSelectionBehavior(QAbstractItemView.SelectRows)
self.dataTable.setSortingEnabled(True)
self.dataTable.setModel(self.dataModel)
self.dataModel.signalUpdate()
#self.dataTable.setFont(QFont("Courier New", 8))
layout = QVBoxLayout()
layout.addWidget(self.dataTable)
self.setLayout(layout)
def setFormat(self,fmt):
""" set non-default string formatting for a column """
for colName, f in fmt.iteritems():
self.dataModel.columnFormat[colName]=f
def fitColumns(self):
self.dataTable.horizontalHeader().setResizeMode(QHeaderView.Stretch)
def setDataFrame(self,df):
self.dataModel.setDataFrame(df)
def resizeColumnsToContents(self):
self.dataTable.resizeColumnsToContents()
def insertRow(self,index, data=None):
self.dataModel.appendRow(index,data)
def showClusterTable(self):
if not self.view.sampleTableView.selectedIndexes():#not self.acTree.selectedIndexes():
s, b = QInputDialog.getItem(self.view, "Select one sample", "Select one sample :",
[spl.shortName() for spl in self.model])
if not b:
return
sample = self.model.sample(str(s), fullNameEntry=False)
else:
idx = self.view.sampleTableView.selectedIndexes()[0]
sample = self.model.sample(idx.data().toString(), fullNameEntry=False)
if sample is None:
print ("sample not found...")
return
if not sample.mappedPeaks:
self.view.showErrorMessage("No peaks found",
"This sample does not have peaks, please do peak picking before")
view= QTableView()
view.setSortingEnabled(True)
view.horizontalHeader().setStretchLastSection(True)
view.setModel(MSDialogController.getSampleModel(sample, flags='cluster'))
self.view.addMdiSubWindow(view, "ClusterList of%s"%str(sample.shortName()))
class Editor(QDialog):
def __init__(self, model, parent=None):
super().__init__(parent)
self._view = QTableView()
self._view.setModel(model)
cancelButton = QPushButton(_("cancel"))
cancelButton.clicked.connect(self.reject)
okButton = QPushButton(_("OK"))
okButton.setDefault(True)
okButton.clicked.connect(self.accept)
addButton = QPushButton("+")
addButton.clicked.connect(self.add)
buttonBox = QHBoxLayout()
buttonBox.addWidget(addButton)
buttonBox.addStretch(1)
buttonBox.addWidget(cancelButton)
buttonBox.addWidget(okButton)
layout = QVBoxLayout()
layout.addWidget(self._view)
layout.addLayout(buttonBox)
self.setLayout(layout)
self.accepted.connect(self.save)
self.setWindowTitle(_("Admin.edit" + model.table))
self.resize(600, 300)
self._view.resizeColumnsToContents()
def save(self):
self._view.model().save()
def add(self):
raise NotImplementedError()
class MTestForm(QDialog):
def __init__(self, parent=None):
super(MTestForm, self).__init__(parent)
self.model = SnapshotTableModel()
self.model.addSnapshot('test.h5')
self.tablev = QTableView()
self.tablev.setModel(self.model)
self.tablev.setItemDelegate(SnapshotItemDelegate(self))
#for i in range(self.model.columnCount()):
self.tablev.resizeColumnsToContents()
self.tablev.setColumnHidden(1, True)
hb1 = QHBoxLayout()
vb2 = QVBoxLayout()
vb2.addWidget(ApPlotWidget(), 1.0)
vb3a = QVBoxLayout()
cb = QCheckBox("show PV")
self.connect(cb, SIGNAL("stateChanged(int)"), self._set_pv_view)
vb3a.addWidget(cb)
vb3a.addStretch()
vb3b = QVBoxLayout()
vb3b.addWidget(QPushButton("Refresh"))
vb3b.addWidget(QPushButton("Plot"))
vb3b.addWidget(QPushButton("Ramp"))
vb3b.addStretch()
hb2 = QHBoxLayout()
hb2.addLayout(vb3a, 1.0)
hb2.addLayout(vb3b)
vb2.addLayout(hb2)
hb1.addWidget(self.tablev, .6)
hb1.addLayout(vb2, 0.4)
self.setLayout(hb1)
def _set_pv_view(self, st):
if st == Qt.Checked: self.tablev.setColumnHidden(1, False)
elif st == Qt.Unchecked: self.tablev.setColumnHidden(1, True)
class Editor(QDialog):
def __init__(self, model, parent=None):
super().__init__(parent)
self._view = QTableView()
self._view.setModel(model)
cancelButton = QPushButton(_('cancel'))
cancelButton.clicked.connect(self.reject)
okButton = QPushButton(_('OK'))
okButton.setDefault(True)
okButton.clicked.connect(self.accept)
addButton = QPushButton('+')
addButton.clicked.connect(self.add)
buttonBox = QHBoxLayout()
buttonBox.addWidget(addButton)
buttonBox.addStretch(1)
buttonBox.addWidget(cancelButton)
buttonBox.addWidget(okButton)
layout = QVBoxLayout()
layout.addWidget(self._view)
layout.addLayout(buttonBox)
self.setLayout(layout)
self.accepted.connect(self.save)
self.setWindowTitle(_('Admin.edit' + model.table))
self.resize(600, 300)
self._view.resizeColumnsToContents()
def save(self):
self._view.model().save()
def add(self):
raise NotImplementedError()
class MTestForm(QDialog):
def __init__(self, parent=None):
super(MTestForm, self).__init__(parent)
self.model = SnapshotTableModel()
self.model.addSnapshot('test.h5')
self.tablev = QTableView()
self.tablev.setModel(self.model)
self.tablev.setItemDelegate(SnapshotItemDelegate(self))
#for i in range(self.model.columnCount()):
self.tablev.resizeColumnsToContents()
self.tablev.setColumnHidden(1, True)
hb1 = QHBoxLayout()
vb2 = QVBoxLayout()
vb2.addWidget(ApPlotWidget(), 1.0)
vb3a = QVBoxLayout()
cb = QCheckBox("show PV")
self.connect(cb, SIGNAL("stateChanged(int)"), self._set_pv_view)
vb3a.addWidget(cb)
vb3a.addStretch()
vb3b = QVBoxLayout()
vb3b.addWidget(QPushButton("Refresh"))
vb3b.addWidget(QPushButton("Plot"))
vb3b.addWidget(QPushButton("Ramp"))
vb3b.addStretch()
hb2 = QHBoxLayout()
hb2.addLayout(vb3a, 1.0)
hb2.addLayout(vb3b)
vb2.addLayout(hb2)
hb1.addWidget(self.tablev, .6)
hb1.addLayout(vb2, 0.4)
self.setLayout(hb1)
def _set_pv_view(self, st):
if st == Qt.Checked: self.tablev.setColumnHidden(1, False)
elif st == Qt.Unchecked: self.tablev.setColumnHidden(1, True)
def __init__(self, parent=None):
super(MainWindow, self).__init__(None)
uic.loadUi("mainwindow.ui", self)
model = QStandardItemModel(7, 4, self)
for row in range(7):
for column in range(4):
item = QStandardItem(QString("%1").arg(row * 4 + column))
model.setItem(row, column, item)
tableView = QTableView()
tableView.setModel(model)
self.setCentralWidget(tableView)
#获取视图的项目选择模型
selectionModel = tableView.selectionModel()
# 定义左上角和右下角的索引，然后使用这两个索引创建选择
topLeft = model.index(1, 1, QModelIndex())
bottomRight = model.index(5, 2, QModelIndex())
selection = QItemSelection(topLeft, bottomRight)
# 使用指定的选择模式来选择项目
selectionModel.select(selection, QItemSelectionModel.Select)
class OrderDlg(QDialog):
def __init__(self, parent=None):
super(OrderDlg, self).__init__(parent)
self.create_widgets()
self.layout_widgets()
self.setMinimumWidth(850)
self.setWindowTitle(u'Список Order Item')
def create_widgets(self):
self.model = QtSql.QSqlRelationalTableModel(self)
self.model.setTable("CustomerOrder")
self.model.setEditStrategy(QtSql.QSqlTableModel.OnManualSubmit)
self.model.setRelation(CUSTOMER, QtSql.QSqlRelation('customerview', 'CustomerID', 'Name'))
self.model.setSort(ID, Qt.AscendingOrder)
self.model.setHeaderData(ID, Qt.Horizontal, QVariant("Order"))
self.model.setHeaderData(DATE, Qt.Horizontal, QVariant("DATE"))
self.model.setHeaderData(TIME, Qt.Horizontal, QVariant("TIME"))
self.model.setHeaderData(CUSTOMER, Qt.Horizontal,
QVariant("CUSTOMER"))
self.model.select()
self.view = QTableView()
self.view.setModel(self.model)
self.view.resizeColumnsToContents()
def layout_widgets(self):
layout = QVBoxLayout()
layout.addWidget(self.view)
self.setLayout(layout)
def on_copy(self):
select_group_dialog = QDialog(self)
select_group_dialog.resize(300, 400)
select_group_dialog.setWindowTitle(self.tr("Choose source group"))
layout = QVBoxLayout(select_group_dialog)
select_group_dialog.setLayout(layout)
groups_list_view = QTableView(self)
layout.addWidget(groups_list_view)
groups_list_view.setModel(self.ds_model)
groups_list_view.setColumnHidden(DSManagerModel.COLUMN_VISIBILITY,
True)
groups_list_view.horizontalHeader().setResizeMode(
DSManagerModel.COLUMN_GROUP_DS, QHeaderView.Stretch)
groups_list_view.setSelectionMode(QTableView.NoSelection)
groups_list_view.setAlternatingRowColors(True)
groups_list_view.setShowGrid(False)
groups_list_view.verticalHeader().setResizeMode(
QHeaderView.ResizeToContents)
groups_list_view.verticalHeader().hide()
groups_list_view.clicked.connect(
lambda index: select_group_dialog.accept() \
if self.ds_model.isGroup(index) and \
index.column() == DSManagerModel.COLUMN_GROUP_DS \
else None
)
if select_group_dialog.exec_() == QDialog.Accepted:
group_info = self.ds_model.data(groups_list_view.currentIndex(),
Qt.UserRole)
group_info.id += "_copy"
edit_dialog = GroupEditDialog()
edit_dialog.setWindowTitle(self.tr('Create group from existing'))
edit_dialog.fill_group_info(group_info)
if edit_dialog.exec_() == QDialog.Accepted:
self.feel_list()
self.ds_model.resetModel()
class OrderDlg(QDialog):
def __init__(self, parent=None):
super(OrderDlg, self).__init__(parent)
self.create_widgets()
self.layout_widgets()
self.setMinimumWidth(850)
self.setWindowTitle(u'Список Order Item')
def create_widgets(self):
self.model = QtSql.QSqlRelationalTableModel(self)
self.model.setTable("CustomerOrder")
self.model.setEditStrategy(QtSql.QSqlTableModel.OnManualSubmit)
self.model.setRelation(
CUSTOMER, QtSql.QSqlRelation('customerview', 'CustomerID', 'Name'))
self.model.setSort(ID, Qt.AscendingOrder)
self.model.setHeaderData(ID, Qt.Horizontal, QVariant("Order"))
self.model.setHeaderData(DATE, Qt.Horizontal, QVariant("DATE"))
self.model.setHeaderData(TIME, Qt.Horizontal, QVariant("TIME"))
self.model.setHeaderData(CUSTOMER, Qt.Horizontal, QVariant("CUSTOMER"))
self.model.select()
self.view = QTableView()
self.view.setModel(self.model)
self.view.resizeColumnsToContents()
def layout_widgets(self):
layout = QVBoxLayout()
layout.addWidget(self.view)
self.setLayout(layout)
class StaffDataDlg(QDialog):
def __init__(self, parent=None):
super(StaffDataDlg, self).__init__(parent)
self.create_widgets()
self.layout_widgets()
self.setMinimumWidth(850)
self.setWindowTitle(u'Список сотрудников компании')
def create_widgets(self):
self.model = QSqlTableModel(self)
self.view = QTableView()
self.view.setModel(self.model)
self.view.setSelectionMode(QTableView.SingleSelection)
self.view.setSelectionBehavior(QTableView.SelectRows)
self.view.resizeColumnsToContents()
self.buttonBox = QDialogButtonBox()
self.addButton = self.buttonBox.addButton(u"&Добавить",
QDialogButtonBox.ActionRole)
self.deleteButton = self.buttonBox.addButton(u"&Удалить",
QDialogButtonBox.ActionRole)
self.sortButton = self.buttonBox.addButton(u"&Сортировать",
QDialogButtonBox.ActionRole)
def layout_widgets(self):
layout = QVBoxLayout()
layout.addWidget(self.view)
layout.addWidget(self.buttonBox)
self.setLayout(layout)
class WCSWindow(QtGui.QWidget):
def __init__(self):
super(WCSWindow, self).__init__()
self.initUI()
def initUI(self):
self.resize(800, 600)
self.center()
self.setWindowTitle('Center')
tablemodel = CoverageTableModel(my_array, self)
self.tableview = QTableView()
self.tableview.setModel(tablemodel)
layout = QVBoxLayout(self)
layout.addWidget(self.tableview)
self.setLayout(layout)
self.show()
self.tableview.doubleClicked.connect(self.coverageDoubleClick)
self.combobox = QComboBox()
self.combobox.addItem("AAAAA")
self.combobox.addItem("BBBBB")
layout.addWidget(self.combobox)
self.combobox.activated.connect(self.changedCombo)
def coverageDoubleClick(self, mi):
row = mi.row()
col = mi.column()
print(row, col)
def changedCombo(self, ob):
print(self.combobox.currentText())
def center(self):
qr = self.frameGeometry()
cp = QtGui.QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
class QtTableView(QtControl, ProxyTableView):
""" A Qt implementation of an Enaml ProxyTableView.
"""
#: A reference to the widget created by the proxy.
widget = Typed(QTableView)
#--------------------------------------------------------------------------
# Initialization API
#--------------------------------------------------------------------------
def create_widget(self):
""" Create the underlying widget.
"""
self.widget = QTableView(self.parent_widget())
self.widget.setAttribute(Qt.WA_StaticContents, True)
def init_widget(self):
""" Create and initialize the underlying control.
"""
super(QtTableView, self).init_widget()
d = self.declaration
self.set_item_model(d.item_model)
#--------------------------------------------------------------------------
# ProxyListView API
#--------------------------------------------------------------------------
def set_item_model(self, model):
""" Set the item model for the widget.
"""
if not model:
self.widget.setModel(None)
else:
self.widget.setModel(QItemModelWrapper(model))
def createTable(self):
# create the view
tableview = QTableView()
# set table model
tablemodel = ArrayDataTableModel(self.savedValue, self.liveValue, self)
tableview.setModel(tablemodel)
# hide vertical header
vh = tableview.verticalHeader()
vh.setVisible(False)
# set horizontal header properties
hh = tableview.horizontalHeader()
hh.setStretchLastSection(True)
# set column width to fit contents
tableview.resizeColumnsToContents()
# enable sorting
# does not work
# tableview.setSortingEnabled(True)
return tableview
def createTable(self):
# create the view
tableview = QTableView()
# set table model
tablemodel = ArrayDataTableModel(self.savedValue, self.liveValue, self)
tableview.setModel(tablemodel)
# hide vertical header
vh = tableview.verticalHeader()
vh.setVisible(False)
# set horizontal header properties
hh = tableview.horizontalHeader()
hh.setStretchLastSection(True)
# set column width to fit contents
tableview.resizeColumnsToContents()
# enable sorting
# does not work
# tableview.setSortingEnabled(True)
return tableview
class OWMPR(OWWidget):
name = 'ModelMap Projection Rank'
description = 'Ranking projections by estimating projection quality'
icon = "icons/ModelMap.svg"
inputs = [('Data', Table, 'set_data', Default)]
outputs = [('Features', AttributeList)]
want_main_area = False
settingsHandler = DomainContextHandler()
variable_changed = Signal()
def __init__(self):
super().__init__()
self.data = None
self.progress = None
self.infoa = gui.widgetLabel(self.controlArea, "No data loaded.")
self.projectionTable = QTableView()
self.controlArea.layout().addWidget(self.projectionTable)
self.projectionTable.setSelectionBehavior(QTableView.SelectRows)
self.projectionTable.setSelectionMode(QTableView.SingleSelection)
self.projectionTable.setSortingEnabled(True)
self.projectionTableModel = QStandardItemModel(self)
self.projectionTableModel.setHorizontalHeaderLabels(["P-Index", "", ""])
self.projectionTable.setModel(self.projectionTableModel)
self.projectionTable.setColumnWidth(0, 90)
self.projectionTable.sortByColumn(0, Qt.DescendingOrder)
self.projectionTable.selectionModel().selectionChanged.connect(self.on_selection_changed)
gui.button(self.controlArea, self, "Rank Projections", callback=self.rank, default=True)
self.resize(370, 600)
def set_data(self, data):
self.data = data
self.infoa.setText("Data set: {}".format(data.name) if self.data else "No data loaded.")
def rank(self):
if self.progress:
return
disc = Orange.preprocess.discretize.EqualWidth(n=10)
ndomain = Orange.data.Domain(
[disc(self.data, attr) if type(attr) == Orange.data.variable.ContinuousVariable
else attr for attr in self.data.domain.attributes], self.data.domain.class_vars)
t = self.data.from_table(ndomain, self.data)
attrs = t.domain.attributes
tables = {}
l = 0
self.progress = gui.ProgressBar(self, len(attrs) * (len(attrs) - 1) / 2)
for i in range(len(attrs)):
for j in range(i):
ct = np.array(contingency.get_contingency(t, attrs[j], attrs[i]))
pindex, _, _ = p_index(ct)
tables[i, j] = ct
item = QStandardItem()
item.setData(float(pindex), Qt.DisplayRole)
self.projectionTableModel.setItem(l, 0, item)
item = QStandardItem()
item.setData(attrs[i].name, Qt.DisplayRole)
self.projectionTableModel.setItem(l, 1, item)
item = QStandardItem()
item.setData(attrs[j].name, Qt.DisplayRole)
self.projectionTableModel.setItem(l, 2, item)
self.progress.advance()
l += 1
self.progress.finish()
self.progress = None
def on_selection_changed(self, selected, deselected):
"""Called when the ranks view selection changes."""
a1 = selected.indexes()[1].data().replace('D_', '')
a2 = selected.indexes()[2].data().replace('D_', '')
d = self.data.domain
self.send("Features", AttributeList([d[a1], d[a2]]))
class DlgSelectInvoice( QDialog ):
def __init__( self, parent = None ):
super( DlgSelectInvoice, self ).__init__( parent )
self.billsmodel = QSqlQueryModel()
query = """
SELECT * FROM (
SELECT
factura.iddocumento,
CONCAT_WS(' ', tdc.descripcion, factura.ndocimpreso) AS 'Numero de Factura',
factura.fechacreacion AS 'Fecha',
p.nombre AS 'Cliente',
-SUM(axd.unidades) -
IFNULL((
SELECT
SUM(axddev.unidades)
FROM documentos devoluciones
JOIN docpadrehijos dpddev ON devoluciones.iddocumento = dpddev.idhijo
JOIN articulosxdocumento axddev ON axddev.iddocumento = devoluciones.iddocumento
WHERE devoluciones.idtipodoc = %d AND dpddev.idpadre = factura.iddocumento
GROUP BY dpddev.idpadre
),0) as unittotal,
p.idpersona,
ca.valorcosto,
ca.idcostoagregado,
tc.tasa,
tc.idtc,
b.nombrebodega AS 'Bodega',
b.idbodega
FROM documentos factura
JOIN bodegas b ON factura.idbodega = b.idbodega
JOIN tiposdoc tdc ON tdc.idtipodoc = factura.idtipodoc
JOIN articulosxdocumento axd ON axd.iddocumento = factura.iddocumento AND factura.idtipodoc = %d
JOIN tiposcambio tc ON tc.idtc = factura.idtipocambio
JOIN personasxdocumento pxd ON pxd.iddocumento = factura.iddocumento
JOIN personas p ON pxd.idpersona = p.idpersona AND p.tipopersona=%d
LEFT JOIN costosxdocumento cxd ON cxd.iddocumento = factura.iddocumento
LEFT JOIN costosagregados ca ON cxd.idcostoagregado = ca.idcostoagregado
JOIN (
SELECT
dpdk.idpadre
FROM documentos kardex
JOIN docpadrehijos dpdk ON kardex.iddocumento = dpdk.idhijo
WHERE kardex.idtipodoc = %d
) as kardex ON kardex.idpadre = factura.iddocumento
GROUP BY factura.iddocumento
) as tbl
WHERE unittotal > 0
""" % ( constantes.IDNC, constantes.IDFACTURA, constantes.CLIENTE, constantes.IDKARDEX )
self.billsmodel.setQuery( query )
self.setWindowTitle( "Seleccione la factura para la devolucion" )
self.filtermodel = QSortFilterProxyModel()
self.filtermodel.setSourceModel( self.billsmodel )
self.filtermodel.setFilterCaseSensitivity( Qt.CaseInsensitive )
self.filtermodel.setFilterKeyColumn( -1 )
iddoc, _ndocimpreso, _fechacreacion, _nombre, total, idpersona, valorcosto, idcosto, tasacambio, idcambio, _nombrebodega, idbodega = range( 12 )
self.tblBills = QTableView()
self.tblBills.setSelectionMode( QAbstractItemView.SingleSelection )
self.tblBills.setSelectionBehavior( QAbstractItemView.SelectRows )
self.tblBills.selectRow( 0 )
self.tblBills.setModel( self.filtermodel )
self.tblBills.setColumnHidden( iddoc, True )
self.tblBills.setColumnHidden( idpersona, True )
self.tblBills.setColumnHidden( total, True )
self.tblBills.setColumnHidden( valorcosto, True )
self.tblBills.setColumnHidden( idcosto, True )
self.tblBills.setColumnHidden( idcambio, True )
self.tblBills.setColumnHidden( tasacambio, True )
self.tblBills.setColumnHidden( idbodega, True )
self.tblBills.horizontalHeader().setStretchLastSection( True )
self.tblBills.resizeColumnsToContents()
buttonbox = QDialogButtonBox( QDialogButtonBox.Ok | QDialogButtonBox.Cancel )
self.txtSearch = QLineEdit()
formlayout = QFormLayout()
formlayout.addRow( "&Buscar", self.txtSearch )
layout = QVBoxLayout()
layout.addWidget( self.tblBills )
layout.addLayout( formlayout )
layout.addWidget( buttonbox )
self.setLayout( layout )
self.setMinimumWidth( 400 )
buttonbox.accepted.connect( self.accept )
buttonbox.rejected.connect( self.reject )
self.txtSearch.textChanged[unicode].connect( self.updateFilter )
#FIXME: Que pasa cuando no hay facturas?
# def exec_( self ):
# if self.billsmodel.rowCount() == 0:
# QMessageBox.critical( None,
# self.trUtf8( "Llantera Esquipulas: Inventario" ),
# self.trUtf8( """No hay facturas a las cuales hacer devoluciones""" ),
# QMessageBox.StandardButtons( \
# QMessageBox.Ok ) )
# self.reject()
# else:
# QDialog.exec_( self )
def updateFilter( self, str ):
self.filtermodel.setFilterWildcard( str )
class TalkEditorApp(FreeseerApp):
'''Freeseer talk database editor main gui class'''
def __init__(self, config, db):
super(TalkEditorApp, self).__init__(config)
self.db = db
icon = QIcon()
icon.addPixmap(QPixmap(':/freeseer/logo.png'), QIcon.Normal, QIcon.Off)
self.setWindowIcon(icon)
self.resize(960, 600)
#
# Setup Layout
#
self.mainWidget = QWidget()
self.mainLayout = QVBoxLayout()
self.mainWidget.setLayout(self.mainLayout)
self.setCentralWidget(self.mainWidget)
self.mainLayout.setAlignment(Qt.AlignTop)
# Add custom widgets
self.commandButtons = CommandButtons()
self.tableView = QTableView()
self.tableView.setSortingEnabled(True)
self.tableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.talkDetailsWidget = TalkDetailsWidget()
self.importTalksWidget = ImportTalksWidget()
self.newTalkWidget = NewTalkWidget()
self.mainLayout.addWidget(self.importTalksWidget)
#self.mainLayout.addLayout(self.titleLayout)
self.mainLayout.addWidget(self.commandButtons)
self.mainLayout.addWidget(self.tableView)
self.mainLayout.addWidget(self.talkDetailsWidget)
self.mainLayout.addWidget(self.importTalksWidget)
# --- End Layout
# Keep track of index of the most recently selected talk
self.currentTalkIndex = QPersistentModelIndex()
# Prompt user to "Continue Editing", "Discard Changes" or "Save Changes"
self.savePromptBox = QMessageBox()
self.savePromptBox.setWindowTitle("Unsaved Changes Exist")
self.savePromptBox.setIcon(QMessageBox.Information)
self.savePromptBox.setText(
"The talk you were editing has unsaved changes.")
self.continueButton = self.savePromptBox.addButton(
"Continue Editing", QMessageBox.RejectRole)
self.discardButton = self.savePromptBox.addButton(
"Discard Changes", QMessageBox.DestructiveRole)
self.saveButton = self.savePromptBox.addButton("Save Changes",
QMessageBox.AcceptRole)
self.savePromptBox.setDefaultButton(self.saveButton)
# Initialize geometry, to be used for restoring window positioning.
self.geometry = None
#
# Setup Menubar
#
self.actionExportCsv = QAction(self)
self.actionExportCsv.setObjectName('actionExportCsv')
self.actionRemoveAll = QAction(self)
self.actionRemoveAll.setObjectName('actionRemoveAll')
# Actions
self.menuFile.insertAction(self.actionExit, self.actionExportCsv)
self.menuFile.insertAction(self.actionExit, self.actionRemoveAll)
# --- End Menubar
#
# TableView Connections
#
self.connect(self.tableView, SIGNAL('activated(const QModelIndex)'),
self.click_talk)
self.connect(self.tableView, SIGNAL('selected(const QModelIndex)'),
self.click_talk)
self.connect(self.tableView, SIGNAL('clicked(const QModelIndex)'),
self.click_talk)
# Import Widget
self.connect(self.importTalksWidget.csvRadioButton,
SIGNAL('toggled(bool)'), self.toggle_import)
self.connect(self.importTalksWidget.importButton, SIGNAL('clicked()'),
self.import_talks)
self.connect(self.importTalksWidget.cancelButton, SIGNAL('clicked()'),
self.hide_import_talks_widget)
self.importTalksWidget.setHidden(True)
self.connect(self.importTalksWidget.csvFileSelectButton,
SIGNAL('clicked()'), self.csv_file_select)
self.connect(self.importTalksWidget.csvLineEdit,
SIGNAL('returnPressed()'),
self.importTalksWidget.importButton.click)
self.connect(self.importTalksWidget.rssLineEdit,
SIGNAL('returnPressed()'),
self.importTalksWidget.importButton.click)
self.connect(self.actionExportCsv, SIGNAL('triggered()'),
self.export_talks_to_csv)
self.connect(self.actionRemoveAll, SIGNAL('triggered()'),
self.confirm_reset)
# Command Buttons
self.connect(self.commandButtons.addButton, SIGNAL('clicked()'),
self.click_add_button)
self.connect(self.commandButtons.removeButton, SIGNAL('clicked()'),
self.remove_talk)
self.connect(self.commandButtons.removeAllButton, SIGNAL('clicked()'),
self.confirm_reset)
self.connect(self.commandButtons.importButton, SIGNAL('clicked()'),
self.show_import_talks_widget)
self.connect(self.commandButtons.exportButton, SIGNAL('clicked()'),
self.export_talks_to_csv)
self.connect(self.commandButtons.searchButton, SIGNAL('clicked()'),
self.search_talks)
self.connect(self.commandButtons.searchLineEdit,
SIGNAL('textEdited(QString)'), self.search_talks)
self.connect(self.commandButtons.searchLineEdit,
SIGNAL('returnPressed()'), self.search_talks)
# Talk Details Buttons
self.connect(self.talkDetailsWidget.saveButton, SIGNAL('clicked()'),
self.update_talk)
# Talk Details Widget
self.connect(self.talkDetailsWidget.titleLineEdit,
SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.presenterLineEdit,
SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.categoryLineEdit,
SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.eventLineEdit,
SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.roomLineEdit,
SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.descriptionTextEdit,
SIGNAL('modificationChanged(bool)'), self.enable_save)
self.connect(self.talkDetailsWidget.dateEdit,
SIGNAL('dateChanged(const QDate)'), self.enable_save)
self.connect(self.talkDetailsWidget.startTimeEdit,
SIGNAL('timeChanged(const QTime)'), self.enable_save)
self.connect(self.talkDetailsWidget.endTimeEdit,
SIGNAL('timeChanged(const QTime)'), self.enable_save)
# New Talk Widget
self.newTalkWidget.connect(self.newTalkWidget.addButton,
SIGNAL('clicked()'), self.add_talk)
self.newTalkWidget.connect(self.newTalkWidget.cancelButton,
SIGNAL('clicked()'),
self.newTalkWidget.reject)
# Load default language
actions = self.menuLanguage.actions()
for action in actions:
if action.data().toString() == self.config.default_language:
action.setChecked(True)
self.translate(action)
break
# Load Talk Database
self.load_presentations_model()
# Setup Autocompletion
self.update_autocomplete_fields()
self.talkDetailsWidget.saveButton.setEnabled(False)
# Select first item
#self.tableView.setCurrentIndex(self.proxy.index(0,0))
#self.talk_selected(self.proxy.index(0,0))
#
# Translation
#
def retranslate(self):
self.setWindowTitle(
self.app.translate("TalkEditorApp", "Freeseer Talk Editor"))
#
# Reusable Strings
#
self.confirmDBClearTitleString = self.app.translate(
"TalkEditorApp", "Remove All Talks from Database")
self.confirmDBClearQuestionString = self.app.translate(
"TalkEditorApp", "Are you sure you want to clear the DB?")
self.confirmTalkDetailsClearTitleString = self.app.translate(
"TalkEditorApp", "Unsaved Data")
self.confirmTalkDetailsClearQuestionString = self.app.translate(
"TalkEditorApp", "Unsaved talk details will be lost. Continue?")
# --- End Reusable Strings
#
# Menubar
#
self.actionExportCsv.setText(
self.app.translate("TalkEditorApp", "&Export to CSV"))
self.actionRemoveAll.setText(
self.app.translate("TalkEditorApp", "&Remove All Talks"))
# --- End Menubar
#
# TalkDetailsWidget
#
self.talkDetailsWidget.titleLabel.setText(
self.app.translate("TalkEditorApp", "Title"))
self.talkDetailsWidget.presenterLabel.setText(
self.app.translate("TalkEditorApp", "Presenter"))
self.talkDetailsWidget.categoryLabel.setText(
self.app.translate("TalkEditorApp", "Category"))
self.talkDetailsWidget.eventLabel.setText(
self.app.translate("TalkEditorApp", "Event"))
self.talkDetailsWidget.roomLabel.setText(
self.app.translate("TalkEditorApp", "Room"))
self.talkDetailsWidget.dateLabel.setText(
self.app.translate("TalkEditorApp", "Date"))
self.talkDetailsWidget.startTimeLabel.setText(
self.app.translate("TalkEditorApp", "Start Time"))
self.talkDetailsWidget.endTimeLabel.setText(
self.app.translate("TalkEditorApp", "End Time"))
# --- End TalkDetailsWidget
#
# Import Talks Widget Translations
#
self.importTalksWidget.rssRadioButton.setText(
self.app.translate("TalkEditorApp", "RSS URL"))
self.importTalksWidget.csvRadioButton.setText(
self.app.translate("TalkEditorApp", "CSV File"))
self.importTalksWidget.importButton.setText(
self.app.translate("TalkEditorApp", "Import"))
# --- End Talks Widget Translations
#
# Command Button Translations\
#
self.commandButtons.importButton.setText(
self.app.translate("TalkEditorApp", "Import"))
self.commandButtons.exportButton.setText(
self.app.translate("TalkEditorApp", "Export"))
self.commandButtons.addButton.setText(
self.app.translate("TalkEditorApp", "Add New Talk"))
self.commandButtons.removeButton.setText(
self.app.translate("TalkEditorApp", "Remove"))
self.commandButtons.removeAllButton.setText(
self.app.translate("TalkEditorApp", "Remove All"))
# --- End Command Butotn Translations
#
# Search Widget Translations
#
self.commandButtons.searchButton.setText(
self.app.translate("TalkEditorApp", "Search"))
# --- End Command Button Translations
def load_presentations_model(self):
# Load Presentation Model
self.presentationModel = self.db.get_presentations_model()
self.proxy = QSortFilterProxyModel()
self.proxy.setSourceModel(self.presentationModel)
self.tableView.setModel(self.proxy)
self.proxy.setFilterCaseSensitivity(Qt.CaseInsensitive)
# Fill table whitespace.
self.tableView.horizontalHeader().setStretchLastSection(False)
self.tableView.horizontalHeader().setResizeMode(1, QHeaderView.Stretch)
# Hide the ID field
self.tableView.setColumnHidden(0, True)
# Map data to widgets
self.mapper = QDataWidgetMapper()
self.mapper.setModel(self.proxy)
self.mapper.setSubmitPolicy(QDataWidgetMapper.ManualSubmit)
self.mapper.addMapping(self.talkDetailsWidget.titleLineEdit, 1)
self.mapper.addMapping(self.talkDetailsWidget.presenterLineEdit, 2)
self.mapper.addMapping(self.talkDetailsWidget.categoryLineEdit, 4)
self.mapper.addMapping(self.talkDetailsWidget.eventLineEdit, 5)
self.mapper.addMapping(self.talkDetailsWidget.roomLineEdit, 6)
self.mapper.addMapping(self.talkDetailsWidget.descriptionTextEdit, 3)
self.mapper.addMapping(self.talkDetailsWidget.dateEdit, 7)
self.mapper.addMapping(self.talkDetailsWidget.startTimeEdit, 8)
self.mapper.addMapping(self.talkDetailsWidget.endTimeEdit, 9)
# Load StringLists
self.titleList = QStringList(self.db.get_string_list("Title"))
#self.speakerList = QStringList(self.db.get_speaker_list())
#self.categoryList = QStringList(self.db.get_category_list())
#self.eventList = QStringList(self.db.get_event_list())
#self.roomList = QStringList(self.db.get_room_list())
#Disble input
self.talkDetailsWidget.disable_input_fields()
def search_talks(self):
# The default value is 0. If the value is -1, the keys will be read from all columns.
self.proxy.setFilterKeyColumn(-1)
self.proxy.setFilterFixedString(
self.commandButtons.searchLineEdit.text())
def show_save_prompt(self):
"""Prompts the user to save or discard changes, or continue editing."""
self.savePromptBox.exec_()
self.savePromptBox.setDefaultButton(self.saveButton)
return self.savePromptBox.clickedButton()
def click_talk(self, model):
"""Warns user if there are unsaved changes, and selects talk clicked by the user."""
log.info("Selecting row %d", model.row())
modelRow = model.row()
if self.unsaved_details_exist():
log.info("Unsaved changes exist in row %d",
self.currentTalkIndex.row())
confirm = self.show_save_prompt()
if confirm == self.saveButton:
log.info("Saving changes in row %d...",
self.currentTalkIndex.row())
self.tableView.selectRow(self.currentTalkIndex.row())
self.update_talk()
newModel = self.tableView.currentIndex().sibling(modelRow, 0)
self.select_talk(newModel)
elif confirm == self.discardButton:
log.info("Discarding changes in row %d...",
self.currentTalkIndex.row())
self.talk_selected(model)
else:
log.info("Continue editing row %d",
self.currentTalkIndex.row())
self.tableView.selectRow(self.currentTalkIndex.row())
else:
self.talk_selected(model)
def click_add_button(self):
"""Warns user if there are unsaved changes, and shows the New Talk window."""
if self.unsaved_details_exist():
log.info("Unsaved changes exist in row %d",
self.currentTalkIndex.row())
confirm = self.show_save_prompt()
if confirm == self.saveButton:
log.info("Saving changes in row %d...",
self.currentTalkIndex.row())
self.update_talk()
self.show_new_talk_popup()
elif confirm == self.discardButton:
log.info("Discarding changes in row %d...",
self.currentTalkIndex.row())
# Ensure that changes are discarded
self.talk_selected(self.currentTalkIndex)
self.show_new_talk_popup()
else:
log.info("Continue editing row %d",
self.currentTalkIndex.row())
else:
self.show_new_talk_popup()
def talk_selected(self, model):
self.mapper.setCurrentIndex(model.row())
self.talkDetailsWidget.enable_input_fields()
self.talkDetailsWidget.saveButton.setEnabled(False)
self.currentTalkIndex = QPersistentModelIndex(model)
def toggle_import(self):
if self.importTalksWidget.csvRadioButton.isChecked():
self.importTalksWidget.csvLineEdit.setEnabled(True)
self.importTalksWidget.csvFileSelectButton.setEnabled(True)
self.importTalksWidget.rssLineEdit.setEnabled(False)
else:
self.importTalksWidget.csvLineEdit.setEnabled(False)
self.importTalksWidget.csvFileSelectButton.setEnabled(False)
self.importTalksWidget.rssLineEdit.setEnabled(True)
def show_import_talks_widget(self):
self.commandButtons.setHidden(True)
self.tableView.setHidden(True)
self.talkDetailsWidget.setHidden(True)
self.importTalksWidget.setHidden(False)
def hide_import_talks_widget(self):
self.commandButtons.setHidden(False)
self.tableView.setHidden(False)
self.talkDetailsWidget.setHidden(False)
self.importTalksWidget.setHidden(True)
def add_talk(self):
"""Adds a new talk to the database using data from the NewTalkWidget input fields"""
presentation = self.create_presentation(
self.newTalkWidget.talkDetailsWidget)
if presentation:
self.db.insert_presentation(presentation)
self.newTalkWidget.accept() # Close the dialog
def update_talk(self):
"""Updates the currently selected talk using data from the TalkEditorApp input fields"""
selected_talk = self.tableView.currentIndex()
if selected_talk.row(
) >= 0: # The tableView index begins at 0 and is -1 by default
talk_id = selected_talk.sibling(selected_talk.row(),
0).data().toString()
presentation = self.create_presentation(self.talkDetailsWidget)
if presentation:
self.db.update_presentation(talk_id, presentation)
self.apply_changes(selected_talk)
self.talkDetailsWidget.saveButton.setEnabled(False)
def create_presentation(self, talkDetailsWidget):
"""Creates and returns an instance of Presentation using data from the input fields"""
date = talkDetailsWidget.dateEdit.date()
startTime = talkDetailsWidget.startTimeEdit.time()
endTime = talkDetailsWidget.endTimeEdit.time()
title = unicode(talkDetailsWidget.titleLineEdit.text()).strip()
if title:
return Presentation(
unicode(talkDetailsWidget.titleLineEdit.text()).strip(),
unicode(talkDetailsWidget.presenterLineEdit.text()).strip(),
unicode(talkDetailsWidget.descriptionTextEdit.toPlainText()).
strip(),
unicode(talkDetailsWidget.categoryLineEdit.text()).strip(),
unicode(talkDetailsWidget.eventLineEdit.text()).strip(),
unicode(talkDetailsWidget.roomLineEdit.text()).strip(),
unicode(date.toString(Qt.ISODate)),
unicode(startTime.toString(Qt.ISODate)),
unicode(endTime.toString(Qt.ISODate)))
def show_new_talk_popup(self):
"""Displays a modal dialog with a talk details view
When Add is selected, a new talk is added to the database using the input field data.
When Cancel is selected, no talk is added.
"""
log.info('Opening Add Talk window...')
self.clear_new_talk_fields()
self.remove_new_talk_placeholder_text()
self.newTalkWidget.talkDetailsWidget.titleLineEdit.setFocus()
if self.newTalkWidget.exec_() == 1:
self.apply_changes()
self.talkDetailsWidget.disable_input_fields()
else:
log.info('No talk added...')
def apply_changes(self, updated_talk=None):
"""Repopulates the model to display the effective changes
Updates the autocomplete fields.
Displays the updated model in the table view, and selects the newly updated/added talk.
"""
self.presentationModel.select()
self.select_talk(updated_talk)
self.update_autocomplete_fields()
def select_talk(self, talk=None):
"""Selects the given talk in the table view
If no talk is given, the last row in the table view is selected.
"""
if talk:
row = talk.row()
column = talk.column()
else:
row = self.presentationModel.rowCount() - 1 # Select last row
column = 0
self.tableView.selectRow(row)
self.tableView.setCurrentIndex(self.proxy.index(row, column))
self.talk_selected(self.proxy.index(row, column))
def remove_talk(self):
try:
rows_selected = self.tableView.selectionModel().selectedRows()
except:
return
# Reversed because rows in list change position once row is removed
for row in reversed(rows_selected):
self.presentationModel.removeRow(row.row())
self.talkDetailsWidget.clear_input_fields()
self.talkDetailsWidget.disable_input_fields()
def load_talk(self):
try:
self.tableView.currentIndex().row()
except:
return
self.mapper.addMapping(self.talkDetailsWidget.roomLineEdit, 6)
self.presentationModel.select()
def reset(self):
self.db.clear_database()
self.presentationModel.select()
self.talkDetailsWidget.clear_input_fields()
self.talkDetailsWidget.disable_input_fields()
def confirm_reset(self):
"""Presents a confirmation dialog to ask the user if they are sure they wish to remove the talk database.
If Yes call the reset() function"""
confirm = QMessageBox.question(self, self.confirmDBClearTitleString,
self.confirmDBClearQuestionString,
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if confirm == QMessageBox.Yes:
self.reset()
def add_talks_from_rss(self):
rss_url = unicode(self.importTalksWidget.rssLineEdit.text())
if rss_url:
self.db.add_talks_from_rss(rss_url)
self.presentationModel.select()
self.hide_import_talks_widget()
else:
error = QMessageBox()
error.setText("Please enter a RSS URL")
error.exec_()
def closeEvent(self, event):
if self.unsaved_details_exist():
log.info("Unsaved changes exist in row %d",
self.currentTalkIndex.row())
confirm = self.show_save_prompt()
if confirm == self.saveButton:
log.info("Saving changes in row %d...",
self.currentTalkIndex.row())
self.update_talk()
log.info('Exiting talk database editor...')
self.geometry = self.saveGeometry()
event.accept()
elif confirm == self.discardButton:
log.info("Discarding changes in row %d...",
self.currentTalkIndex.row())
# Ensure that changes are discarded
self.talk_selected(self.currentTalkIndex)
log.info('Exiting talk database editor...')
self.geometry = self.saveGeometry()
event.accept()
else:
log.info("Continue editing row %d",
self.currentTalkIndex.row())
event.ignore()
else:
log.info('Exiting talk database editor...')
self.geometry = self.saveGeometry()
event.accept()
def csv_file_select(self):
fname = QFileDialog.getOpenFileName(self, 'Select file', "", "*.csv")
if fname:
self.importTalksWidget.csvLineEdit.setText(fname)
def add_talks_from_csv(self):
fname = self.importTalksWidget.csvLineEdit.text()
if fname:
self.db.add_talks_from_csv(fname)
self.presentationModel.select()
self.hide_import_talks_widget()
else:
error = QMessageBox()
error.setText("Please select a file")
error.exec_()
def import_talks(self):
if self.importTalksWidget.csvRadioButton.isChecked():
self.add_talks_from_csv()
else:
self.add_talks_from_rss()
self.update_autocomplete_fields()
def export_talks_to_csv(self):
fname = QFileDialog.getSaveFileName(self, 'Select file', "", "*.csv")
if fname:
self.db.export_talks_to_csv(fname)
def update_autocomplete_fields(self):
self.titleList = QStringList(self.db.get_string_list("Title"))
self.speakerList = QStringList(self.db.get_string_list("Speaker"))
self.categoryList = QStringList(self.db.get_string_list("Category"))
self.eventList = QStringList(self.db.get_string_list("Event"))
self.roomList = QStringList(self.db.get_string_list("Room"))
self.titleCompleter = QCompleter(self.titleList)
self.titleCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.speakerCompleter = QCompleter(self.speakerList)
self.speakerCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.categoryCompleter = QCompleter(self.categoryList)
self.categoryCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.eventCompleter = QCompleter(self.eventList)
self.eventCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.roomCompleter = QCompleter(self.roomList)
self.roomCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.talkDetailsWidget.titleLineEdit.setCompleter(self.titleCompleter)
self.talkDetailsWidget.presenterLineEdit.setCompleter(
self.speakerCompleter)
self.talkDetailsWidget.categoryLineEdit.setCompleter(
self.categoryCompleter)
self.talkDetailsWidget.eventLineEdit.setCompleter(self.eventCompleter)
self.talkDetailsWidget.roomLineEdit.setCompleter(self.roomCompleter)
def are_fields_enabled(self):
return (self.talkDetailsWidget.titleLineEdit.isEnabled()
and self.talkDetailsWidget.presenterLineEdit.isEnabled()
and self.talkDetailsWidget.categoryLineEdit.isEnabled()
and self.talkDetailsWidget.eventLineEdit.isEnabled()
and self.talkDetailsWidget.roomLineEdit.isEnabled()
and self.talkDetailsWidget.dateEdit.isEnabled()
and self.talkDetailsWidget.startTimeEdit.isEnabled()
and self.talkDetailsWidget.endTimeEdit.isEnabled())
def unsaved_details_exist(self):
"""Checks if changes have been made to new/existing talk details
Looks for text in the input fields and check the enabled state of the Save Talk button
If the Save Talk button is enabled, the input fields contain modified values
"""
return (self.talkDetailsWidget.saveButton.isEnabled() and
(self.talkDetailsWidget.titleLineEdit.text()
or self.talkDetailsWidget.presenterLineEdit.text()
or self.talkDetailsWidget.categoryLineEdit.text()
or self.talkDetailsWidget.descriptionTextEdit.toPlainText()))
def enable_save(self):
self.talkDetailsWidget.saveButton.setEnabled(True)
def clear_new_talk_fields(self):
"""Removes existing data from all NewTalkWidget fields except event, room, date and time"""
self.newTalkWidget.talkDetailsWidget.titleLineEdit.clear()
self.newTalkWidget.talkDetailsWidget.presenterLineEdit.clear()
self.newTalkWidget.talkDetailsWidget.descriptionTextEdit.clear()
self.newTalkWidget.talkDetailsWidget.categoryLineEdit.clear()
def remove_new_talk_placeholder_text(self):
"""Removes placeholder text in NewTalkWidget originally set by TalkDetailsWidget"""
self.newTalkWidget.talkDetailsWidget.titleLineEdit.setPlaceholderText(
"")
self.newTalkWidget.talkDetailsWidget.presenterLineEdit.setPlaceholderText(
"")
self.newTalkWidget.talkDetailsWidget.categoryLineEdit.setPlaceholderText(
"")
self.newTalkWidget.talkDetailsWidget.eventLineEdit.setPlaceholderText(
"")
self.newTalkWidget.talkDetailsWidget.roomLineEdit.setPlaceholderText(
"")
class TalkEditorApp(FreeseerApp):
'''Freeseer talk database editor main gui class'''
def __init__(self, config, db):
FreeseerApp.__init__(self)
self.config = config
self.db = db
icon = QIcon()
icon.addPixmap(QPixmap(':/freeseer/logo.png'), QIcon.Normal, QIcon.Off)
self.setWindowIcon(icon)
self.resize(960, 600)
#
# Setup Layout
#
self.mainWidget = QWidget()
self.mainLayout = QVBoxLayout()
self.mainWidget.setLayout(self.mainLayout)
self.setCentralWidget(self.mainWidget)
self.mainLayout.setAlignment(Qt.AlignTop)
# Add custom widgets
self.commandButtons = CommandButtons()
self.tableView = QTableView()
self.tableView.setSortingEnabled(True)
self.tableView.setSelectionBehavior(QAbstractItemView.SelectRows)
self.talkDetailsWidget = TalkDetailsWidget()
self.importTalksWidget = ImportTalksWidget()
self.newTalkWidget = NewTalkWidget()
self.mainLayout.addWidget(self.importTalksWidget)
#self.mainLayout.addLayout(self.titleLayout)
self.mainLayout.addWidget(self.commandButtons)
self.mainLayout.addWidget(self.tableView)
self.mainLayout.addWidget(self.talkDetailsWidget)
self.mainLayout.addWidget(self.importTalksWidget)
# --- End Layout
# Initialize geometry, to be used for restoring window positioning.
self.geometry = None
#
# Setup Menubar
#
self.actionExportCsv = QAction(self)
self.actionExportCsv.setObjectName('actionExportCsv')
self.actionRemoveAll = QAction(self)
self.actionRemoveAll.setObjectName('actionRemoveAll')
# Actions
self.menuFile.insertAction(self.actionExit, self.actionExportCsv)
self.menuFile.insertAction(self.actionExit, self.actionRemoveAll)
# --- End Menubar
#
# TableView Connections
#
self.connect(self.tableView, SIGNAL('activated(const QModelIndex)'), self.talk_selected)
self.connect(self.tableView, SIGNAL('selected(const QModelIndex)'), self.talk_selected)
self.connect(self.tableView, SIGNAL('clicked(const QModelIndex)'), self.talk_selected)
# Import Widget
self.connect(self.importTalksWidget.csvRadioButton, SIGNAL('toggled(bool)'), self.toggle_import)
self.connect(self.importTalksWidget.importButton, SIGNAL('clicked()'), self.import_talks)
self.connect(self.importTalksWidget.cancelButton, SIGNAL('clicked()'), self.hide_import_talks_widget)
self.importTalksWidget.setHidden(True)
self.connect(self.importTalksWidget.csvFileSelectButton, SIGNAL('clicked()'), self.csv_file_select)
self.connect(self.importTalksWidget.csvLineEdit, SIGNAL('returnPressed()'),
self.importTalksWidget.importButton.click)
self.connect(self.importTalksWidget.rssLineEdit, SIGNAL('returnPressed()'),
self.importTalksWidget.importButton.click)
self.connect(self.actionExportCsv, SIGNAL('triggered()'), self.export_talks_to_csv)
self.connect(self.actionRemoveAll, SIGNAL('triggered()'), self.confirm_reset)
# Command Buttons
self.connect(self.commandButtons.addButton, SIGNAL('clicked()'), self.show_new_talk_popup)
self.connect(self.commandButtons.removeButton, SIGNAL('clicked()'), self.remove_talk)
self.connect(self.commandButtons.removeAllButton, SIGNAL('clicked()'), self.confirm_reset)
self.connect(self.commandButtons.importButton, SIGNAL('clicked()'), self.show_import_talks_widget)
self.connect(self.commandButtons.exportButton, SIGNAL('clicked()'), self.export_talks_to_csv)
self.connect(self.commandButtons.searchButton, SIGNAL('clicked()'), self.search_talks)
self.connect(self.commandButtons.searchLineEdit, SIGNAL('textEdited(QString)'), self.search_talks)
self.connect(self.commandButtons.searchLineEdit, SIGNAL('returnPressed()'), self.search_talks)
# Talk Details Buttons
self.connect(self.talkDetailsWidget.saveButton, SIGNAL('clicked()'), self.update_talk)
# Talk Details Widget
self.connect(self.talkDetailsWidget.titleLineEdit, SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.presenterLineEdit, SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.categoryLineEdit, SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.eventLineEdit, SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.roomLineEdit, SIGNAL('textEdited(const QString)'), self.enable_save)
self.connect(self.talkDetailsWidget.descriptionTextEdit, SIGNAL('modificationChanged(bool)'), self.enable_save)
self.connect(self.talkDetailsWidget.dateEdit, SIGNAL('dateChanged(const QDate)'), self.enable_save)
self.connect(self.talkDetailsWidget.startTimeEdit, SIGNAL('timeChanged(const QTime)'), self.enable_save)
self.connect(self.talkDetailsWidget.endTimeEdit, SIGNAL('timeChanged(const QTime)'), self.enable_save)
# New Talk Widget
self.newTalkWidget.connect(self.newTalkWidget.addButton, SIGNAL('clicked()'), self.add_talk)
self.newTalkWidget.connect(self.newTalkWidget.cancelButton, SIGNAL('clicked()'), self.newTalkWidget.reject)
# Load default language
actions = self.menuLanguage.actions()
for action in actions:
if action.data().toString() == self.config.default_language:
action.setChecked(True)
self.translate(action)
break
# Load Talk Database
self.load_presentations_model()
# Setup Autocompletion
self.update_autocomplete_fields()
self.talkDetailsWidget.saveButton.setEnabled(False)
# Select first item
#self.tableView.setCurrentIndex(self.proxy.index(0,0))
#self.talk_selected(self.proxy.index(0,0))
#
# Translation
#
def retranslate(self):
self.setWindowTitle(self.app.translate("TalkEditorApp", "Freeseer Talk Editor"))
#
# Reusable Strings
#
self.confirmDBClearTitleString = self.app.translate("TalkEditorApp", "Remove All Talks from Database")
self.confirmDBClearQuestionString = self.app.translate("TalkEditorApp",
"Are you sure you want to clear the DB?")
self.confirmTalkDetailsClearTitleString = self.app.translate("TalkEditorApp", "Unsaved Data")
self.confirmTalkDetailsClearQuestionString = self.app.translate("TalkEditorApp",
"Unsaved talk details will be lost. Continue?")
# --- End Reusable Strings
#
# Menubar
#
self.actionExportCsv.setText(self.app.translate("TalkEditorApp", "&Export to CSV"))
self.actionRemoveAll.setText(self.app.translate("TalkEditorApp", "&Remove All Talks"))
# --- End Menubar
#
# TalkDetailsWidget
#
self.talkDetailsWidget.titleLabel.setText(self.app.translate("TalkEditorApp", "Title"))
self.talkDetailsWidget.presenterLabel.setText(self.app.translate("TalkEditorApp", "Presenter"))
self.talkDetailsWidget.categoryLabel.setText(self.app.translate("TalkEditorApp", "Category"))
self.talkDetailsWidget.eventLabel.setText(self.app.translate("TalkEditorApp", "Event"))
self.talkDetailsWidget.roomLabel.setText(self.app.translate("TalkEditorApp", "Room"))
self.talkDetailsWidget.dateLabel.setText(self.app.translate("TalkEditorApp", "Date"))
self.talkDetailsWidget.startTimeLabel.setText(self.app.translate("TalkEditorApp", "Start Time"))
self.talkDetailsWidget.endTimeLabel.setText(self.app.translate("TalkEditorApp", "End Time"))
# --- End TalkDetailsWidget
#
# Import Talks Widget Translations
#
self.importTalksWidget.rssRadioButton.setText(self.app.translate("TalkEditorApp", "RSS URL"))
self.importTalksWidget.csvRadioButton.setText(self.app.translate("TalkEditorApp", "CSV File"))
self.importTalksWidget.importButton.setText(self.app.translate("TalkEditorApp", "Import"))
# --- End Talks Widget Translations
#
# Command Button Translations\
#
self.commandButtons.importButton.setText(self.app.translate("TalkEditorApp", "Import"))
self.commandButtons.exportButton.setText(self.app.translate("TalkEditorApp", "Export"))
self.commandButtons.addButton.setText(self.app.translate("TalkEditorApp", "Add New Talk"))
self.commandButtons.removeButton.setText(self.app.translate("TalkEditorApp", "Remove"))
self.commandButtons.removeAllButton.setText(self.app.translate("TalkEditorApp", "Remove All"))
# --- End Command Butotn Translations
#
# Search Widget Translations
#
self.commandButtons.searchButton.setText(self.app.translate("TalkEditorApp", "Search"))
# --- End Command Button Translations
def load_presentations_model(self):
# Load Presentation Model
self.presentationModel = self.db.get_presentations_model()
self.proxy = QSortFilterProxyModel()
self.proxy.setSourceModel(self.presentationModel)
self.tableView.setModel(self.proxy)
self.proxy.setFilterCaseSensitivity(Qt.CaseInsensitive)
# Fill table whitespace.
self.tableView.horizontalHeader().setStretchLastSection(False)
self.tableView.horizontalHeader().setResizeMode(1, QHeaderView.Stretch)
# Hide the ID field
self.tableView.setColumnHidden(0, True)
# Map data to widgets
self.mapper = QDataWidgetMapper()
self.mapper.setModel(self.proxy)
self.mapper.setSubmitPolicy(QDataWidgetMapper.ManualSubmit)
self.mapper.addMapping(self.talkDetailsWidget.titleLineEdit, 1)
self.mapper.addMapping(self.talkDetailsWidget.presenterLineEdit, 2)
self.mapper.addMapping(self.talkDetailsWidget.categoryLineEdit, 4)
self.mapper.addMapping(self.talkDetailsWidget.eventLineEdit, 5)
self.mapper.addMapping(self.talkDetailsWidget.roomLineEdit, 6)
self.mapper.addMapping(self.talkDetailsWidget.descriptionTextEdit, 3)
self.mapper.addMapping(self.talkDetailsWidget.dateEdit, 7)
self.mapper.addMapping(self.talkDetailsWidget.startTimeEdit, 8)
self.mapper.addMapping(self.talkDetailsWidget.endTimeEdit, 9)
# Load StringLists
self.titleList = QStringList(self.db.get_string_list("Title"))
#self.speakerList = QStringList(self.db.get_speaker_list())
#self.categoryList = QStringList(self.db.get_category_list())
#self.eventList = QStringList(self.db.get_event_list())
#self.roomList = QStringList(self.db.get_room_list())
#Disble input
self.talkDetailsWidget.disable_input_fields()
def search_talks(self):
# The default value is 0. If the value is -1, the keys will be read from all columns.
self.proxy.setFilterKeyColumn(-1)
self.proxy.setFilterFixedString(self.commandButtons.searchLineEdit.text())
def talk_selected(self, model):
self.mapper.setCurrentIndex(model.row())
self.talkDetailsWidget.enable_input_fields()
self.talkDetailsWidget.saveButton.setEnabled(False)
def toggle_import(self):
if self.importTalksWidget.csvRadioButton.isChecked():
self.importTalksWidget.csvLineEdit.setEnabled(True)
self.importTalksWidget.csvFileSelectButton.setEnabled(True)
self.importTalksWidget.rssLineEdit.setEnabled(False)
else:
self.importTalksWidget.csvLineEdit.setEnabled(False)
self.importTalksWidget.csvFileSelectButton.setEnabled(False)
self.importTalksWidget.rssLineEdit.setEnabled(True)
def show_import_talks_widget(self):
self.commandButtons.setHidden(True)
self.tableView.setHidden(True)
self.talkDetailsWidget.setHidden(True)
self.importTalksWidget.setHidden(False)
def hide_import_talks_widget(self):
self.commandButtons.setHidden(False)
self.tableView.setHidden(False)
self.talkDetailsWidget.setHidden(False)
self.importTalksWidget.setHidden(True)
def add_talk(self):
"""Adds a new talk to the database using data from the NewTalkWidget input fields"""
presentation = self.create_presentation(self.newTalkWidget.talkDetailsWidget)
if presentation:
self.db.insert_presentation(presentation)
self.newTalkWidget.accept() # Close the dialog
def update_talk(self):
"""Updates the currently selected talk using data from the TalkEditorApp input fields"""
selected_talk = self.tableView.currentIndex()
if selected_talk.row() >= 0: # The tableView index begins at 0 and is -1 by default
talk_id = selected_talk.sibling(selected_talk.row(), 0).data().toString()
presentation = self.create_presentation(self.talkDetailsWidget)
if presentation:
self.db.update_presentation(talk_id, presentation)
self.apply_changes(selected_talk)
self.talkDetailsWidget.saveButton.setEnabled(False)
def create_presentation(self, talkDetailsWidget):
"""Creates and returns an instance of Presentation using data from the input fields"""
date = talkDetailsWidget.dateEdit.date()
startTime = talkDetailsWidget.startTimeEdit.time()
endTime = talkDetailsWidget.endTimeEdit.time()
title = unicode(talkDetailsWidget.titleLineEdit.text()).strip()
if title:
return Presentation(
unicode(talkDetailsWidget.titleLineEdit.text()).strip(),
unicode(talkDetailsWidget.presenterLineEdit.text()).strip(),
unicode(talkDetailsWidget.descriptionTextEdit.toPlainText()).strip(),
unicode(talkDetailsWidget.categoryLineEdit.text()).strip(),
unicode(talkDetailsWidget.eventLineEdit.text()).strip(),
unicode(talkDetailsWidget.roomLineEdit.text()).strip(),
unicode(date.toString(Qt.ISODate)),
unicode(startTime.toString(Qt.ISODate)),
unicode(endTime.toString(Qt.ISODate)))
def show_new_talk_popup(self):
"""Displays a modal dialog with a talk details view
When Add is selected, a new talk is added to the database using the input field data.
When Cancel is selected, no talk is added.
"""
log.info('Opening Add Talk window...')
self.clear_new_talk_fields()
self.remove_new_talk_placeholder_text()
self.newTalkWidget.talkDetailsWidget.titleLineEdit.setFocus()
if self.newTalkWidget.exec_() == 1:
self.apply_changes()
self.talkDetailsWidget.disable_input_fields()
else:
log.info('No talk added...')
def apply_changes(self, updated_talk=None):
"""Repopulates the model to display the effective changes
Updates the autocomplete fields.
Displays the updated model in the table view, and selects the newly updated/added talk.
"""
self.presentationModel.select()
self.select_talk(updated_talk)
self.update_autocomplete_fields()
def select_talk(self, talk=None):
"""Selects the given talk in the table view
If no talk is given, the last row in the table view is selected.
"""
if talk:
row = talk.row()
column = talk.column()
else:
row = self.presentationModel.rowCount() - 1 # Select last row
column = 0
self.tableView.selectRow(row)
self.tableView.setCurrentIndex(self.proxy.index(row, column))
self.talk_selected(self.proxy.index(row, column))
def remove_talk(self):
try:
rows_selected = self.tableView.selectionModel().selectedRows()
except:
return
# Reversed because rows in list change position once row is removed
for row in reversed(rows_selected):
self.presentationModel.removeRow(row.row())
def load_talk(self):
try:
self.tableView.currentIndex().row()
except:
return
self.mapper.addMapping(self.talkDetailsWidget.roomLineEdit, 6)
self.presentationModel.select()
def reset(self):
self.db.clear_database()
self.presentationModel.select()
def confirm_reset(self):
"""Presents a confirmation dialog to ask the user if they are sure they wish to remove the talk database.
If Yes call the reset() function"""
confirm = QMessageBox.question(self,
self.confirmDBClearTitleString,
self.confirmDBClearQuestionString,
QMessageBox.Yes |
QMessageBox.No,
QMessageBox.No)
if confirm == QMessageBox.Yes:
self.reset()
def add_talks_from_rss(self):
rss_url = unicode(self.importTalksWidget.rssLineEdit.text())
if rss_url:
self.db.add_talks_from_rss(rss_url)
self.presentationModel.select()
self.hide_import_talks_widget()
else:
error = QMessageBox()
error.setText("Please enter a RSS URL")
error.exec_()
def closeEvent(self, event):
log.info('Exiting talk database editor...')
self.geometry = self.saveGeometry()
event.accept()
def csv_file_select(self):
fname = QFileDialog.getOpenFileName(
self, 'Select file', "", "*.csv")
if fname:
self.importTalksWidget.csvLineEdit.setText(fname)
def add_talks_from_csv(self):
fname = self.importTalksWidget.csvLineEdit.text()
if fname:
self.db.add_talks_from_csv(fname)
self.presentationModel.select()
self.hide_import_talks_widget()
else:
error = QMessageBox()
error.setText("Please select a file")
error.exec_()
def import_talks(self):
if self.importTalksWidget.csvRadioButton.isChecked():
self.add_talks_from_csv()
else:
self.add_talks_from_rss()
self.update_autocomplete_fields()
def export_talks_to_csv(self):
fname = QFileDialog.getSaveFileName(self, 'Select file', "", "*.csv")
if fname:
self.db.export_talks_to_csv(fname)
def update_autocomplete_fields(self):
self.titleList = QStringList(self.db.get_string_list("Title"))
self.speakerList = QStringList(self.db.get_string_list("Speaker"))
self.categoryList = QStringList(self.db.get_string_list("Category"))
self.eventList = QStringList(self.db.get_string_list("Event"))
self.roomList = QStringList(self.db.get_string_list("Room"))
self.titleCompleter = QCompleter(self.titleList)
self.titleCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.speakerCompleter = QCompleter(self.speakerList)
self.speakerCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.categoryCompleter = QCompleter(self.categoryList)
self.categoryCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.eventCompleter = QCompleter(self.eventList)
self.eventCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.roomCompleter = QCompleter(self.roomList)
self.roomCompleter.setCaseSensitivity(Qt.CaseInsensitive)
self.talkDetailsWidget.titleLineEdit.setCompleter(self.titleCompleter)
self.talkDetailsWidget.presenterLineEdit.setCompleter(self.speakerCompleter)
self.talkDetailsWidget.categoryLineEdit.setCompleter(self.categoryCompleter)
self.talkDetailsWidget.eventLineEdit.setCompleter(self.eventCompleter)
self.talkDetailsWidget.roomLineEdit.setCompleter(self.roomCompleter)
def are_fields_enabled(self):
return (self.talkDetailsWidget.titleLineEdit.isEnabled() and
self.talkDetailsWidget.presenterLineEdit.isEnabled() and
self.talkDetailsWidget.categoryLineEdit.isEnabled() and
self.talkDetailsWidget.eventLineEdit.isEnabled() and
self.talkDetailsWidget.roomLineEdit.isEnabled() and
self.talkDetailsWidget.dateEdit.isEnabled() and
self.talkDetailsWidget.startTimeEdit.isEnabled() and
self.talkDetailsWidget.endTimeEdit.isEnabled())
def unsaved_details_exist(self):
"""Checks if changes have been made to new/existing talk details
Looks for text in the input fields and check the enabled state of the Save Talk button
If the Save Talk button is enabled, the input fields contain modified values
"""
return (self.talkDetailsWidget.saveButton.isEnabled() and
(self.talkDetailsWidget.titleLineEdit.text() or
self.talkDetailsWidget.presenterLineEdit.text() or
self.talkDetailsWidget.categoryLineEdit.text() or
self.talkDetailsWidget.descriptionTextEdit.toPlainText()))
def enable_save(self):
self.talkDetailsWidget.saveButton.setEnabled(True)
def clear_new_talk_fields(self):
"""Removes existing data from all NewTalkWidget fields except event, room, date and time"""
self.newTalkWidget.talkDetailsWidget.titleLineEdit.clear()
self.newTalkWidget.talkDetailsWidget.presenterLineEdit.clear()
self.newTalkWidget.talkDetailsWidget.descriptionTextEdit.clear()
self.newTalkWidget.talkDetailsWidget.categoryLineEdit.clear()
def remove_new_talk_placeholder_text(self):
"""Removes placeholder text in NewTalkWidget originally set by TalkDetailsWidget"""
self.newTalkWidget.talkDetailsWidget.titleLineEdit.setPlaceholderText("")
self.newTalkWidget.talkDetailsWidget.presenterLineEdit.setPlaceholderText("")
self.newTalkWidget.talkDetailsWidget.categoryLineEdit.setPlaceholderText("")
self.newTalkWidget.talkDetailsWidget.eventLineEdit.setPlaceholderText("")
self.newTalkWidget.talkDetailsWidget.roomLineEdit.setPlaceholderText("")
class ComponentSearchForm():
data_source = None
related_mashups = None
widget = None
table = None
add_btn = None
graph_form = None
highlighted_api = None
highlighted_mashup = None
def show_main_window(self):
self.graph_form = matplotlibWidget()
if self.widget:
self.widget.destroy()
self.widget = None;
self.widget = QWidget()
self.widget.setMinimumSize(800, 600)
btn_api = QtGui.QPushButton("Recommend Modules", self.widget)
btn_api.move(30, 20)
btn_mashup = QtGui.QPushButton("Recommend Workflows", self.widget)
btn_mashup.move(200, 20)
self.textboxLabel = QLabel(self.widget)
self.textboxLabel.setText("Describe your goals:")
self.textboxLabel.move(35, 60)
self.textboxLabel.show
self.textbox = QTextEdit(self.widget)
self.textbox.move(30, 80)
self.textbox.setFixedWidth(300)
self.textbox.setFixedHeight(28)
btn_api.clicked.connect(self.api_button_clicked)
btn_mashup.clicked.connect(self.mashups_button_clicked)
self.table = QTableView(self.widget)
self.table.clicked.connect(self.table_clicked)
self.table.setMinimumSize(740, 500)
self.table.resizeColumnsToContents()
self.table.move(30, 120)
self.widget.show()
self.add_btn = QPushButton(self.widget)
self.add_btn.clicked.connect(self.add_new_api)
self.add_btn.setText("Add to Palette")
self.add_btn.hide()
self.add_btn.move(650, 20)
self.recommendLabel = QLabel(self.widget)
self.recommendLabel.setText("Also Used")
self.recommendLabel.move(30, 95)
self.recommendLabel.hide()
self.switch_btn_apis = QPushButton(self.widget)
self.switch_btn_apis.clicked.connect(self._show_related_apis)
self.switch_btn_apis.setText("Related Mashup")
self.switch_btn_apis.move(500, 20)
self.switch_btn_apis.hide()
self.switch_btn_mashups = QPushButton(self.widget)
self.switch_btn_mashups.clicked.connect(self._show_related_mashups)
self.switch_btn_mashups.setText("Related API")
self.switch_btn_mashups.move(500, 20)
self.switch_btn_mashups.hide()
def __init__(self, parent=None):
self.data_source = DataSource()
def table_clicked(self):
"""
Click the table, the graph form may change according to the selection.
"""
model = self.table.selectionModel()
indexes = model.selectedIndexes()
for index in indexes:
row = index.row()
data = index.model().headerData(0,Qt.Horizontal).toString()
newIndex = index.model().index(row, 0)
if data == "API":
api_id = get_api_full_name(newIndex.model().data(newIndex).toString())
api = self.data_source.api_by_id(api_id)
print api
mashups = self.data_source.mashups_by_api(api)
apis = []
for mashup in mashups:
apis.extend(self.data_source.apis_by_mashup(mashup))
self.graph_form.draw_apis(apis, api, self.highlighted_api)
else:
mashup_id = get_mashup_full_name(newIndex.model().data(newIndex).toString())
mashup = self.data_source.mashup_by_id(mashup_id)
if not mashup:
return
apis = self.data_source.apis_by_mashup(mashup)
mashups = []
if len(apis) > 0:
mashups.extend(self.data_source.mashups_by_api(apis[0]))
self.graph_form.draw_mashups(mashups, mashup, self.highlighted_mashup)
return
def _show_apis(self, apis, recommend=False):
self.switch_btn_apis.hide()
self.switch_btn_mashups.hide()
self.recommendLabel.hide()
row = 0
model = QStandardItemModel(len(apis), 4)
model.setColumnCount(4)
for api in apis:
model.setData(model.index(row, 0), QVariant(get_api_name(api)))
model.setData(model.index(row, 1), QVariant(api['protocols']))
model.setData(model.index(row, 2), QVariant(api['provider']))
model.setData(model.index(row, 3), QVariant(api['version']))
row += 1
model.setHeaderData(0, Qt.Horizontal, QVariant("Module"))
model.setHeaderData(1, Qt.Horizontal, QVariant("Protocol"))
model.setHeaderData(2, Qt.Horizontal, QVariant("Provider"))
model.setHeaderData(3, Qt.Horizontal, QVariant("Version"))
# model.setHeaderData(0, Qt.Horizontal, QVariant("API"))
# model.setHeaderData(1, Qt.Horizontal, QVariant("Protocols"))
# model.setHeaderData(2, Qt.Horizontal, QVariant("Provider"))
# model.setHeaderData(3, Qt.Horizontal, QVariant("Version"))
self.table.setModel(model)
self.table.resizeColumnsToContents()
if recommend:
self.recommendLabel.show()
self.add_btn.show()
def _show_mashups(self, mashups):
self.switch_btn_apis.hide()
self.switch_btn_mashups.hide()
self.recommendLabel.hide()
row = 0
model = QStandardItemModel(len(mashups), 4)
model.setColumnCount(4)
for mashup in mashups:
model.setData(model.index(row, 0), QVariant(get_mashup_name(mashup)))
model.setData(model.index(row, 1), QVariant(mashup['title']))
model.setData(model.index(row, 2), QVariant(mashup['self']))
model.setData(model.index(row, 3), QVariant(mashup['description']))
row += 1
model.setHeaderData(0, Qt.Horizontal, QVariant("Workflow"))
model.setHeaderData(1, Qt.Horizontal, QVariant("Short Description"))
model.setHeaderData(2, Qt.Horizontal, QVariant("Provider"))
model.setHeaderData(3, Qt.Horizontal, QVariant("Detailed Info"))
# model.setHeaderData(0, Qt.Horizontal, QVariant("Info"))
# model.setHeaderData(1, Qt.Horizontal, QVariant("Title"))
# model.setHeaderData(2, Qt.Horizontal, QVariant("self"))
# model.setHeaderData(3, Qt.Horizontal, QVariant("Description"))
self.table.setModel(model)
self.table.resizeColumnsToContents()
self.add_btn.show()
def api_button_clicked(self):
"""
Trigger to search APIs
"""
self.graph_form.draw_api()
self.graph_form.show()
apis = self.data_source.apis()
key = str(self.textbox.toPlainText())
#Has key or not has key, it is different.
if key:
self.api_search_button_clicked()
else:
self._show_apis(apis)
def mashups_button_clicked(self):
"""
Trigger to search mashups
"""
self.graph_form.draw_mashup()
self.graph_form.show()
key = str(self.textbox.toPlainText())
if key:
self.mashup_search_button_clicked()
else:
self._show_mashups(self.data_source.mashups())
#Should probably refactor this into one method.
def api_search_button_clicked(self):
"""
Search when no keyword
"""
self.highlighted_api = None
self.highlighted_mashup = None
key = str(self.textbox.toPlainText())
if key != "":
apis = self.data_source.search_api(key)
self._show_apis(apis)
def mashup_search_button_clicked(self):
"""
Search when no keyword
"""
self.highlighted_api = None
self.highlighted_mashup = None
key = str(self.textbox.toPlainText())
if key != "":
mashups = self.data_source.search_mashup(key)
self._show_mashups(mashups)
def add_new_api(self):
"""
Add new api to the modules package.
"""
apis = self.data_source.apis()
model = self.table.selectionModel()
indexes = model.selectedIndexes()
for index in indexes:
api = apis[index.row()]
self._add_new_api(api)
return
def add_related_api(self):
objs = []
for mashup in self.related_mashups:
objs.append(mashup)
for api in mashup["related_mashups"]:
objs.append(api)
model = self.table.selectionModel()
indexes = model.selectedIndexes()
for index in indexes:
api = objs[index.row()]
if api.get("protocols"):
self._add_new_api(api)
return
def _show_related_mashups(self):
self.switch_btn_apis.hide()
self.switch_btn_mashups.hide()
self.recommendLabel.hide()
apis = []
objs = []
for mashup in self.related_mashups:
apis.extend(mashup["related_mashups"])
for api in apis:
objs.append(api)
mashups = self.data_source.mashups_by_api(api)
objs.extend(mashups)
row = 0
model = QStandardItemModel(len(objs), 4)
model.setColumnCount(4)
for obj in objs:
if obj.get('protocols'):
model.setData(model.index(row, 0), QVariant(get_api_name(obj)))
model.setData(model.index(row, 1), QVariant(obj['protocols']))
model.setData(model.index(row, 2), QVariant(obj['provider']))
else:
model.setData(model.index(row, 3), QVariant(get_mashup_name(obj)))
row += 1
model.setHeaderData(0, Qt.Horizontal, QVariant("API"))
model.setHeaderData(1, Qt.Horizontal, QVariant("Protocols"))
model.setHeaderData(2, Qt.Horizontal, QVariant("Provider"))
model.setHeaderData(3, Qt.Horizontal, QVariant("Mashup"))
self.table.setModel(model)
self.table.resizeColumnsToContents()
self.switch_btn_apis.show()
def _show_related_apis(self):
self.switch_btn_apis.hide()
self.switch_btn_mashups.hide()
self.recommendLabel.hide()
row = 0
objs = []
for mashup in self.related_mashups:
objs.append(mashup)
for api in mashup["related_mashups"]:
objs.append(api)
#Combining similarity and related.
similar_apis = self.data_source.search_api_similarity(self.highlighted_api)
#return str(mashup['id'])[(len("http://www.programmableweb.com/mashup/")):]
objs.append({'id': "http://www.programmableweb.com/mashup/Using-Similarity-Metric"})
objs.extend(similar_apis)
#Combining similarity and related.
model = QStandardItemModel(len(objs), 5)
for obj in objs:
if obj.get('protocols'):
model.setData(model.index(row, 1), QVariant(get_api_name(obj)))
model.setData(model.index(row, 2), QVariant(obj['protocols']))
model.setData(model.index(row, 3), QVariant(obj['provider']))
model.setData(model.index(row, 4), QVariant(obj['version']))
else:
model.setData(model.index(row, 0), QVariant(get_mashup_name(obj)))
row += 1
model.setHeaderData(0, Qt.Horizontal, QVariant("Mashup"))
model.setHeaderData(1, Qt.Horizontal, QVariant("API"))
model.setHeaderData(2, Qt.Horizontal, QVariant("Protocols"))
model.setHeaderData(3, Qt.Horizontal, QVariant("Provider"))
model.setHeaderData(4, Qt.Horizontal, QVariant("Version"))
self.table.setModel(model)
self.switch_btn_mashups.show()
def _add_new_api(self, api):
self.highlighted_api = api
mashups = self.data_source.mashups_by_api(api)
for mashup in mashups:
mashup['related_mashups'] = (self.data_source.apis_by_mashup(mashup))
if len(mashups) > 0:
self.related_mashups = mashups
self._show_related_apis()
manager = core.packagemanager.get_package_manager()
reg = core.modules.module_registry.get_module_registry()
package = manager.get_package("edu.cmu.sv.components", "1.0.0")
core.modules.module_registry.set_current_package(package)
if (api["protocols"] == "SOAP" and api["wsdl"] != ""):
s = Service(api["wsdl"])
add_new_service(s, api["wsdl"])
else:
new_module = vistrails_module.new_module(Module, get_api_name(api))
reg.add_module(new_module)
reg.add_input_port(new_module, "value1",
(core.modules.basic_modules.String, 'the first argument'))
reg.add_input_port(new_module, "value2",
(core.modules.basic_modules.String, 'the second argument'))
reg.add_output_port(new_module, "value",
(core.modules.basic_modules.String, 'the result'))
from PyQt4.QtSql import QSqlQueryModel, QSqlDatabase
from PyQt4.QtGui import QApplication, QListView, QTableView
import sys
app = QApplication(sys.argv)
db = QSqlDatabase.addDatabase("QMYSQL")
db.setHostName("localhost")
db.setDatabaseName("app")
db.setUserName("root")
db.setPassword("")
db.open()
projectModel = QSqlQueryModel()
projectModel.setQuery("select name from customers", db)
projectView = QTableView()
projectView.setModel(projectModel)
projectView.show()
model = projectView.model()
indexes = projectView.selectionModel().currentIndex().row()
app.exec_()
class Regression_PLOT_PyQt(QDialog):
"""
This calss return a PyQt GUI with a regression plot
"""
def __init__(self, Minerals=None, string=None, flag=None):
super(Regression_PLOT_PyQt, self).__init__()
if string is not None:
self.stringK, self.stringG, self.stringRho, self.user_input = string
else:
self.stringK = None
self.stringG = None
self.stringRho = None
self.user_input = False
self.resize(1400, 600)
self.Minerals = Minerals
self.dirty = False
#self.user_input = user_input
self.Minerals.original_flag()
self.Minerals.read_data()
self.table = QTableView()
self.model = QStandardItemModel(25, 7, self)
self.model.setHorizontalHeaderLabels([
'flag', 'Water Content', 'Iron Content', 'K (Gpa)', 'G (Gpa)',
'Rho (g/cm³)', 'Reference'
])
for i in range(len(self.Minerals.Flag)):
a = self.Minerals.Return_original_data(i)
for j in range(0, 7):
item = QStandardItem(str(a[j]))
self.model.setItem(i, j, item)
if j != 0:
item.setFlags(Qt.ItemIsEnabled)
item.setBackground(QColor(211, 211, 211))
if flag is not None:
self.Minerals.change_flag(flag)
for i in range(len(self.Minerals.Flag)):
item = QStandardItem(str(self.Minerals.Flag[i]))
self.model.setItem(i, 0, item)
self.table.setModel(self.model)
self.button = QPushButton('Update and use in thermoelastic model')
self.button.clicked.connect(self.Update)
self.button.setAutoDefault(False)
self.button1 = QPushButton('Add data file ')
self.button1.clicked.connect(self.Export)
self.button1.setAutoDefault(False)
self.layout = QGridLayout()
self.label = QLabel()
self.label.setText('''
Please input equation, Water: water content (wt%) and Fe: iron content (mol%)
for example -2.41*Water-30*Fe+81,K'=4.1,
''')
self.Kinput_formula = QLineEdit()
self.Ginput_formula = QLineEdit()
self.Rhoinput_formula = QLineEdit()
if self.stringK is not None:
self.Kinput_formula.setText(self.stringK)
self.Ginput_formula.setText(self.stringG)
self.Rhoinput_formula.setText(self.stringRho)
self.Userinput()
else:
self.Kinput_formula.setText(
self.Minerals.Show_fit_function(self.Minerals.function_K()[0],
self.Minerals.function_K()[1],
"K'",
error=False))
self.Ginput_formula.setText(
self.Minerals.Show_fit_function(self.Minerals.function_G()[0],
self.Minerals.function_G()[1],
"G'",
error=False))
self.Rhoinput_formula.setText(
self.Minerals.Show_fit_function(
self.Minerals.function_Rho()[0],
self.Minerals.function_Rho()[1],
'',
error=False))
self.Kinput_formula.returnPressed.connect(self.Kformula)
self.Ginput_formula.returnPressed.connect(self.Gformula)
self.Rhoinput_formula.returnPressed.connect(self.Rhoformula)
#self.connect(self.Kinput_formula,SIGNAL("returnPressed()"),self.Kformula)
#self.connect(self.Ginput_formula,SIGNAL("returnPressed()"),self.Gformula)
#self.connect(self.Rhoinput_formula,SIGNAL("returnPressed()"),self.Rhoformula)
self.user_change_confrim = QPushButton('Change to user input')
self.user_change_confrim.clicked.connect(self.Userinput)
self.user_change_confrim.setAutoDefault(False)
self.extension = QWidget()
self.extensionLayout = QGridLayout()
self.extensionLayout.setContentsMargins(0, 0, 0, 0)
labelK = QLabel()
labelK.setText('K<sub>0</sub>=')
labelG = QLabel()
labelG.setText('G<sub>0</sub>=')
labelRho = QLabel()
labelRho.setText('Rho<sub>0</sub>=')
self.extensionLayout.addWidget(labelK, 0, 0, 1, 3)
self.extensionLayout.addWidget(labelG, 1, 0, 1, 3)
self.extensionLayout.addWidget(labelRho, 2, 0, 1, 3)
self.extensionLayout.addWidget(self.Kinput_formula, 0, 1, 1, 3)
self.extensionLayout.addWidget(self.Ginput_formula, 1, 1, 1, 3)
self.extensionLayout.addWidget(self.Rhoinput_formula, 2, 1, 1, 3)
self.extensionLayout.addWidget(self.user_change_confrim, 3, 0, 9, 4)
self.extension.setLayout(self.extensionLayout)
self.check_change = QCheckBox("user input")
self.check_change.setChecked(False)
self.check_change.toggled.connect(self.extension.setVisible)
#self.PLOT(switch=True)
self.PLOT()
self.layout.addWidget(self.table, 0, 1, 1, 17)
self.layout.addWidget(self.button, 2, 0, 1, 9)
self.layout.addWidget(self.button1, 2, 9, 1, 9)
self.layout.addWidget(self.check_change, 3, 0, 1, 1)
self.layout.addWidget(self.label, 3, 3, 1, 5)
self.layout.addWidget(self.extension, 4, 0, 1, 9)
self.setLayout(self.layout)
self.extension.hide()
def Kformula(self):
self.stringK = str(self.Kinput_formula.text())
self.K0 = np.array(
re.findall(r"[+-]? *(?:\d+(?:\.\d*)?|\.\d+)(?:[eE][+-]?\d+)?",
self.stringK))
return self.K0, self.stringK
def Gformula(self):
self.stringG = str(self.Ginput_formula.text())
self.G0 = np.array(
re.findall(r"[+-]? *(?:\d+(?:\.\d*)?|\.\d+)(?:[eE][+-]?\d+)?",
self.stringG))
return self.G0, self.stringG
def Rhoformula(self):
self.stringRho = str(self.Rhoinput_formula.text())
self.Rho0 = np.array(
re.findall(r"[+-]? *(?:\d+(?:\.\d*)?|\.\d+)(?:[eE][+-]?\d+)?",
self.stringRho))
return self.Rho0, self.stringRho
def Userinput(self):
self.Minerals.User_Change(self.Kformula(), self.Gformula(),
self.Rhoformula())
self.user_input = True
self.PLOT()
def PLOT(self, switch=True):
if self.dirty == False:
#self.Minerals.Clean_content()
params = []
for i in range(20):
index = self.model.index(i, 0)
try:
aa = (float(self.model.itemData(index)[0]))
#print (aa)
params.append(aa)
except:
break
#print (params)
self.Minerals.Change_data_from_table(params)
#print ('wf')
else:
self.model.setHorizontalHeaderLabels([
'flag', 'Water Content', 'Iron Content', 'K (Gpa)', 'G (Gpa)',
'Rho (g/cm3)', 'Reference'
])
for i in range(len(self.Minerals.Flag)):
#a=self.Minerals.Return_original_data(i)
item = QStandardItem(str(self.Minerals.Flag[i]))
self.model.setItem(i, 0, item)
#==============================================================================
# for j in range(1,7):
# item = QStandardItem(str(a[j]))
# self.model.setItem(i, j,item)
# if j != 0:
# item.setFlags(Qt.ItemIsEnabled)
# item.setBackground(QColor(211,211,211))
#==============================================================================
#print (self.Minerals.Change)
if self.user_input == False:
self.a, self.b, self.c = self.Minerals.PLOT(return_fig=False)
#print (self.Minerals.number_of_data)
self.Kinput_formula.setText(
self.Minerals.Show_fit_function(self.Minerals.function_K()[0],
self.Minerals.function_K()[1],
"K'",
error=False))
self.Ginput_formula.setText(
self.Minerals.Show_fit_function(self.Minerals.function_G()[0],
self.Minerals.function_G()[1],
"G'",
error=False))
self.Rhoinput_formula.setText(
self.Minerals.Show_fit_function(
self.Minerals.function_Rho()[0],
self.Minerals.function_Rho()[1],
'',
error=False))
else:
self.a, self.b, self.c = self.Minerals.PLOT_input_formula(
return_fig=False)
self.canvas1 = FigureCanvas3D(self.a)
self.canvas2 = FigureCanvas3D(self.b)
self.canvas3 = FigureCanvas3D(self.c)
self.canvas1.mpl_connect('pick_event', self.onpick)
self.canvas2.mpl_connect('pick_event', self.onpick)
self.canvas3.mpl_connect('pick_event', self.onpick)
self.toolbar1 = NavigationToolbar(self.canvas1, self)
self.toolbar2 = NavigationToolbar(self.canvas2, self)
self.toolbar3 = NavigationToolbar(self.canvas3, self)
self.layout1_widget = QWidget()
self.layout1 = QGridLayout(self.layout1_widget)
self.layout1_widget.setFixedSize(600, 600)
self.layout1.addWidget(self.canvas1, 0, 1, 5, 5)
self.layout1.addWidget(self.toolbar1, 5, 1, 1, 5)
self.layout1.addWidget(self.canvas2, 6, 1, 5, 5)
self.layout1.addWidget(self.toolbar2, 11, 1, 1, 5)
self.layout1.addWidget(self.canvas3, 12, 1, 5, 5)
self.layout1.addWidget(self.toolbar3, 17, 1, 1, 5)
self.layout.addWidget(self.layout1_widget, 0, 0, 1, 1)
def onpick(self, event):
try:
for i in range(6):
self.model.item(self.ind,
i + 1).setBackground(QColor(211, 211, 211))
except:
pass
count = -1
for j in range(len((self.Minerals.Flag))):
if self.Minerals.Flag[j] == 1:
count += 1
if count == event.ind[0]:
self.ind = j
break
#print (self.ind)
#self.ind = event.ind
for i in range(6):
self.model.item(self.ind,
i + 1).setBackground(QColor(111, 111, 111))
def Update(self):
self.user_input = False
self.dirty = False
self.check_change.setChecked(False)
self.PLOT()
def Export(self):
self.dirty = True
dialog = TextEditor(name=self.Minerals.name)
if dialog.exec_():
pass
self.Minerals.read_data()
self.PLOT()
def ReturnString(self):
aa = self.Minerals.Flag
bb = str(int(aa[0]))
for i in range(1, len(aa)):
bb += str(int(aa[i]))
return [self.stringK, self.stringG, self.stringRho,
self.user_input], bb
class QueueDialog(Window):
def __init__(self, base):
Window.__init__(self, base, i18n.get('messages_queue'))
self.setFixedSize(500, 400)
self.last_timestamp = None
self.showed = False
self.list_ = QTableView()
self.list_.setSelectionBehavior(QAbstractItemView.SelectRows)
self.list_.clicked.connect(self.__account_clicked)
self.caption = QLabel()
self.caption.setWordWrap(True)
self.caption.setAlignment(Qt.AlignCenter)
self.estimated_time = QLabel()
self.estimated_time.setWordWrap(True)
self.estimated_time.setAlignment(Qt.AlignCenter)
self.delete_button = QPushButton(i18n.get('delete'))
self.delete_button.setEnabled(False)
self.delete_button.setToolTip(i18n.get('delete_selected_message'))
self.delete_button.clicked.connect(self.__delete_message)
self.clear_button = QPushButton(i18n.get('delete_all'))
self.clear_button.setEnabled(False)
self.clear_button.setToolTip(i18n.get('delete_all_messages_in_queue'))
self.clear_button.clicked.connect(self.__delete_all)
button_box = QHBoxLayout()
button_box.addStretch(1)
button_box.addWidget(self.clear_button)
button_box.addWidget(self.delete_button)
layout = QVBoxLayout()
layout.addWidget(self.list_, 1)
layout.addWidget(self.caption)
layout.addWidget(self.estimated_time)
layout.addLayout(button_box)
layout.setSpacing(5)
layout.setContentsMargins(5, 5, 5, 5)
self.setLayout(layout)
def __account_clicked(self, point):
self.delete_button.setEnabled(True)
self.clear_button.setEnabled(True)
def __delete_message(self):
self.__disable()
selection = self.list_.selectionModel()
index = selection.selectedIndexes()[0]
message = i18n.get('delete_message_from_queue_confirm')
confirmation = self.base.show_confirmation_message(i18n.get('confirm_delete'),
message)
if not confirmation:
self.__enable()
return
self.base.delete_message_from_queue(index.row())
def __delete_all(self):
self.__disable()
message = i18n.get('clear_message_queue_confirm')
confirmation = self.base.show_confirmation_message(i18n.get('confirm_delete'),
message)
if not confirmation:
self.__enable()
return
self.base.clear_queue()
def __enable(self):
self.list_.setEnabled(True)
self.delete_button.setEnabled(False)
if len(self.base.core.list_statuses_queue()) > 0:
self.clear_button.setEnabled(True)
else:
self.clear_button.setEnabled(False)
def __disable(self):
self.list_.setEnabled(False)
self.delete_button.setEnabled(False)
self.clear_button.setEnabled(False)
def __on_timeout(self):
now = int(time.time())
interval = self.base.core.get_queue_interval() * 60
if self.last_timestamp:
est_time = ((self.last_timestamp + interval) - now) / 60
else:
est_time = 0
humanized_est_time = self.base.humanize_time_intervals(est_time)
next_message = ' '.join([i18n.get('next_message_should_be_posted_in'), humanized_est_time])
if len(self.base.core.list_statuses_queue()) == 0:
self.estimated_time.setText('')
else:
self.estimated_time.setText(next_message)
def start(self):
self.timer = QTimer()
self.timer.timeout.connect(self.__on_timeout)
self.timer.start(60000)
def closeEvent(self, event=None):
if event:
event.ignore()
self.hide()
self.showed = False
def show(self):
if self.showed:
self.raise_()
return
self.update()
Window.show(self)
self.showed = True
def update(self):
model = QStandardItemModel()
model.setHorizontalHeaderItem(0, QStandardItem(i18n.get('account')))
model.setHorizontalHeaderItem(1, QStandardItem(i18n.get('message')))
self.list_.setModel(model)
now = int(time.time())
interval = self.base.core.get_queue_interval() * 60
if self.last_timestamp:
est_time = ((self.last_timestamp + interval) - now) / 60
else:
est_time = 0
row = 0
for status in self.base.core.list_statuses_queue():
username = get_username_from(status.account_id)
protocol_image = "%s.png" % get_protocol_from(status.account_id)
item = QStandardItem(QString.fromUtf8(username))
item.setIcon(QIcon(self.base.load_image(protocol_image, True)))
model.setItem(row, 0, item)
model.setItem(row, 1, QStandardItem(QString.fromUtf8(status.text)))
row += 1
humanized_interval = self.base.humanize_time_intervals(self.base.core.get_queue_interval())
humanized_est_time = self.base.humanize_time_intervals(est_time)
warning = i18n.get('messages_will_be_send') % humanized_interval
next_message = ' '.join([i18n.get('next_message_should_be_posted_in'), humanized_est_time])
self.caption.setText(warning)
if row == 0:
self.estimated_time.setText('')
else:
self.estimated_time.setText(next_message)
self.list_.horizontalHeader().setResizeMode(1, QHeaderView.Stretch)
self.list_.resizeColumnsToContents()
self.__enable()
def update_timestamp(self):
if len(self.base.core.list_statuses_queue()) > 0:
self.last_timestamp = int(time.time())
else:
self.last_timestamp = None
class VizRank(OWWidget):
name = "Rank projections (Scatter Plot)"
def __init__(self, parent_widget):
super().__init__()
self.parent_widget = parent_widget
self.want_control_area = False
self.running = False
self.progress = None
self.k = 10
self.projectionTable = QTableView()
self.mainArea.layout().addWidget(self.projectionTable)
self.projectionTable.setSelectionBehavior(QTableView.SelectRows)
self.projectionTable.setSelectionMode(QTableView.SingleSelection)
self.projectionTable.setSortingEnabled(True)
self.projectionTableModel = QStandardItemModel(self)
self.projectionTable.setModel(self.projectionTableModel)
self.projectionTable.selectionModel().selectionChanged.connect(
self.on_selection_changed)
self.button = gui.button(self.mainArea, self, "Start evaluation",
callback=self.toggle, default=True)
self.resize(380, 512)
self._initialize()
def _initialize(self):
self.running = False
self.projectionTableModel.clear()
self.projectionTableModel.setHorizontalHeaderLabels(
["Score", "Feature 1", "Feature 2"])
self.projectionTable.setColumnWidth(0, 60)
self.projectionTable.setColumnWidth(1, 120)
self.projectionTable.setColumnWidth(2, 120)
self.button.setText("Start evaluation")
self.button.setEnabled(False)
self.pause = False
self.data = None
self.attrs = []
self.scores = []
self.i, self.j = 0, 0
if self.progress:
self.progress.finish()
self.progress = None
self.information(0)
if self.parent_widget.data:
if not self.parent_widget.data.domain.class_var:
self.information(
0, "Data with a class variable is required.")
return
if len(self.parent_widget.data.domain.attributes) < 2:
self.information(
0, 'At least 2 unique features are needed.')
return
self.button.setEnabled(True)
def on_selection_changed(self, selected, deselected):
"""Called when the ranks view selection changes."""
a1 = selected.indexes()[1].data()
a2 = selected.indexes()[2].data()
self.parent_widget.update_attr(attributes=(a1, a2))
def toggle(self):
self.running ^= 1
if self.running:
self.button.setText("Pause")
self.run()
else:
self.button.setText("Continue")
self.button.setEnabled(False)
def run(self):
graph = self.parent_widget.graph
y_full = self.parent_widget.data.Y
norm = 1 / (len(y_full) * self.k)
if not self.attrs:
self.attrs = self.score_heuristic()
if not self.progress:
self.progress = gui.ProgressBar(
self, len(self.attrs) * (len(self.attrs) - 1) / 2)
for i in range(self.i, len(self.attrs)):
ind1 = graph.attribute_name_index[self.attrs[i]]
for j in range(self.j, i):
if not self.running:
self.i, self.j = i, j
if not self.projectionTable.selectedIndexes():
self.projectionTable.selectRow(0)
self.button.setEnabled(True)
return
ind2 = graph.attribute_name_index[self.attrs[j]]
X = graph.scaled_data[[ind1, ind2], :]
valid = graph.get_valid_list([ind1, ind2])
X = X[:, valid].T
y = y_full[valid]
knn = NearestNeighbors(n_neighbors=self.k).fit(X)
ind = knn.kneighbors(return_distance=False)
score = norm * np.sum(y[ind] == y.reshape(-1, 1))
pos = bisect_left(self.scores, score)
self.projectionTableModel.insertRow(
len(self.scores) - pos,
[QStandardItem("{:.4f}".format(score)),
QStandardItem(self.attrs[j]),
QStandardItem(self.attrs[i])])
self.scores.insert(pos, score)
self.progress.advance()
self.j = 0
self.progress.finish()
if not self.projectionTable.selectedIndexes():
self.projectionTable.selectRow(0)
self.button.setText("Finished")
self.button.setEnabled(False)
def score_heuristic(self):
X = self.parent_widget.graph.scaled_data.T
Y = self.parent_widget.data.Y
dom = Orange.data.Domain([ContinuousVariable(str(i))
for i in range(X.shape[1])],
self.parent_widget.data.domain.class_vars)
data = Orange.data.Table(dom, X, Y)
weights = ReliefF(n_iterations=100, k_nearest=self.k)(data)
attrs = sorted(
zip(weights,
(x.name for x in self.parent_widget.data.domain.attributes)
),
reverse=True)
return [a for _, a in attrs]
