|
|
|
|
|
|
|
|
""" |
|
|
Module implementing ChooseMTModelDialog. |
|
|
""" |
|
|
|
|
|
import sys |
|
|
|
|
|
from PyQt4.QtCore import ( |
|
|
pyqtSignature, |
|
|
QObject, |
|
|
SIGNAL, |
|
|
) |
|
|
from PyQt4.QtGui import QDialog |
|
|
from PyQt4.QtSql import QSqlQueryModel |
|
|
|
|
|
from Ui_chooseMTModel import Ui_Dialog |
|
|
from util import doAlert |
|
|
|
|
|
|
|
|
class ChooseMTModelDialog(QDialog, Ui_Dialog): |
|
|
""" |
|
|
Class documentation goes here. |
|
|
""" |
|
|
def __init__(self, parent=None, datamodel=None): |
|
|
""" |
|
|
Constructor |
|
|
""" |
|
|
QDialog.__init__(self, parent) |
|
|
self.setupUi(self) |
|
|
self.model = QSqlQueryModel() |
|
|
self.selTableView.setModel(self.model) |
|
|
self.database = datamodel.getQSqlDatabase() |
|
|
self.updateModel() |
|
|
self.selTableView.hideColumn(0) |
|
|
self.selTableView.hideColumn(5) |
|
|
self.selTableView.hideColumn(6) |
|
|
|
|
|
QObject.connect( |
|
|
datamodel, SIGNAL("modelInstalled()"), |
|
|
self.on_datamodel_modelInstalled) |
|
|
|
|
|
def updateModel(self): |
|
|
self.model.setQuery( |
|
|
'SELECT ID, name, srclang, trglang, status, path, mosesini ' |
|
|
'FROM models ' |
|
|
'WHERE status = "READY" AND deleted != "True"', |
|
|
self.database) |
|
|
|
|
|
def on_datamodel_recordUpdated(self, bRecord): |
|
|
"""Deal with the selection changed problem.""" |
|
|
try: |
|
|
if bRecord: |
|
|
current = self.selTableView.currentIndex() |
|
|
if current and current.row() != -1: |
|
|
self.curSelection = current.row() |
|
|
else: |
|
|
self.curSelection = None |
|
|
else: |
|
|
if self.curSelection is not None: |
|
|
self.selTableView.selectRow(self.curSelection) |
|
|
except Exception as e: |
|
|
print >> sys.stderr, str(e) |
|
|
|
|
|
def on_datamodel_modelInstalled(self): |
|
|
self.updateModel() |
|
|
|
|
|
@pyqtSignature("") |
|
|
def on_buttonBox_accepted(self): |
|
|
""" |
|
|
Slot documentation goes here. |
|
|
""" |
|
|
current = self.selTableView.currentIndex() |
|
|
if not current: |
|
|
doAlert("Please choose a model to start.") |
|
|
return |
|
|
record = self.model.record(current.row()) |
|
|
self.ID = record.value("ID").toString() |
|
|
self.modelName = record.value("name").toString() |
|
|
self.srcLang = record.value('srclang').toString() |
|
|
self.trgLang = record.value('trglang').toString() |
|
|
self.path = record.value("path").toString() |
|
|
self.mosesini = record.value("mosesini").toString() |
|
|
self.accept() |
|
|
|
