def main():
import sys
from PyQt4 import QtGui
from library.ProgressBar import CProgressBar
app = QtGui.QApplication(sys.argv)
p = CProgressBar()
p.setRange(0, 0)
p.show()
app.processEvents()
pi = CProgressInformer(description=u'test')
pi.progressChanged.connect(p.setProgress)
pi.total = 1000
for i in xrange(pi.total):
for j in xrange(100000):
x = 8 * 9
pi.current += 1
sys.exit(app.exec_())
class CImportDialog(QtGui.QDialog):
InitState = 0
ImportState = 1
def __init__(self, db, parent = None):
super(CImportDialog, self).__init__(parent = parent)
self._db = db
self._pi = CProgressInformer(processEventsFlag=QtCore.QEventLoop.AllEvents)
self._engine = CImportR85ERZCentralEngine(db, progressInformer=self._pi)
self._engine.setUserConfirmationFunction(self.userConfirmation)
self._loggerHandler = CLogHandler(self._engine.logger(), self)
self._loggerHandler.setLevel(logging.INFO)
self._currentState = self.InitState
self.setupUi()
self.onStateChanged()
def setParams(self, params):
if isinstance(params, dict):
reqFileName = forceStringEx(getPref(params, 'reqFile', u''))
if os.path.isfile(reqFileName ):
self.edtReqFileName.setText(reqFileName )
respFileName = forceStringEx(getPref(params, 'respFile', u''))
if os.path.isfile(respFileName):
self.edtRespFileName.setText(respFileName)
def engine(self):
return self._engine
def params(self):
params = {}
setPref(params, 'reqFile', forceStringEx(self.edtReqFileName.text()))
setPref(params, 'respFile', forceStringEx(self.edtRespFileName.text()))
return params
@property
def currentState(self):
return self._currentState
@currentState.setter
def currentState(self, value):
if value in [self.InitState, self.ImportState]:
self._currentState = value
self.onStateChanged()
def onStateChanged(self):
self.btnNextAction.setText(self._actionNames.get(self.currentState, u'<Error>'))
self.btnClose.setEnabled(self.currentState != self.ImportState)
self.gbInit.setEnabled(self.currentState == self.InitState)
self.gbImport.setEnabled(self.currentState == self.ImportState)
# self.btnSave.setEnabled(bool(self._engine.documents))
QtCore.QCoreApplication.processEvents()
# noinspection PyAttributeOutsideInit,PyUnresolvedReferences
def setupUi(self):
layout = QtGui.QVBoxLayout()
# init block
self.gbInit = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
lineLayout = QtGui.QHBoxLayout()
self.edtReqFileName = QtGui.QLineEdit()
self.edtReqFileName.setReadOnly(True)
lineLayout.addWidget(self.edtReqFileName)
self.btnBrowseReqFile = QtGui.QToolButton()
self.btnBrowseReqFile.clicked.connect(self.onBrowseReqFileClicked)
lineLayout.addWidget(self.btnBrowseReqFile)
gbLayout.addLayout(lineLayout)
lineLayout = QtGui.QHBoxLayout()
self.edtRespFileName = QtGui.QLineEdit()
self.edtRespFileName.setReadOnly(True)
lineLayout.addWidget(self.edtRespFileName)
self.btnBrowseRespFile = QtGui.QToolButton()
self.btnBrowseRespFile.clicked.connect(self.onBrowseRespFileClicked)
lineLayout.addWidget(self.btnBrowseRespFile)
gbLayout.addLayout(lineLayout)
self.gbInit.setLayout(gbLayout)
layout.addWidget(self.gbInit)
# import block
self.gbImport = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
self.progressBar = CProgressBar()
self.progressBar.setAlignment(QtCore.Qt.AlignCenter)
self.progressBar.setRange(0, 0)
self.progressBar.setProgressFormat(u'(%v/%m)')
self._pi.progressChanged.connect(self.progressBar.setProgress)
gbLayout.addWidget(self.progressBar)
self.gbImport.setLayout(gbLayout)
layout.addWidget(self.gbImport)
# log block
self.logInfo = QtGui.QTextEdit()
layout.addWidget(self.logInfo)
self._loggerHandler.logged.connect(self.logInfo.append)
# buttons block
subLayout = QtGui.QHBoxLayout()
self.btnNextAction = QtGui.QPushButton()
# noinspection PyUnresolvedReferences
self.btnNextAction.clicked.connect(self.onNextActionClicked)
subLayout.addStretch()
subLayout.addWidget(self.btnNextAction)
self.btnClose = QtGui.QPushButton()
self.btnClose.clicked.connect(self.onCloseClicked)
subLayout.addWidget(self.btnClose)
layout.addLayout(subLayout)
self.setLayout(layout)
self.setMinimumWidth(512)
self.retranslateUi()
# noinspection PyAttributeOutsideInit
def retranslateUi(self):
context = unicode(getClassName(self))
self.setWindowTitle(forceTr(u'Импорт. Крым. ЦС ЕРЗ.', context))
self.gbInit.setTitle(forceTr(u'Параметры импорта', context))
self.gbImport.setTitle(forceTr(u'Импорт', context))
self._actionNames = {self.InitState: forceTr(u'Импорт',context),
self.ImportState: forceTr(u'Прервать', context)}
self.btnClose.setText(forceTr(u'Закрыть', context))
def userConfirmation(self, title, message):
# noinspection PyTypeChecker,PyCallByClass
return QtGui.QMessageBox.Yes == QtGui.QMessageBox.question(self,
title,
message,
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
QtGui.QMessageBox.No)
@QtCore.pyqtSlot()
def onNextActionClicked(self):
if self.currentState == self.InitState:
self.currentState = self.ImportState
try:
self._engine.process(forceStringEx(self.edtReqFileName.text()),
forceStringEx(self.edtRespFileName.text()))
except Exception, e:
self.currentState = self.InitState
raise
self.currentState = self.InitState
elif self.currentState == self.ImportState:
self._engine.abort()
self.currentState = self.InitState
class CExportDialog(QtGui.QDialog):
InitState = 0
ExportState = 1
SaveState = 2
def __init__(self, db, parent=None):
QtGui.QDialog.__init__(self, parent)
self.db = db
self.pi = CProgressInformer(
processEventsFlag=QtCore.QEventLoop.AllEvents)
self.currentState = self.InitState
self.setupUi()
self.engine = CExportR23AttachClientMIACEngine(
self.db,
self.cmbOrgStructure.value(),
self.edtDate.date(),
progressInformer=self.pi)
self.loggerHandler = CLogHandler(self.engine.getLogger(), self)
self.btnMiac.setChecked(True)
self.onStateChange()
def onStateChange(self):
self.btnNextAction.setText(
self.actionNames.get(self.currentState, u'<Error>'))
self.btnClose.setEnabled(self.currentState != self.ExportState)
self.grInit.setEnabled(self.currentState == self.InitState)
self.grExport.setEnabled(self.currentState == self.ExportState)
self.grSave.setEnabled(self.currentState == self.SaveState)
self.btnSave.setEnabled(bool(self.engine.documents))
QtCore.QCoreApplication.processEvents()
def setupUi(self):
layout = QtGui.QVBoxLayout()
self.grInit = QtGui.QGroupBox()
grLayout = QtGui.QVBoxLayout()
orgStructureLayout = QtGui.QHBoxLayout()
self.lblOrgStructure = QtGui.QLabel()
orgStructureLayout.addWidget(self.lblOrgStructure)
self.cmbOrgStructure = COrgStructureComboBox(None)
orgStructureLayout.addWidget(self.cmbOrgStructure)
grLayout.addLayout(orgStructureLayout)
dateLayout = QtGui.QHBoxLayout()
self.lblDate = QtGui.QLabel()
dateLayout.addWidget(self.lblDate)
self.edtDate = CDateEdit()
dateLayout.addWidget(self.edtDate)
grLayout.addLayout(dateLayout)
modeExportLayout = QtGui.QHBoxLayout()
self.btnMiac = QtGui.QRadioButton()
modeExportLayout.addWidget(self.btnMiac)
self.btnSMO = QtGui.QRadioButton()
modeExportLayout.addWidget(self.btnSMO)
grLayout.addLayout(modeExportLayout)
self.grInit.setLayout(grLayout)
layout.addWidget(self.grInit)
self.grExport = QtGui.QGroupBox()
grLayout = QtGui.QVBoxLayout()
self.progressBar = CProgressBar()
self.progressBar.setAlignment(QtCore.Qt.AlignCenter)
self.progressBar.setRange(0, 0)
self.progressBar.setProgressFormat(u'(%v/%m)')
self.pi.progressChanged.connect(self.progressBar.setProgress)
grLayout.addWidget(self.progressBar)
self.grExport.setLayout(grLayout)
layout.addWidget(self.grExport)
self.grSave = QtGui.QGroupBox()
grLayout = QtGui.QVBoxLayout()
lineLayout = QtGui.QHBoxLayout()
self.edtSaveDir = QtGui.QLineEdit(QtCore.QDir.homePath())
self.edtSaveDir.setReadOnly(True)
lineLayout.addWidget(self.edtSaveDir)
self.btnBrowseDir = QtGui.QToolButton()
self.btnBrowseDir.clicked.connect(self.onBrowseDir)
lineLayout.addWidget(self.btnBrowseDir)
grLayout.addLayout(lineLayout)
lineLayout = QtGui.QHBoxLayout()
self.btnSave = QtGui.QPushButton()
self.btnSave.clicked.connect(self.onSaveClicked)
lineLayout.addWidget(self.btnSave)
grLayout.addLayout(lineLayout)
self.grSave.setLayout(grLayout)
layout.addWidget(self.grSave)
self.logInfo = QtGui.QTextEdit()
layout.addWidget(self.logInfo)
# self.loggerHandler.logged.connect(self.logInfo.append)
subLayout = QtGui.QHBoxLayout()
self.btnNextAction = QtGui.QPushButton()
self.btnNextAction.clicked.connect(self.onNextActionClicked)
subLayout.addStretch()
subLayout.addWidget(self.btnNextAction)
self.btnClose = QtGui.QPushButton()
self.btnClose.clicked.connect(self.onCloseClicked)
subLayout.addWidget(self.btnClose)
layout.addLayout(subLayout)
self.setLayout(layout)
self.setMinimumWidth(512)
self.retranselateUi()
def retranselateUi(self):
context = unicode(getClassName(self))
self.setWindowTitle(
forceTr(u'Экспорт прикрипленного населения. Краснодарский край',
context))
self.grInit.setTitle(forceTr(u'Параметры экспорта', context))
self.lblOrgStructure.setText(forceTr(u'Подразделение', context))
self.btnMiac.setText(forceTr(u'Экспорт МИАЦ', context))
self.btnSMO.setText(forceTr(u'Экспорт СМО', context))
self.lblDate.setText(forceTr(u'Отчетная дата', context))
self.grExport.setTitle(forceTr(u'Экспотр', context))
self.grSave.setTitle(forceTr(u'Сохранение результата', context))
self.btnSave.setText(forceTr(u'Сохранить', context))
self.actionNames = {
self.InitState: forceTr(u'Экспорт', context),
self.ExportState: forceTr(u'Прервать', context),
self.SaveState: forceTr(u'Повторить', context)
}
self.btnClose.setText(forceTr(u'Закрыть', context))
@QtCore.pyqtSlot()
def onNextActionClicked(self):
if self.currentState == self.InitState:
self.currentState = self.ExportState
try:
result = self.engine.process()
except:
self.currentState = self.InitState
raise
self.currentState = self.SaveState if result else self.InitState
elif self.currentState == self.ExportState:
self.engine.terminate()
elif self.currentState == self.SaveState:
self.currentState = self.InitState
self.onStateChange()
def canClose(self):
return not ((self.engine.documents) or QtGui.QMessageBox.warning(
self, u'Внимание!', u'Остались несохраненные файлы выгрузок\n'
u'которые будут утеряны.\n'
u'Вы уверены, что хотите покинуть менеджер экспорта?\n',
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No, QtGui.QMessageBox.No)
== QtGui.QMessageBox.Yes)
def closeEvent(self, event):
if self.canClose():
event.accept()
else:
event.ignore()
def done(self, result):
if self.canClose():
super(CExportDialog, self).done(result)
@QtCore.pyqtSlot()
def onCloseClicked(self):
self.accept()
@QtCore.pyqtSlot()
def onBrowseDir(self):
saveDir = forceStringEx(
QtGui.QFileDialog.getExistingDirectory(
self, u'Укажите директорию для сохранения файлов выгрузки',
self.edtSaveDir.text()))
if os.path.isdir(saveDir):
self.edtSaveDir.setText(saveDir)
@QtCore.pyqtSlot()
def onSaveClicked(self):
self.btnClose.setEnabled(False)
self.engine.save(self.edtSaveDir.text())
self.onStateChange()
@QtCore.pyqtSlot(bool)
def on_btnMiac_toggled(self, checked):
if not checked:
self.engine = CExportR23AttachClientSMOEngine(
self.db,
self.cmbOrgStructure.value(),
self.edtDate.date(),
progressInformer=self.pi)
self.loggerHandler = CLogHandler(self.engine.logger(), self)
else:
self.engine = CExportR23AttachClientMIACEngine(
self.db,
self.cmbOrgStructure.value(),
self.edtDate.date(),
progressInformer=self.pi)
self.loggerHandler = CLogHandler(self.engine.logger(), self)
class CImportPayRefuseR85Dialog(QtGui.QDialog):
InitState = 0
ImportState = 1
MainEngine = CImportPayRefuseR85Engine
# DDEngine = CImportR85DDTFEngine
# TODO: Возможно, стоит отсюда выпилить эту разбивку, если принципиальных различий не появится
engineByFormat = {'HS': MainEngine,
'HT': MainEngine,
'DP': MainEngine,
'DV': MainEngine,
'DO': MainEngine,
'DS': MainEngine,
'DU': MainEngine,
'DF': MainEngine,
'DD': MainEngine,
'DR': MainEngine,
'TS': MainEngine,
'TT': MainEngine,
}
def __init__(self, db, orgId, parent = None, isFLC = False):
super(CImportPayRefuseR85Dialog, self).__init__(parent)
if isFLC:
self.MainEngine = CImportPayRefuseR85EngineFLC
self._db = db
self._orgId = orgId
self._pi = CProgressInformer(processEventsFlag=QtCore.QEventLoop.AllEvents)
# self._engine = CImportR85TFEngine(db, orgId, progressInformer=self._pi) #TODO?
# self._engine.setUserConfirmationFunction(self.userConfirmation)
self._lastExceptionHandler = None
# self._loggerHandler = CLogHandler(self._engine.logger(), self)
# self._loggerHandler.setLevel(logging.INFO)
self._currentState = self.InitState
self.setupUi()
self.onStateChanged()
self._engines = {}
self._isFLC = isFLC
def setLastExceptionHandler(self, handler):
self._lastExceptionHandler = handler
def setParams(self, params):
if isinstance(params, dict):
fileName = forceStringEx(getPref(params, 'inputFile', u''))
if os.path.isfile(fileName):
self.edtFileName.setText(fileName)
def engine(self, format):
engine = self._engines.get(format, None) # self._engine
if engine is None:
engine = self.engineByFormat.get(format, self.MainEngine)(self._db, self._orgId, progressInformer=self._pi, format=format)
engine.setLastExceptionHandler(self._lastExceptionHandler)
engine.setUserConfirmationFunction(self.userConfirmation)
loggerHandler = CLogHandler(engine.logger(), self)
loggerHandler.setLevel(logging.INFO)
loggerHandler.logged.connect(self.logInfo.append)
self._engines[format] = engine
return engine
def params(self):
params = {}
setPref(params, 'inputFile', forceStringEx(self.edtFileName.text()))
return params
@property
def currentState(self):
return self._currentState
@currentState.setter
def currentState(self, value):
if value in [self.InitState, self.ImportState]:
self._currentState = value
self.onStateChanged()
def onStateChanged(self):
self.btnNextAction.setText(self._actionNames.get(self.currentState, u'<Error>'))
self.btnClose.setEnabled(self.currentState != self.ImportState)
self.gbInit.setEnabled(self.currentState == self.InitState)
self.gbImport.setEnabled(self.currentState == self.ImportState)
# self.btnSave.setEnabled(bool(self._engine.documents))
QtCore.QCoreApplication.processEvents()
# noinspection PyAttributeOutsideInit
def setupUi(self):
layout = QtGui.QVBoxLayout()
# init block
self.gbInit = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
lineLayout = QtGui.QHBoxLayout()
self.edtFileName = QtGui.QLineEdit()
self.edtFileName.setReadOnly(True)
self.edtFileName.textChanged.connect(self.onFileNameChanged)
lineLayout.addWidget(self.edtFileName)
self.btnBrowseFile = QtGui.QToolButton()
# noinspection PyUnresolvedReferences
self.btnBrowseFile.clicked.connect(self.onBrowseFileClicked)
lineLayout.addWidget(self.btnBrowseFile)
gbLayout.addLayout(lineLayout)
self.gbInit.setLayout(gbLayout)
layout.addWidget(self.gbInit)
# import block
self.gbImport = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
self.progressBar = CProgressBar()
self.progressBar.setAlignment(QtCore.Qt.AlignCenter)
self.progressBar.setRange(0, 0)
self.progressBar.setProgressFormat(u'(%v/%m)')
self._pi.progressChanged.connect(self.progressBar.setProgress)
gbLayout.addWidget(self.progressBar)
self.gbImport.setLayout(gbLayout)
layout.addWidget(self.gbImport)
# log block
self.logInfo = QtGui.QTextEdit()
layout.addWidget(self.logInfo)
# self._loggerHandler.logged.connect(self.logInfo.append)
# buttons block
subLayout = QtGui.QHBoxLayout()
self.btnNextAction = QtGui.QPushButton()
# noinspection PyUnresolvedReferences
self.btnNextAction.clicked.connect(self.onNextActionClicked)
subLayout.addStretch()
subLayout.addWidget(self.btnNextAction)
self.btnClose = QtGui.QPushButton()
# noinspection PyUnresolvedReferences
self.btnClose.clicked.connect(self.onCloseClicked)
subLayout.addWidget(self.btnClose)
layout.addLayout(subLayout)
self.setLayout(layout)
self.setMinimumWidth(512)
self.retranslateUi()
# noinspection PyAttributeOutsideInit
def retranslateUi(self):
context = unicode(getClassName(self))
self.setWindowTitle(forceTr(u'Импорт отказов. Крым. ТФОМС.', context))
self.gbInit.setTitle(forceTr(u'Параметры импорта', context))
self.gbImport.setTitle(forceTr(u'Импорт', context))
self._actionNames = {self.InitState: forceTr(u'Импорт',context),
self.ImportState: forceTr(u'Прервать', context)}
self.btnClose.setText(forceTr(u'Закрыть', context))
def userConfirmation(self, title, message):
return QtGui.QMessageBox.Yes == QtGui.QMessageBox.question(self,
title,
message,
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
QtGui.QMessageBox.No)
@QtCore.pyqtSlot()
def onNextActionClicked(self):
if self.currentState == self.InitState:
self.currentState = self.ImportState
try:
self.engine(self._currentFormat).process(forceStringEx(self.edtFileName.text()))
except Exception, e:
self.currentState = self.InitState
raise
self.currentState = self.InitState
elif self.currentState == self.ImportState:
self.engine(self._currentFormat).abort()
self.currentState = self.InitState
class CMultipleExportDialog(QtGui.QDialog):
InitState = 0
ExportState = 1
SaveState = 2
def __init__(self, db, accountIdList, parent=None):
super(CMultipleExportDialog, self).__init__(parent=parent)
self._db = db
self._accountIdList = accountIdList
self._engines = []
self._pi = CProgressInformer(
processEventsFlag=QtCore.QEventLoop.AllEvents)
# self._engine = engine(db, accountRecord, progressInformer=self._pi)
# self._loggerHandler = CLogHandler(self._engine.logger(), self)
self._currentState = self.InitState
self.setupUi()
self.onStateChanged()
def setParams(self, params):
if isinstance(params, dict):
outDir = forceStringEx(getPref(params, 'outDir', u''))
if os.path.isdir(outDir):
self.edtSaveDir.setText(outDir)
def params(self):
params = {}
setPref(params, 'outDir', forceStringEx(self.edtSaveDir.text()))
return params
@property
def currentState(self):
return self._currentState
@currentState.setter
def currentState(self, value):
if value in [self.InitState, self.ExportState, self.SaveState]:
self._currentState = value
self.onStateChanged()
def onStateChanged(self):
self.btnNextAction.setText(
self._actionNames.get(self.currentState, u'<Error>'))
self.btnClose.setEnabled(self.currentState != self.ExportState)
# self.gbInit.setEnabled(self.currentState == self.InitState)
self.gbExport.setEnabled(self.currentState == self.ExportState)
self.gbSave.setEnabled(self.currentState == self.SaveState)
self.btnSave.setEnabled(self.hasDocuments())
QtCore.QCoreApplication.processEvents()
def setupUi(self):
layout = QtGui.QVBoxLayout()
# init block
# self.gbInit = QtGui.QGroupBox()
# gbLayout = QtGui.QVBoxLayout()
# self.gbInit.setLayout(gbLayout)
# layout.addWidget(self.gbInit)
# export block
self.gbExport = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
self.progressBar = CProgressBar()
self.progressBar.setAlignment(QtCore.Qt.AlignCenter)
self.progressBar.setRange(0, 0)
self.progressBar.setProgressFormat(u'(%v/%m)')
self._pi.progressChanged.connect(self.progressBar.setProgress)
gbLayout.addWidget(self.progressBar)
self.gbExport.setLayout(gbLayout)
layout.addWidget(self.gbExport)
# save block
self.gbSave = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
lineLayout = QtGui.QHBoxLayout()
self.edtSaveDir = QtGui.QLineEdit(QtCore.QDir.homePath())
self.edtSaveDir.setReadOnly(True)
lineLayout.addWidget(self.edtSaveDir)
self.btnBrowseDir = QtGui.QToolButton()
self.btnBrowseDir.clicked.connect(self.onBrowseDir)
lineLayout.addWidget(self.btnBrowseDir)
gbLayout.addLayout(lineLayout)
lineLayout = QtGui.QHBoxLayout()
self.btnSave = QtGui.QPushButton()
self.btnSave.clicked.connect(self.onSaveClicked)
lineLayout.addWidget(self.btnSave)
gbLayout.addLayout(lineLayout)
self.gbSave.setLayout(gbLayout)
layout.addWidget(self.gbSave)
# log block
self.logInfo = QtGui.QTextEdit()
layout.addWidget(self.logInfo)
# self._loggerHandler.logged.connect(self.logInfo.append)
# buttons block
subLayout = QtGui.QHBoxLayout()
self.btnNextAction = QtGui.QPushButton()
self.btnNextAction.clicked.connect(self.onNextActionClicked)
subLayout.addStretch()
subLayout.addWidget(self.btnNextAction)
self.btnClose = QtGui.QPushButton()
self.btnClose.clicked.connect(self.onCloseClicked)
subLayout.addWidget(self.btnClose)
layout.addLayout(subLayout)
self.setLayout(layout)
self.setMinimumWidth(512)
self.retranslateUi()
self.setModal(True)
def retranslateUi(self):
context = unicode(getClassName(self))
self.setWindowTitle(forceTr(u'Экспорт. Крым. ФЛК', context))
# self.gbInit.setTitle(forceTr(u'Параметры экспорта', context))
self.gbExport.setTitle(forceTr(u'Экспорт', context))
self.gbSave.setTitle(forceTr(u'Сохранение результата', context))
self.btnSave.setText(forceTr(u'Сохранить', context))
self._actionNames = {
self.InitState: forceTr(u'Экспорт', context),
self.ExportState: forceTr(u'Прервать', context),
self.SaveState: forceTr(u'Повторить', context)
}
self.btnClose.setText(forceTr(u'Закрыть', context))
@QtCore.pyqtSlot()
def onNextActionClicked(self):
if self.currentState == self.InitState:
result = False
self.currentState = self.ExportState
for accountId in self._accountIdList:
engine = None
exportInfo = getAccountExportFormat(accountId).split()
if exportInfo:
exportFormat = exportInfo[0]
if exportFormat in ('R85XML'):
engine = CExportR85TFEngine(self._db,
progressInformer=self._pi,
FLC=True)
# Крым, выгрузка диспансеризации и профосмотров (МО->Фонд)
elif exportFormat == 'R85DDXML':
engine = CExportR85DDTFEngine(
self._db, progressInformer=self._pi, FLC=True)
elif exportFormat == 'R85HTXML':
engine = CExportR85HTTFEngine(
self._db, progressInformer=self._pi, FLC=True)
if engine:
# FIXME: может провоцировать падения/ошибки?
CLogHandler(engine.logger(),
self).logged.connect(self.logInfo.append)
self._engines.append(engine)
try:
result = engine.process(accountId)
except:
self.currentState = self.InitState
raise
self.currentState = self.SaveState if result else self.InitState
elif self.currentState == self.ExportState:
for engine in self._engines:
engine.terminate()
self._engines = []
elif self.currentState == self.SaveState:
self.currentState = self.InitState
self.onStateChanged()
def canClose(self):
ok = self.currentState == self.SaveState
return ok or \
QtGui.QMessageBox.warning(self,
u'Внимание!',
u'Остались несохраненные файлы выгрузок,\n'
u'которые будут утеряны.\n'
u'Вы уверены, что хотите покинуть менеджер экспорта?\n',
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
QtGui.QMessageBox.No) == QtGui.QMessageBox.Yes
def hasDocuments(self):
result = False
for engine in self._engines:
if engine.hDocuments or engine.lDocuments:
result = True
break
return result
def closeEvent(self, event):
if self.canClose():
event.accept()
else:
event.ignore()
def done(self, result):
if self.canClose():
super(CMultipleExportDialog, self).done(result)
@QtCore.pyqtSlot()
def onCloseClicked(self):
self.accept()
@QtCore.pyqtSlot()
def onBrowseDir(self):
saveDir = forceStringEx(
QtGui.QFileDialog.getExistingDirectory(
self, u'Укажите директорию для сохранения файлов выгрузки',
self.edtSaveDir.text()))
if os.path.isdir(saveDir):
self.edtSaveDir.setText(saveDir)
@QtCore.pyqtSlot()
def onSaveClicked(self):
self.btnClose.setEnabled(False)
for engine in self._engines:
engine.save(self.edtSaveDir.text())
self.onStateChanged()
class CExport85MTR_Refused(QtGui.QDialog):
InitState = 0
ExportState = 1
SaveState = 2
def __init__(self, db, accountId, parent=None):
super(CExport85MTR_Refused, self).__init__(parent)
self._db = db
self._pi = CProgressInformer(
processEventsFlag=QtCore.QEventLoop.AllEvents)
self._accountId = accountId
self._engine = CExportR85MTREngine_Refused(db,
progressInformer=self._pi)
self._loggerHandler = CLogHandler(self._engine.logger(), self)
self._loggerHandler.setLevel(logging.INFO)
self._currentState = self.InitState
self.setupUi()
self.onStateChanged()
def engine(self):
return self._engine
def setParams(self, params):
if isinstance(params, dict):
outDir = forceStringEx(getPref(params, 'outDir', u''))
if os.path.isdir(outDir):
self.edtSaveDir.setText(outDir)
accNumber = forceInt(getPref(params, 'accNumber', 0)) + 1
self.spbAccountNumber.setValue(accNumber)
def params(self):
params = {}
setPref(params, 'outDir', forceStringEx(self.edtSaveDir.text()))
setPref(params, 'accNumber', self.spbAccountNumber.value())
return params
@property
def currentState(self):
return self._currentState
@currentState.setter
def currentState(self, value):
if value in [self.InitState, self.ExportState, self.SaveState]:
self._currentState = value
self.onStateChanged()
def onStateChanged(self):
self.btnNextAction.setText(
self._actionNames.get(self.currentState, u'<Error>'))
self.btnClose.setEnabled(self.currentState != self.ExportState)
self.gbInit.setEnabled(self.currentState == self.InitState)
self.gbExport.setEnabled(self.currentState == self.ExportState)
self.gbSave.setEnabled(self.currentState == self.SaveState)
self.btnSave.setEnabled(bool(self._engine.documents))
QtCore.QCoreApplication.processEvents()
# noinspection PyAttributeOutsideInit
def setupUi(self):
layout = QtGui.QVBoxLayout()
# init block
self.gbInit = QtGui.QGroupBox()
gbLayout = QtGui.QHBoxLayout()
self.lblAccountNumber = QtGui.QLabel()
gbLayout.addWidget(self.lblAccountNumber)
self.spbAccountNumber = QtGui.QSpinBox()
self.spbAccountNumber.setRange(1, 9999)
gbLayout.addWidget(self.spbAccountNumber)
self.gbInit.setLayout(gbLayout)
layout.addWidget(self.gbInit)
# export block
self.gbExport = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
self.progressBar = CProgressBar()
self.progressBar.setAlignment(QtCore.Qt.AlignCenter)
self.progressBar.setRange(0, 0)
self.progressBar.setProgressFormat(u'(%v/%m)')
self._pi.progressChanged.connect(self.progressBar.setProgress)
gbLayout.addWidget(self.progressBar)
self.gbExport.setLayout(gbLayout)
layout.addWidget(self.gbExport)
# save block
self.gbSave = QtGui.QGroupBox()
gbLayout = QtGui.QVBoxLayout()
lineLayout = QtGui.QHBoxLayout()
self.edtSaveDir = QtGui.QLineEdit(QtCore.QDir.homePath())
self.edtSaveDir.setReadOnly(True)
lineLayout.addWidget(self.edtSaveDir)
self.btnBrowseDir = QtGui.QToolButton()
self.btnBrowseDir.clicked.connect(self.onBrowseDirClicked)
lineLayout.addWidget(self.btnBrowseDir)
gbLayout.addLayout(lineLayout)
lineLayout = QtGui.QHBoxLayout()
# self.chkZipping = QtGui.QCheckBox()
# lineLayout.addWidget(self.chkZipping)
self.btnSave = QtGui.QPushButton()
self.btnSave.clicked.connect(self.onSaveClicked)
lineLayout.addWidget(self.btnSave)
gbLayout.addLayout(lineLayout)
self.gbSave.setLayout(gbLayout)
layout.addWidget(self.gbSave)
# log block
self.logInfo = QtGui.QTextEdit()
layout.addWidget(self.logInfo)
self._loggerHandler.logged.connect(self.logInfo.append)
# buttons block
subLayout = QtGui.QHBoxLayout()
self.btnNextAction = QtGui.QPushButton()
self.btnNextAction.clicked.connect(self.onNextActionClicked)
subLayout.addStretch()
subLayout.addWidget(self.btnNextAction)
self.btnClose = QtGui.QPushButton()
self.btnClose.clicked.connect(self.onCloseClicked)
subLayout.addWidget(self.btnClose)
layout.addLayout(subLayout)
self.setLayout(layout)
self.setMinimumWidth(512)
self.retranslateUi()
# noinspection PyAttributeOutsideInit
def retranslateUi(self):
context = unicode(getClassName(self))
self.setWindowTitle(
forceTr(u'Экспорт. Крым. МежТер. Возврат.', context))
self.gbInit.setTitle(forceTr(u'Параметры экспорта', context))
self.lblAccountNumber.setText(
forceTr(u'Номер выгрузки по счету', context))
self.gbExport.setTitle(forceTr(u'Экспорт', context))
self.gbSave.setTitle(forceTr(u'Сохранение результата', context))
self.btnSave.setText(forceTr(u'Сохранить', context))
self._actionNames = {
self.InitState: forceTr(u'Экспорт', context),
self.ExportState: forceTr(u'Прервать', context),
self.SaveState: forceTr(u'Повторить', context)
}
self.btnClose.setText(forceTr(u'Закрыть', context))
@QtCore.pyqtSlot()
def onNextActionClicked(self):
if self.currentState == self.InitState:
self.currentState = self.ExportState
try:
result = self._engine.process(self._accountId)
except Exception, e:
self.currentState = self.InitState
raise
self.currentState = self.SaveState if result else self.InitState
elif self.currentState == self.ExportState:
self._engine.abort()
