|
|
|
|
|
""" |
|
Module implementing Dialog. |
|
""" |
|
|
|
from PyQt4.QtGui import ( |
|
QDialog, |
|
QFileDialog, |
|
) |
|
from PyQt4.QtCore import pyqtSignature |
|
|
|
import datetime |
|
import os |
|
|
|
from Ui_addMTModel import Ui_Dialog |
|
from util import doAlert |
|
|
|
|
|
class AddMTModelDialog(QDialog, Ui_Dialog): |
|
""" |
|
Class documentation goes here. |
|
""" |
|
def __init__(self, parent=None, workdir=None): |
|
""" |
|
Constructor |
|
""" |
|
QDialog.__init__(self, parent) |
|
self.setupUi(self) |
|
self.timestr = None |
|
self.workdir = workdir |
|
todir, timestr = self.findEmptyDirWithTime(self.workdir) |
|
self.editPath.setText(todir) |
|
self.editName.setText("SampleModel_" + timestr) |
|
|
|
def findEmptyDirWithTime(self, workdir): |
|
if not self.timestr: |
|
self.timestr = datetime.datetime.now().strftime('%Y-%m-%d_%H%M%S') |
|
while True: |
|
todir = os.path.join(workdir, "Model_%s" % self.timestr) |
|
if not os.path.exists(todir): |
|
break |
|
self.timestr = datetime.datetime.now().strftime('%Y-%m-%d_%H%M%S') |
|
return todir, self.timestr |
|
|
|
@pyqtSignature("") |
|
def on_btnLocal_clicked(self): |
|
""" |
|
Slot documentation goes here. |
|
""" |
|
dialog = QFileDialog(self) |
|
dialog.setFileMode(QFileDialog.ExistingFile) |
|
dialog.setNameFilter("Zipped model files (*.zip)") |
|
dialog.setViewMode(QFileDialog.Detail) |
|
if dialog.exec_(): |
|
self.editLocal.setText(dialog.selectedFiles()[0]) |
|
|
|
@pyqtSignature("") |
|
def on_btnPath_clicked(self): |
|
""" |
|
Slot documentation goes here. |
|
""" |
|
dialog = QFileDialog(self, directory=self.workdir) |
|
dialog.setFileMode(QFileDialog.Directory) |
|
dialog.setViewMode(QFileDialog.Detail) |
|
dialog.setOptions(QFileDialog.ShowDirsOnly) |
|
if dialog.exec_(): |
|
root = str(dialog.selectedFiles()[0]) |
|
todir, _ = self.findEmptyDirWithTime(root) |
|
self.editPath.setText(todir) |
|
|
|
@pyqtSignature("bool") |
|
def on_grpBoxInternet_toggled(self, p0): |
|
""" |
|
Slot documentation goes here. |
|
""" |
|
self.grpBoxLocal.setChecked(not p0) |
|
|
|
@pyqtSignature("bool") |
|
def on_grpBoxLocal_toggled(self, p0): |
|
""" |
|
Slot documentation goes here. |
|
""" |
|
self.grpBoxInternet.setChecked(not p0) |
|
|
|
@pyqtSignature("") |
|
def on_buttonBox_accepted(self): |
|
""" |
|
Slot documentation goes here. |
|
""" |
|
def checkEmpty(mystr): |
|
return len(str(mystr).strip()) <= 0 |
|
|
|
|
|
self.modelName = self.editName.text() |
|
if checkEmpty(self.modelName): |
|
doAlert("Please provide non-empty Model Name") |
|
return |
|
if self.grpBoxInternet.isChecked(): |
|
self.source = self.editInternet.text() |
|
self.sourceMode = "Internet" |
|
elif self.grpBoxLocal.isChecked(): |
|
self.source = self.editLocal.text() |
|
self.sourceMode = "Local" |
|
if not os.path.exists(str(self.source)): |
|
doAlert("Please provide valid local file as source") |
|
return |
|
else: |
|
doAlert("Please provide Install Source") |
|
return |
|
if checkEmpty(self.source): |
|
doAlert("Please provide non-empty Install Source") |
|
return |
|
self.dest = self.editPath.text() |
|
if checkEmpty(self.dest): |
|
doAlert("Please provide non-empty Install Destination Folder") |
|
return |
|
self.accept() |
|
|
