def save_report(self):
"""Save report"""
filename = QFileDialog.getSaveFileName(
self, "Save Report", self.save_dir,
"HTML (*.html);;PDF (*.pdf);;Report (*.report)")
if not filename:
return QDialog.Rejected
self.save_dir = os.path.dirname(filename)
self.saveSettings()
_, extension = os.path.splitext(filename)
if extension == ".pdf":
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(filename)
self.report_view.print_(printer)
elif extension == ".report":
with open(filename, 'wb') as f:
pickle.dump(self, f)
else:
frame = self.report_view.page().currentFrame()
with open(filename, "w") as f:
f.write(frame.documentElement().toInnerXml())
self.report_changed = False
return QDialog.Accepted
def save_report(self):
"""Save report"""
filename = QFileDialog.getSaveFileName(
self, "Save Report", self.save_dir,
"HTML (*.html);;PDF (*.pdf);;Report (*.report)")
if not filename:
return QDialog.Rejected
self.save_dir = os.path.dirname(filename)
self.saveSettings()
_, extension = os.path.splitext(filename)
if extension == ".pdf":
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(filename)
self.report_view.print_(printer)
elif extension == ".report":
self.save(filename)
else:
frame = self.report_view.page().currentFrame()
with open(filename, "w", encoding="utf-8") as f:
f.write(frame.documentElement().toInnerXml())
self.report_changed = False
return QDialog.Accepted
def savePdf(self):
"""
Save to pdf file
"""
fileName = QFileDialog.getSaveFileName(
self, 'Save to PDF', "", "PDF file (*.pdf);;All Files (*.*)")
# new in v17.1
if QtHelper.IS_QT5:
_filename, _type = fileName
else:
_filename = fileName
# end of new
if _filename:
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setColorMode(QPrinter.Color)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(_filename)
doc = QTextDocument()
if self.toXml:
doc.setPlainText(self.txtEdit.text())
else:
doc.setHtml(self.txtEdit.toHtml())
doc.print_(printer)
def on_actionPreview_activated( self ):
printer = QPrinter()
printer.setPageSize( QPrinter.Letter )
web = "balancegeneral.php?date=%d+%d" % (
self.dtPicker.date().month() ,
self.dtPicker.date().year() )
report = frmReportes( web , printer, self )
report.exec_()
def printWindow(self, printer=None):
if not printer:
printer = QPrinter()
printer.setFullPage(True)
printer.setPageSize(QPrinter.A4)
dialog = QPrintDialog(printer, self)
dialog.setWindowTitle(QCoreApplication.translate('QwtPlot', 'Print Document'))
if dialog.exec_() != QDialog.Accepted:
return
self.plot.print_(printer, QwtPlotPrintFilter())
def writePdf(dir_):
newbody = originHTML.format(**htmlpara)
printer = QPrinter()
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(dir_ + '.pdf')
printer.setPageSize(QPrinter.A4)
text = QTextDocument()
text.setHtml(newbody.decode('utf-8'))
text.print_(printer)
def writePdf(dir_):
newbody = originHTML.format(**htmlpara)
printer = QPrinter()
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(dir_+'.pdf')
printer.setPageSize(QPrinter.A4)
text = QTextDocument()
text.setHtml(newbody.decode('utf-8'))
text.print_(printer)
def handlePrintPdf(self):
printer = QPrinter()
pdffile = 'test.pdf'
printer.setResolution(96)
printer.setPageSize(QPrinter.Letter)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(pdffile)
printer.setPageMargins(5, 5, 5, 10, QPrinter.Millimeter)
document = self.makeTableDocument()
document.setPageSize(QSizeF(printer.pageRect().size()))
document.print_(printer)
def print_pdf(html, destination):
app = QApplication(sys.argv)
web = QWebView()
web.setHtml(html)
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(destination)
web.print_(printer)
app.exit()
def lunchPrintPdf(self, printer):
fileName = QFileDialog.getSaveFileName(self, 'Save File as', '', '*.pdf')
document = self.doc1
#document = QTextDocument(document)
printer = QPrinter()
printer.setResolution(96)
printer.setPageMargins(5, 5, 5, 5, QPrinter.Millimeter)
printer.setPageSize(QPrinter.Letter)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(fileName)
#document.setPageSize(QSizeF(printer.pageRect().size()))
#print(document.pageSize(), printer.resolution(), printer.pageRect())
document.print_(printer)
def printWindow(self, printer=None):
if not printer:
printer = QPrinter()
printer.setFullPage(True)
printer.setPageSize(QPrinter.A4)
dialog = QPrintDialog(printer, self)
dialog.setWindowTitle(QCoreApplication.translate('Graphics', 'Print Document'))
if dialog.exec_() != QDialog.Accepted:
return
# FIXME: on windows the resolutions seems to be very low, why?
#printer.setResolution(600)
painter = QPainter(printer)
painter.setRenderHint(QPainter.Antialiasing)
self.graphicsView.scene().render(painter)
def writePdfabs(dir_):
updates = SETTING().get('pdfpara')
htmlpara.update(updates)
newbody = originHTML.format(**htmlpara)
# with open('t.html', 'wb') as f:
# f.write(newbody)
# f.close()
printer = QPrinter()
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(dir_)
printer.setPageSize(QPrinter.A4)
text = QTextDocument()
text.setHtml(newbody.decode('utf-8'))
text.print_(printer)
def writePdfabs(dir_):
updates = PDF_PARAMETER
htmlpara.update(updates)
newbody = originHTML.format(**htmlpara)
# with open('t.html', 'wb') as f:
# f.write(newbody)
# f.close()
printer = QPrinter()
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(dir_)
printer.setPageSize(QPrinter.A4)
text = QTextDocument()
text.setHtml(newbody.decode('utf-8'))
text.print_(printer)
def init_print(self, linenos=True, style="default"):
app = QApplication(sys.argv) # noqa
doc = QTextDocument()
doc.setHtml(
self.highlight_file(linenos=linenos, style=style)
)
printer = QPrinter()
printer.setOutputFileName(self.pdf_file)
printer.setOutputFormat(QPrinter.PdfFormat)
page_size_dict = {"a2": QPrinter.A2, "a3": QPrinter.A3, "a4": QPrinter.A4}
printer.setPageSize(page_size_dict.get(self.size.lower(), QPrinter.A4))
printer.setPageMargins(15, 15, 15, 15, QPrinter.Millimeter)
doc.print_(printer)
logging.info("PDF created at %s" % (self.pdf_file))
def init_print(self, linenos=True, style="default"):
app = QApplication(sys.argv) # noqa
doc = QTextDocument()
doc.setHtml(self.highlight_file(linenos=linenos, style=style))
printer = QPrinter()
printer.setOutputFileName(self.pdf_file)
printer.setOutputFormat(QPrinter.PdfFormat)
page_size_dict = {
"a2": QPrinter.A2,
"a3": QPrinter.A3,
"a4": QPrinter.A4
}
printer.setPageSize(page_size_dict.get(self.size.lower(), QPrinter.A4))
printer.setPageMargins(15, 15, 15, 15, QPrinter.Millimeter)
doc.print_(printer)
logging.info("PDF created at %s" % (self.pdf_file))
def print_pdf(html, filename):
""" Print HTML to PDF. """
wv = QWebView()
wv.setHtml(html)
# doc = QTextDocument()
# doc.setHtml(html)
printer = QPrinter()
printer.setOutputFileName(filename)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setPageSize(QPrinter.A4)
printer.setPageMargins(15, 15, 15, 15, QPrinter.Millimeter)
# doc.print_(printer)
wv.print_(printer)
print("PDF Generated: " + filename)
def print_pdf(html, filename):
""" Print HTML to PDF. """
wv = QWebView()
wv.setHtml(html)
# doc = QTextDocument()
# doc.setHtml(html)
printer = QPrinter()
printer.setOutputFileName(filename)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setPageSize(QPrinter.A4)
printer.setPageMargins(15, 15, 15, 15, QPrinter.Millimeter)
# doc.print_(printer)
wv.print_(printer)
print("PDF Generated: " + filename)
def renderpdf(url, dest):
app = QApplication(sys.argv)
web = QWebView()
request = QNetworkRequest(QUrl(url))
request.setRawHeader("X-Rendering-PDF", "true")
web.load(request)
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(dest)
def print_file():
web.print_(printer)
QApplication.exit()
QObject.connect(web, SIGNAL("loadFinished(bool)"), print_file)
app.exec_()
def print_file(fileName, printFunction):
"""This method print a file
This method print a file, fileName is the default fileName,
and printFunction is a funcion that takes a QPrinter
object and print the file,
the print method
More info on:http://doc.qt.nokia.com/latest/printing.html"""
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fileName)
printer.setDocName(fileName)
preview = QPrintPreviewDialog(printer)
preview.paintRequested[QPrinter].connect(printFunction)
size = QApplication.instance().desktop().screenGeometry()
width = size.width() - 100
height = size.height() - 100
preview.setMinimumSize(width, height)
preview.exec_()
def open_html_in_print_preview_from_gui_thread(html,
html_document=QTextDocument, page_size=None, page_orientation=None):
printer = QPrinter()
printer.setPageSize(page_size or QPrinter.A4)
printer.setOrientation(page_orientation or QPrinter.Portrait)
dialog = QPrintPreviewDialog(printer)
@QtCore.pyqtSlot(QPrinter)
def render(printer):
doc = html_document()
doc.setHtml(html)
doc.print_(printer)
dialog.paintRequested.connect(render)
# show maximized seems to trigger a bug in qt which scrolls the page down
#dialog.showMaximized()
desktop = QCoreApplication.instance().desktop()
# use the size of the desktop instead to set the dialog size
dialog.resize(desktop.width() * 0.75, desktop.height() * 0.75)
dialog.exec_()
def print_file(fileName, printFunction):
"""This method print a file
This method print a file, fileName is the default fileName,
and printFunction is a funcion that takes a QPrinter
object and print the file,
the print method
More info on:http://doc.qt.nokia.com/latest/printing.html"""
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fileName)
printer.setDocName(fileName)
preview = QPrintPreviewDialog(printer)
preview.paintRequested[QPrinter].connect(printFunction)
size = QApplication.instance().desktop().screenGeometry()
width = size.width() - 100
height = size.height() - 100
preview.setMinimumSize(width, height)
preview.exec_()
def preview(self):
"""
Muestra el dialogo de vista previa de impresión
"""
try:
printer = QPrinter()
printer.setOrientation(self.orientation)
printer.setPageSize(self.pageSize)
web = self.web + self.printIdentifier
report = reports.frmReportes(web, printer, self)
report.exec_()
except NotImplementedError as inst:
QMessageBox.information(
self, qApp.organizationName(), u"No se ha implementado la función de impresión para este modulo"
)
logging.error(unicode(inst))
except UserWarning as inst:
QMessageBox.critical(self, qApp.organizationName(), unicode(inst))
logging.error(unicode(inst))
except Exception as inst:
QMessageBox.critical(self, qApp.organizationName(), "Hubo un error al intentar mostrar su reporte")
logging.critical(unicode(inst))
def open_html_in_print_preview_from_gui_thread(html,
html_document=QTextDocument,
page_size=None,
page_orientation=None):
printer = QPrinter()
printer.setPageSize(page_size or QPrinter.A4)
printer.setOrientation(page_orientation or QPrinter.Portrait)
dialog = QPrintPreviewDialog(printer)
@QtCore.pyqtSlot(QPrinter)
def render(printer):
doc = html_document()
doc.setHtml(html)
doc.print_(printer)
dialog.paintRequested.connect(render)
# show maximized seems to trigger a bug in qt which scrolls the page down
#dialog.showMaximized()
desktop = QCoreApplication.instance().desktop()
# use the size of the desktop instead to set the dialog size
dialog.resize(desktop.width() * 0.75, desktop.height() * 0.75)
dialog.exec_()
def generate_pdf(template, filename, media_path=None, template_context=None):
global app
if template_context is None:
template_context = {}
template_context['media_path'] = media_path
#put your templates in the pdf subdirectory
template_path = '{0}.html'.format(template)
template = jinja_env.get_template(template_path)
html = template.render(**template_context)
app = QApplication.instance()
if app is None:
app = QApplication(['--platform', 'minimal'])
web = QWebView()
url = QUrl.fromLocalFile(media_path)
web.setHtml(html, baseUrl=url)
#we need this call to correctly render images...
app.processEvents()
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(filename)
web.print_(printer)
def _save_report(self):
filename = QFileDialog.getSaveFileName(
self, "Save Report", self.save_dir,
"HTML (*.html);;PDF (*.pdf);;Report (*.report)")
if not filename:
return
self.save_dir = os.path.dirname(filename)
self.saveSettings()
_, extension = os.path.splitext(filename)
if extension == ".pdf":
printer = QPrinter()
printer.setPageSize(QPrinter.A4)
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(filename)
self.report_view.print_(printer)
elif extension == ".report":
with open(filename, 'wb') as f:
pickle.dump(self, f)
else:
frame = self.report_view.page().currentFrame()
with open(filename, "w") as f:
f.write(frame.documentElement().toInnerXml())
def print_file(fileName, printFunction):
"""This method print a file
This method print a file, fileName is the default fileName,
and printFunction is a funcion that takes a QPrinter
object and print the file,
the print method
More info on:http://doc.qt.nokia.com/latest/printing.html"""
#TODO: Make that the option it's taken from settings. maybe
#this should be a class to manage all cases?
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fileName)
printer.setDocName(fileName)
#PRINT PREVIEW OPTIONS
#TODO: the follow imp. dipatch an error on execution time we need to review
#if the error belongs to qt
# preview = QPrintPreviewDialog(printer)
# preview.paintRequested[QPrinter].connect(printFunction)
# preview.exec_()
dialog = QPrintDialog(printer)
if dialog.exec_():
printFunction(printer)
def print_file(fileName, printFunction):
"""This method print a file
This method print a file, fileName is the default fileName,
and printFunction is a funcion that takes a QPrinter
object and print the file,
the print method
More info on:http://doc.qt.nokia.com/latest/printing.html"""
#TODO: Make that the option it's taken from settings. maybe
#this should be a class to manage all cases?
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fileName)
printer.setDocName(fileName)
#PRINT PREVIEW OPTIONS
#TODO: the follow imp. dipatch an error on execution time we need to review
#if the error belongs to qt
# preview = QPrintPreviewDialog(printer)
# preview.paintRequested[QPrinter].connect(printFunction)
# preview.exec_()
dialog = QPrintDialog(printer)
if dialog.exec_():
printFunction(printer)
class MainWindow(QWidget, MainWindowGeneric):
def __init__(self, parent):
QWidget.__init__(self)
MainWindowGeneric.__init__(self)
self._parent = parent
self._vbox = QVBoxLayout(self)
#Splitters
self.splitterMain = QSplitter()
self.splitterCentral = QSplitter(Qt.Vertical)
#Properties Panel
self._properties = PropertiesWidget(self)
#Central
self._central = CentralWidget(self)
self.show_start_page()
self.splitterCentral.addWidget(self._central)
#Display Container
self.container = DisplayContainer(self)
self._hide_container()
self.splitterCentral.addWidget(self.container)
height = [(self.height() / 3) * 2, self.height() / 3]
self.splitterCentral.setSizes([height[0], height[1]])
#Size Central Splitter
self.splitterMain.addWidget(self.splitterCentral)
self.splitterMain.addWidget(self._properties)
width = [(self.width() / 6) * 5, self.width() / 6]
self.splitterMain.setSizes([width[0], width[1]])
self._vbox.addWidget(self.splitterMain)
#flag for reload_file
self._reloading = False
#Shortcuts
shortCloseTab = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_W), self)
shortNew = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_N), self)
shortNewProject = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_J), self)
shortOpen = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_O), self)
shortOpenProject = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_P), self)
shortSave = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_S), self)
shortSaveAll = QShortcut(QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_S), self)
shortPrint = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_I), self)
shortRedo = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_Y), self)
shortComment = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_D), self)
shortHorizontalLine = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_R), self)
shortTitleComment = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_T), self)
shortIndentLess = QShortcut(QKeySequence(Qt.SHIFT + Qt.Key_Tab), self)
shortHideContainer = QShortcut(QKeySequence(Qt.Key_F4), self)
shortHideEditor = QShortcut(QKeySequence(Qt.Key_F3), self)
shortHideExplorer = QShortcut(QKeySequence(Qt.Key_F2), self)
shortRunFile = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_F6), self)
shortRunProgram = QShortcut(QKeySequence(Qt.Key_F6), self)
shortHideAll = QShortcut(QKeySequence(Qt.Key_F11), self)
shortFind = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_F), self)
shortFindReplace = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_H), self)
shortHelp = QShortcut(QKeySequence(Qt.Key_F1), self)
shortSplitHorizontal = QShortcut(QKeySequence(Qt.Key_F10), self)
shortSplitVertical = QShortcut(QKeySequence(Qt.Key_F9), self)
shortFollowMode = QShortcut(QKeySequence(Qt.CTRL + Qt.Key_F10), self)
shortReloadFile = QShortcut(QKeySequence(Qt.Key_F5), self)
shortShowProjectsTree = QShortcut(QKeySequence(Qt.ALT + Qt.Key_1), self)
shortShowSymbolsTree = QShortcut(QKeySequence(Qt.ALT + Qt.Key_2), self)
#Signal -> Slot
self.connect(shortCloseTab, SIGNAL("activated()"), self.close_actual_tab)
self.connect(shortNew, SIGNAL("activated()"), self.new_editor)
self.connect(shortNewProject, SIGNAL("activated()"), self.new_project)
self.connect(shortOpen, SIGNAL("activated()"), self.open_file)
self.connect(shortOpenProject, SIGNAL("activated()"), self.open_project_folder)
self.connect(shortSave, SIGNAL("activated()"), self.save)
self.connect(shortSaveAll, SIGNAL("activated()"), self.save_project)
self.connect(shortPrint, SIGNAL("activated()"), self._print_file)
self.connect(shortComment, SIGNAL("activated()"), lambda: self._central.obtain_editor().comment())
self.connect(shortIndentLess, SIGNAL("activated()"), lambda: self._central.obtain_editor().indent_less())
self.connect(shortHorizontalLine, SIGNAL("activated()"), lambda: self._central.obtain_editor().insert_horizontal_line())
self.connect(shortTitleComment, SIGNAL("activated()"), lambda: self._central.obtain_editor().insert_title_comment())
self.connect(shortRedo, SIGNAL("activated()"), lambda: self._central.obtain_editor().redo())
self.connect(shortHideContainer, SIGNAL("activated()"), self._hide_container)
self.connect(shortHideEditor, SIGNAL("activated()"), self._hide_editor)
self.connect(shortHideExplorer, SIGNAL("activated()"), self._hide_explorer)
self.connect(shortRunFile, SIGNAL("activated()"), self._run_code)
self.connect(shortRunProgram, SIGNAL("activated()"), self._run_program)
self.connect(shortHideAll, SIGNAL("activated()"), self._hide_all)
self.connect(shortFind, SIGNAL("activated()"), self._open_find)
self.connect(shortFindReplace, SIGNAL("activated()"), self._open_find_replace)
self.connect(shortHelp, SIGNAL("activated()"), self._show_python_doc)
self.connect(shortSplitHorizontal, SIGNAL("activated()"), lambda: self.split_tab(True))
self.connect(shortSplitVertical, SIGNAL("activated()"), lambda: self.split_tab(False))
self.connect(shortFollowMode, SIGNAL("activated()"), self._view_follow_mode)
self.connect(shortReloadFile, SIGNAL("activated()"), lambda: self.reload_file())
self.connect(shortShowProjectsTree, SIGNAL("activated()"), self.show_projects_tree)
self.connect(shortShowSymbolsTree, SIGNAL("activated()"), self.show_symbols_tree)
def change_window_title(self, title):
self._parent.setWindowTitle('NINJA-IDE - ' + title)
def _open_find(self):
if not self._parent._status.isVisible():
self._parent._status.show()
self._parent._status.focus_find(self._central.obtain_editor())
def _open_find_replace(self):
if not self._parent._status.isVisible():
self._parent._status.show()
self._parent._status.replace_visibility(True)
self._parent._status.focus_find(self._central.obtain_editor())
def _print_file(self):
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.A4)
if self._central.obtain_editor().path:
fileName = manage_files.get_basename(self._central.obtain_editor().path)
fileName = fileName[:fileName.rfind('.')] + '.pdf'
else:
fileName = 'newDocument.pdf'
self.printer.setOutputFileName(fileName)
dialog = QPrintDialog(self.printer, self)
if dialog.exec_():
self.printer.setDocName(manage_files.get_basename(self._central.obtain_editor().path))
self._central.obtain_editor().document().print_(self.printer)
def _hide_container(self):
if self.containerIsVisible:
self.container.hide()
self.containerIsVisible = False
self._central.obtain_editor().setFocus()
else:
self.container.show()
self.containerIsVisible = True
self.container.gain_focus()
def _hide_editor(self):
if self._central.isVisible():
self._central.hide()
else:
self._central.show()
def _hide_explorer(self):
if self._properties.isVisible():
self._properties.hide()
else:
self._properties.show()
def _splitter_central_orientation(self):
if self.splitterCentral.orientation() == Qt.Horizontal:
self.splitterCentral.setOrientation(Qt.Vertical)
else:
self.splitterCentral.setOrientation(Qt.Horizontal)
def get_splitter_central_orientation(self):
return self.splitterCentral.orientation()
def _splitter_main_orientation(self):
if self.splitterMain.orientation() == Qt.Horizontal:
self.splitterMain.setOrientation(Qt.Vertical)
else:
self.splitterMain.setOrientation(Qt.Horizontal)
def get_splitter_main_orientation(self):
return self.splitterMain.orientation()
def _splitter_main_rotate(self):
w = self.splitterMain.widget(0)
w1 = self.splitterMain.widget(1)
if type(w) is PropertiesWidget:
w.setTabPosition(QTabWidget.East)
else:
w1.setTabPosition(QTabWidget.West)
self.splitterMain.insertWidget(0, w1)
self.splitterMain.insertWidget(1, w)
def get_splitter_main_position(self):
w = self.splitterMain.widget(0)
return w.__class__
def _splitter_central_rotate(self):
w = self.splitterCentral.widget(0)
w1 = self.splitterCentral.widget(1)
self.splitterCentral.insertWidget(0, w1)
self.splitterCentral.insertWidget(1, w)
def get_splitter_central_position(self):
w = self.splitterCentral.widget(0)
return w.__class__
def get_splitter_position_0(self):
return type(self.splitterCentral.widget(0))
def get_splitter_main_position_0(self):
return type(self.splitterMain.widget(0))
def reload_panels_position(self):
self.settings = QSettings('NINJA-IDE', 'Kunai')
self.settings.beginGroup('Preferences')
self.settings.beginGroup('Interface')
#first with the splitterCentral
c = self.splitterCentral.widget(0)
c1 = self.splitterCentral.widget(1)
if (type(c) == CentralWidget):
self.splitterCentral.insertWidget(self.settings.value('central_tab_position', 0).toInt()[0], c)
self.splitterCentral.insertWidget(self.settings.value('container_tab_position', 1).toInt()[0], c1)
else:
self.splitterCentral.insertWidget(self.settings.value('central_tab_position', 0).toInt()[0], c1)
self.splitterCentral.insertWidget(self.settings.value('container_tab_position', 1).toInt()[0], c)
#now with the splitterMain
m = self.splitterMain.widget(0)
m1 = self.splitterMain.widget(1)
if (type(m) == QSplitter):
self.splitterMain.insertWidget(self.settings.value('central_tab_position', 0).toInt()[0], m)
self.splitterMain.insertWidget(self.settings.value('container_tab_position', 1).toInt()[0], m1)
else:
self.splitterMain.insertWidget(self.settings.value('central_tab_position', 0).toInt()[0], m1)
self.splitterMain.insertWidget(self.settings.value('container_tab_position', 1).toInt()[0], m)
def get_open_projects(self):
return self._properties._treeProjects.get_open_projects()
def _run_code(self):
if self.save():
self.container.show()
self.containerIsVisible = True
editor = self._central.obtain_editor()
ext = self.get_file_extension(editor.path)
if ext == 'html':
height = self.height() / 3
self.splitterCentral.setSizes([height, height * 2])
self.container.render_web_page(editor.path)
elif ext == 'py':
height = self.height() / 3
self.splitterCentral.setSizes([height * 2, height])
self.container.run_application(editor.path)
else:
self.execute_file(editor.path, ext)
def _run_program(self, actual=None):
self.container.show()
self.containerIsVisible = True
if actual is None:
actual = self._properties._treeProjects.actualProject
if actual is None:
return
mainFile = actual.mainFile
if mainFile == '':
self._properties._treeProjects.open_project_properties()
self.containerIsVisible = False
self.container.hide()
return
path = manage_files.create_abs_path(actual.path, mainFile)
self._central.save_project_files(actual.path)
lang = actual.lang()
type_ = actual.projectType
if lang == 'html':
height = self.height() / 3
self.splitterCentral.setSizes([height, height * 2])
self.container.render_web_page(path)
elif lang == 'py':
height = self.height() / 3
self.splitterCentral.setSizes([height * 2, height])
self.container.run_application(path, actual.pythonPath)
else:
self.execute_program(path, lang, type_)
def _stop_program(self):
self.container.kill_application()
def _hide_all(self):
if self._properties.isVisible():
self._properties.hide()
self.container.hide()
self._parent._toolbar.hide()
else:
if self.containerIsVisible:
self.container.show()
self._properties.show()
self._parent._toolbar.show()
def show_start_page(self):
startPage = Browser(resources.start_page_url, None, self)
self.add_tab(startPage, 'Start Page')
def show_report_bugs(self):
bugsPage = Browser(resources.bugs_page)
self.add_tab(bugsPage, 'Report Bugs!')
def show_plugins_doc(self):
bugsPage = Browser(resources.plugins_doc)
self.add_tab(bugsPage, 'How to Write NINJA-IDE plugins')
def _show_python_doc(self):
process = runner.start_pydoc()
docPage = Browser(process[1], process[0])
self.add_tab(docPage, 'Python Documentation')
def new_editor(self, lang='py'):
if not self._reloading:
editor = factory_editor(lang, self._central.actual_tab())
self.add_tab(editor, 'New Document')
def add_tab(self, component, title):
self._central.actual_tab().add_tab(component, title)
def split_tab(self, option):
if option:
self._central.show_split(Qt.Horizontal)
else:
self._central.show_split(Qt.Vertical)
def _view_follow_mode(self):
self._central._show_follow_mode()
def new_project(self):
project = WizardNewProject(self)
project.show()
def show_preferences(self):
prefs = PreferencesWindow(self)
prefs.show()
def open_document(self, fileName, project=None):
try:
if not self._central.actual_tab().is_open(fileName):
self._central.actual_tab().notOpening = False
editor = factory_editor(fileName, self._central.actual_tab(), project)
content = self.read_file_content(fileName)
editor.setPlainText(content)
editor.path = fileName
editor.ask_if_externally_modified = True
#self.add_tab(editor, self.get_file_name(fileName))
if not editor.has_write_permission():
fileName += ' (Read-Only)'
self.add_tab(editor, self.get_file_name(fileName))
self.change_window_title(fileName)
editor.find_errors_and_check_style()
else:
self._central.actual_tab().move_to_open(fileName)
except Exception, reason:
print reason
QMessageBox.information(self, 'Incorrect File', 'The file couldn\'t be open')
self._central.actual_tab().notOpening = True
def save(self, imgName, w=None, h=None, header=None, \
dpi=150, take_region=False):
ext = imgName.split(".")[-1].upper()
root = self.startNode
#aspect_ratio = root.fullRegion.height() / root.fullRegion.width()
aspect_ratio = self.i_height / self.i_width
# auto adjust size
if w is None and h is None and (ext == "PDF" or ext == "PS"):
w = dpi * 6.4
h = w * aspect_ratio
if h > dpi * 11:
h = dpi * 11
w = h / aspect_ratio
elif w is None and h is None:
w = self.i_width
h = self.i_height
elif h is None:
h = w * aspect_ratio
elif w is None:
w = h / aspect_ratio
if ext == "SVG":
svg = QtSvg.QSvgGenerator()
svg.setFileName(imgName)
svg.setSize(QtCore.QSize(w, h))
svg.setViewBox(QtCore.QRect(0, 0, w, h))
#svg.setTitle("SVG Generator Example Drawing")
#svg.setDescription("An SVG drawing created by the SVG Generator")
pp = QtGui.QPainter()
pp.begin(svg)
targetRect = QtCore.QRectF(0, 0, w, h)
self.render(pp, targetRect, self.sceneRect())
pp.end()
elif ext == "PDF" or ext == "PS":
format = QPrinter.PostScriptFormat if ext == "PS" else QPrinter.PdfFormat
printer = QPrinter(QPrinter.HighResolution)
printer.setResolution(dpi)
printer.setOutputFormat(format)
printer.setPageSize(QPrinter.A4)
pageTopLeft = printer.pageRect().topLeft()
paperTopLeft = printer.paperRect().topLeft()
# For PS -> problems with margins
# print paperTopLeft.x(), paperTopLeft.y()
# print pageTopLeft.x(), pageTopLeft.y()
# print printer.paperRect().height(), printer.pageRect().height()
topleft = pageTopLeft - paperTopLeft
printer.setFullPage(True)
printer.setOutputFileName(imgName)
pp = QtGui.QPainter(printer)
if header:
pp.setFont(QtGui.QFont("Verdana", 12))
pp.drawText(topleft.x(), 20, header)
targetRect = QtCore.QRectF(topleft.x(), 20 + (topleft.y() * 2),
w, h)
else:
targetRect = QtCore.QRectF(topleft.x(), topleft.y() * 2, w, h)
if take_region:
self.selector.setVisible(False)
self.render(pp, targetRect, self.selector.rect())
self.selector.setVisible(True)
else:
self.render(pp, targetRect, self.sceneRect())
pp.end()
return
else:
targetRect = QtCore.QRectF(0, 0, w, h)
ii= QtGui.QImage(w, \
h, \
QtGui.QImage.Format_ARGB32)
pp = QtGui.QPainter(ii)
pp.setRenderHint(QtGui.QPainter.Antialiasing)
pp.setRenderHint(QtGui.QPainter.TextAntialiasing)
pp.setRenderHint(QtGui.QPainter.SmoothPixmapTransform)
if take_region:
self.selector.setVisible(False)
self.render(pp, targetRect, self.selector.rect())
self.selector.setVisible(True)
else:
self.render(pp, targetRect, self.sceneRect())
pp.end()
ii.save(imgName)
class Form(QDialog):
def __init__(self, args=None,parent=None):
super(Form, self).__init__(parent)
self.printer = QPrinter()
self.printer.setPageSize(QPrinter.A4)
self.filename = ''
#File name to convert
lblFileName = QLabel("File Name")
self.txtFileName = QLineEdit()
if len(args) > 1:
self.txtFileName.setText(args[1])
self.filename = args[1]
self.txtFileName.setReadOnly(True)
btnGetFileName = QToolButton()
btnGetFileName.setText("..")
getFileLayout = QHBoxLayout()
getFileLayout.addWidget(lblFileName)
getFileLayout.addWidget(self.txtFileName)
getFileLayout.addWidget(btnGetFileName)
#Code page of file
lblCodePage = QLabel("Code Page")
self.cmbCodePage = QComboBox()
self.cmbCodePage.addItem("dos")
self.cmbCodePage.addItem("windows")
self.cmbCodePage.addItem("utf8")
codePageLayout = QHBoxLayout()
codePageLayout.addStretch()
codePageLayout.addWidget(lblCodePage)
codePageLayout.addWidget(self.cmbCodePage)
#Printer Orientation (Portrait / Landscape)
lblOrientation = QLabel("Orientation")
self.cmbOrientation = QComboBox()
self.cmbOrientation.addItem("portait")
self.cmbOrientation.addItem("landscape")
orientationLayout = QHBoxLayout()
orientationLayout.addStretch()
orientationLayout.addWidget(lblOrientation)
orientationLayout.addWidget(self.cmbOrientation)
#Buttons for save or Just quit
bSave = QPushButton("Go for it")
bQuit = QPushButton("Abort")
buttonLayout = QHBoxLayout()
buttonLayout.addWidget(bQuit)
buttonLayout.addStretch()
buttonLayout.addWidget(bSave)
#Final Layout
layout = QVBoxLayout()
layout.addLayout(getFileLayout)
layout.addLayout(codePageLayout)
layout.addLayout(orientationLayout)
layout.addLayout(buttonLayout)
self.setLayout(layout)
#Connections
self.connect(bSave, SIGNAL("clicked()"),self.printViaQPainter)
self.connect(bQuit, SIGNAL("clicked()"), self.accept)
self.connect(btnGetFileName, SIGNAL("clicked()"),self.getFileName)
self.setWindowTitle("dos/windows text to PDF")
def getFileName(self):
fd = QFileDialog(self)
self.filename = fd.getOpenFileName()
self.txtFileName.setText(self.filename)
def printViaQPainter(self):
from os.path import isfile
if isfile(self.filename):
lines = open(self.filename)
else:
return
a = self.filename.split('.')
s = a[-1]
fname = ''
for i in range(len(a)-1) :
fname += a[i]
fname += '.pdf'
self.printer.setOutputFileName(fname)
self.printer.setOutputFormat(QPrinter.PdfFormat)
if self.cmbOrientation.currentText() == 'landscape':
self.printer.setOrientation(QPrinter.Landscape)
pageRect = self.printer.pageRect()
LeftMargin = 15
biggest = findBiggestLine(self.filename)
bestSize = findBestFontSize(biggest,pageRect)
sansFont = QFont("Courier", bestSize)
sansLineHeight = QFontMetrics(sansFont).height()
painter = QPainter(self.printer)
page = 1
y = 20
cpage = codePage(self.cmbCodePage.currentText())
for line in lines:
painter.save()
painter.setFont(sansFont)
y += sansLineHeight
x = LeftMargin
try:
painter.drawText(x,y,line[:-1].decode(cpage))
except:
painter.drawText(x,y,'CodePage error !!!')
if y > (pageRect.height() - 54) :
self.printer.newPage()
y = 20
painter.restore()
lines.close()
self.accept()
class Base(QMainWindow):
"""
Esta clase sirve de base para todos aquellos
formularios que siguen el estandar de dos pestañas, una para navegación
y otra para edición
"""
orientation = QPrinter.Portrait
pageSize = QPrinter.Letter
web = ""
def __init__(self, parent, own_toolbar=False):
"""
@param parent: El widget padre de esta ventana
@param own_toolbar: Si este widget dibujara su toolbar en el padre o en el mismo
@type parent: QWidget
@type own_toolbar: bool
"""
super(Base, self).__init__(parent)
self.user = user.LoggedUser
self._status = True
self.parentWindow = parent
self.own_toolbar = own_toolbar
self.database = QSqlDatabase.database()
"""
@type: QSqlDatabase
@ivar: La base de datos a la cual se conecta el sistema
"""
self.mapper = QDataWidgetMapper(self)
u"""
@type: QDataWidgetMapper
@ivar: El mapper que se encarga de asignar los
datos del modelo de navegación a los distintos widgets
"""
self.printProgressBar = QProgressBar(self)
self.webview = QWebView()
"""
@ivar: EL objeto webview usado para cargar los reportes
@type: QWebView
"""
self.loaded = False
"""
@ivar: Si se pudo o no cargar un reporte
@type: bool
"""
self.startUi()
self.editmodel = None
self.printer = QPrinter()
self._status = False
def startUi(self):
"""
Iniciar todos los elementos graficos
"""
self.setupUi(self)
if not self.own_toolbar:
self.parent().addToolBar(self.toolBar)
self.removeToolBar(self.toolBar)
settings = QSettings()
self.restoreState(settings.value(self.windowTitle() + "/State").toByteArray())
"""
@ivar: El MainWindow al que pertenece este widget
"""
self.createActions()
self.printProgressBar.setVisible(False)
_tab1shortcut = QShortcut(QKeySequence("Ctrl+1"), self, functools.partial(self.tabWidget.setCurrentIndex, 0))
_tab2shortcut = QShortcut(QKeySequence("Ctrl+2"), self, functools.partial(self.tabWidget.setCurrentIndex, 1))
self.mapper.currentIndexChanged[int].connect(self.updateDetailFilter)
self.actionCut.setVisible(False)
self.actionPaste.setVisible(False)
self.actionCopy.setVisible(False)
def closeEvent(self, event):
u"""
Guardar el tamaño, la posición en la pantalla y la posición de
la barra de tareas
Preguntar si realmente se desea cerrar la pestaña cuando
se esta en modo edición
"""
if not self.status:
if (
not QMessageBox.question(
self, qApp.organizationName(), u"¿Está seguro que desea salir?", QMessageBox.Yes | QMessageBox.No
)
== QMessageBox.Yes
):
event.ignore()
# Guardar el tamaño y la posición
settings = QSettings()
settings.setValue(self.windowTitle() + "/Geometry", self.saveGeometry())
if not self.own_toolbar:
self.parent().mdiArea().parent().parent().removeToolBar(self.toolBar)
def editCell(self):
"""
Editar la celda actualmente seleccionada de self.tableedit
"""
self.tabledetails.edit(self.tabledetails.selectionModel().currentIndex())
@pyqtSlot(QDateTime)
@if_edit_model
def on_dtPicker_dateTimeChanged(self, datetime):
"""
Cambiar el tipo de cambio del modelo de edición si cambia la fecha
@param datetime: La fecha contenida en self.dtPicker
@type datetime: QDateTime
"""
query = QSqlQuery()
try:
if not self.database.isOpen():
if not self.database.open():
raise Exception(
"No se pudo conectar a la base de " + "datos para recuperar los tipos " + "de cambio"
)
q = """
SELECT idtc, tasa
FROM tiposcambio
WHERE fecha = %s
LIMIT 1
""" % datetime.toString(
"yyyyMMdd"
)
if not query.exec_(q):
raise UserWarning("No se pudieron recuperar los tipos de " + "cambio")
if not query.size() == 1:
logging.critical(u"La consulta para obtener tipos de " + "cambio no devolvio exactamente un valor")
raise UserWarning(u"Hubo un error al obtener los tipos " + "de cambio")
query.first()
self.editmodel.exchangeRateId = query.value(0).toInt()[0]
self.editmodel.exchangeRate = Decimal(query.value(1).toString())
# self.editmodel.setData( self.editmodel.index( 0, 0 ), self.editmodel.index( 0, 0 ).data() )
self.editmodel.datetime = datetime
except UserWarning as inst:
QMessageBox.critical(self, qApp.organizationName(), unicode(inst))
self.dtPicker.setDateTime(self.editmodel.datetime)
logging.error(inst)
logging.error(query.lastError().text())
except Exception as inst:
QMessageBox.critical(self, qApp.organizationName(), u"Hubo un error al obtener los tipos de" + " cambio")
logging.critical(query.lastError().text())
logging.critical(inst)
self.dtPicker.setDateTime(self.editmodel.datetime)
def navigate(self, to):
"""
Esta funcion se encarga de navegar entro los distintos documentos
@param to: es una string que puede tomar los valores 'next' 'previous'
'first' 'last'
"""
if self.mapper.currentIndex != -1:
row = self.mapper.currentIndex()
if to == "next":
row += 1
if row >= self.navproxymodel.rowCount():
row = self.navproxymodel.rowCount() - 1
self.mapper.setCurrentIndex(row)
elif to == "previous":
if row <= 1:
row = 0
else:
row = row - 1
self.mapper.setCurrentIndex(row)
elif to == "first":
self.mapper.toFirst()
elif to == "last":
self.mapper.toLast()
else:
self.mapper.toLast()()
if self.tabledetails != None:
self.tabledetails.resizeColumnsToContents()
self.tabledetails.horizontalHeader().setStretchLastSection(True)
self.tablenavigation.selectRow(self.mapper.currentIndex())
def updateDetailFilter(self, _index):
"""
Esta función se debe implementar en los formularios para que al
navegar se actualize el filtro de la tabla detalles
@param index: Este es el indice del mapper en el que actualmente
se encuentra navegando
@type index: int
"""
QMessageBox.information(self, qApp.organizationName(), u"Esta parte del sistema no ha " + "sido implementada")
raise NotImplementedError()
def loadModels(self):
"""
Esta función se ejecuta en el constructor del formulario mediante
un QTimer, carga los formularios por primera vez
"""
self.updateModels()
self.navigate("last")
self.status = True
def _setStatus(self, stat):
"""
@param stat: False = editando, True = navegando
@type stat: bool
"""
self._status = stat
self.setControls(self._status)
def _getStatus(self):
"""
esta propiedad cambia entre navegar y editar
"""
return self._status
status = property(_getStatus, _setStatus)
@pyqtSlot(unicode)
def on_txtSearch_textChanged(self, searchstring):
"""
Cambiar el filtro para el navigation model
@param searchstring: Es el contenido por el cual se va a
filtrar el modelo de navegación
@type searchstring: string
"""
self.navproxymodel.setFilterFixedString(searchstring)
@pyqtSlot(QModelIndex)
def on_tablenavigation_doubleClicked(self, index):
"""
Al hacer doble click en la tabla de navegación el se cambia a la
pestaña detalles mostrando el documento seleccionado
@param index: El indice de la tabla en la que se dio doble click
@type index: QModelIndex
"""
self.mapper.setCurrentIndex(index.row())
self.tabWidget.setCurrentIndex(0)
@pyqtSlot(QModelIndex)
def on_tablenavigation_clicked(self, index):
self.mapper.setCurrentIndex(index.row())
def save(self, ask=True):
"""
Guardar el documento actual
@param ask: Si se deberia o no preguntar al usuario si
esta seguro antes de proceder
@type ask: bool
"""
if (
ask == False
or QMessageBox.question(
self, qApp.organizationName(), u"¿Esta seguro que desea guardar?", QMessageBox.Yes | QMessageBox.No
)
== QMessageBox.Yes
):
if self.editmodel.valid:
if self.editmodel.save():
QMessageBox.information(self, qApp.organizationName(), u"El documento se ha guardado con éxito")
self.editmodel = None
self.updateModels()
self.navigate("last")
self.status = True
else:
QMessageBox.critical(self, qApp.organizationName(), "Ha ocurrido un error al guardar el documento")
else:
try:
QMessageBox.warning(self, qApp.organizationName(), self.editmodel.validError)
except AttributeError:
QMessageBox.warning(
self,
qApp.organizationName(),
u"El documento no puede guardarse" + " ya que la información no esta" + " completa",
)
def setControls(self, unused_status):
"""
Habilitar o deshabilitar los controles según status
@param status:
@type status: bool
"""
QMessageBox.information(self, qApp.organizationName(), u"Esta parte del sistema no ha sido implementada")
raise NotImplementedError()
def addLine(self):
"""
añadir una linea a table edit, solo se llama directamente
en una ocasion, al comenzar la edicion de un documento
"""
row = self.editmodel.rowCount()
self.editmodel.insertRows(row)
def createAction(self, text, slot=None, shortcut=None, icon=None, tip=None, checkable=False, signal="triggered"):
"""
Crear un objeto acción
@param text: El texto de la acción
@type text: string
@param slot: El slot que se ejecutara cuando se dispare esta acción
@type slot: callable
@param shortcut: El acceso directo que tiene asignada esta acción
@type shortcut: QKeySequence
@param icon: El icono de esta acción
@type icon: string
@param tip: El tooltip que tendra esta acción
@type tip: string
@param checkable: Si esta acción es checkable o no
@type checkable: bool
@param signal: La señal en la que esta acción ejecutara el slot
@type signal: string
@rtype: QAction
"""
action = QAction(text, self)
if icon is not None:
if type(icon) == QIcon:
action.setIcon(icon)
else:
action.setIcon(QIcon(icon))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
getattr(action, signal).connect(slot)
if checkable:
action.setCheckable(True)
return action
def newDocument(self):
"""
Empezar la edición de un nuevo documento
"""
QMessageBox.information(self, qApp.organizationName(), u"Esta parte del sistema no ha sido implementada")
raise NotImplementedError()
def cancel(self):
"""
Cancelar la edición del nuevo documento
"""
QMessageBox.information(self, qApp.organizationName(), u"Esta parte del sistema no ha sido implementada")
raise NotImplementedError()
@property
def printIdentifier(self):
"""
La identificación de este documento para reporte,
normalmente sera el iddocumento o el ndocimpreso
@rtype:string
"""
raise NotImplementedError(u"printIdentifier debe implementarse para " + "poder imprimir")
def preview(self):
"""
Muestra el dialogo de vista previa de impresión
"""
try:
printer = QPrinter()
printer.setOrientation(self.orientation)
printer.setPageSize(self.pageSize)
web = self.web + self.printIdentifier
report = reports.frmReportes(web, printer, self)
report.exec_()
except NotImplementedError as inst:
QMessageBox.information(
self, qApp.organizationName(), u"No se ha implementado la función de impresión para este modulo"
)
logging.error(unicode(inst))
except UserWarning as inst:
QMessageBox.critical(self, qApp.organizationName(), unicode(inst))
logging.error(unicode(inst))
except Exception as inst:
QMessageBox.critical(self, qApp.organizationName(), "Hubo un error al intentar mostrar su reporte")
logging.critical(unicode(inst))
def printDocument(self):
"""
Imprime el documento actual
"""
try:
base = reports.Reports.url
if base == "":
raise UserWarning(u"No existe una configuración para el " + "servidor de reportes")
self.printer.setOrientation(self.orientation)
self.printer.setPageSize(self.pageSize)
web = base + self.web + self.printIdentifier + "&uname=" + self.user.user + "&hash=" + self.user.hash
self.loaded = False
self.webview.load(QUrl(web))
self.webview.loadFinished[bool].connect(self.on_webview_loadFinished)
self.webview.loadProgress[int].connect(self.on_webview_loadProgress)
except NotImplementedError as inst:
logging.error(unicode(inst))
QMessageBox.information(
self, qApp.organizationName(), u"La función de impresión no se ha " + "implementado para este modulo"
)
except UserWarning as inst:
logging.error(unicode(inst))
QMessageBox.critical(self, qApp.organizationName(), unicode(inst))
except Exception as inst:
logging.critical(unicode(inst))
QMessageBox.critical(self, qApp.organizationName(), "Hubo un problema al intentar imprimir" + " su reporte")
def on_webview_loadProgress(self, progress):
"""
Muestra el progreso de la carga del reporte en un progressBar
"""
self.printProgressBar.setValue(progress)
def on_webview_loadFinished(self, status):
if self.printProgressBar.isVisible():
self.printProgressBar.hide()
if not status:
QMessageBox.critical(self, qApp.organizationName(), "El reporte no se pudo cargar")
logging.error("No se pudo cargar el reporte")
self.loaded = True
printdialog = QPrintDialog(self.printer, self)
if printdialog.exec_() == QDialog.Accepted:
self.webview.print_(self.printer)
del self.webview
def deleteRow(self):
"""
Funcion usada para borrar lineas de la tabla
"""
index = self.tabledetails.currentIndex()
if not index.isValid():
return
row = index.row()
self.editmodel.removeRows(row, 1)
self.updateLabels()
def updateLabels(self):
"""
Este metodo se llama para actualizar las etiquetas de totales en el
formulario
"""
raise NotImplementedError()
def createActions(self):
"""
Crea las acciones predefinidas del sistema
"""
self.actionNew = self.createAction(
text="Nuevo",
tip="Crear un nuevo documento",
icon=QIcon.fromTheme("document-new", QIcon(":/icons/res/document-new.png")),
shortcut="Ctrl+n",
slot=self.newDocument,
)
self.actionPreview = self.createAction(
text="Previsualizar",
tip=u"Vista de impresión del documento",
icon=QIcon.fromTheme("document-preview", QIcon(":/icons/res/document-preview.png")),
shortcut="Ctrl+p",
slot=self.preview,
)
self.actionPrint = self.createAction(
text="Imprimir",
tip="Imprimir el documento",
icon=QIcon.fromTheme("document-print", QIcon(":/icons/res/document-print.png")),
slot=self.printDocument,
)
self.actionSave = self.createAction(
text="Guardar",
tip="Guardar el documento",
icon=QIcon.fromTheme("document-save", QIcon(":/icons/res/document-save.png")),
shortcut="Ctrl+g",
slot=self.save,
)
self.actionCancel = self.createAction(
text="Cancelar",
tip=u"Cancelar la creación del nuevo documento",
icon=QIcon.fromTheme("dialog-cancel", QIcon(":/icons/res/dialog-cancel.png")),
shortcut="Esc",
slot=self.cancel,
)
# edicion, TODO: QUE FUNCIONEN ESTAS ACCIONES
self.actionCopy = self.createAction(
text="Copiar", icon=QIcon.fromTheme("edit-copy", QIcon(":/icons/res/edit-copy.png")), shortcut="Ctrl+c"
)
self.actionCut = self.createAction(
text="Cortar", icon=QIcon.fromTheme("edit-cut", QIcon(":/icons/res/edit-cut.png")), shortcut="Ctrl+x"
)
self.actionPaste = self.createAction(text="Pegar", icon=":/icons/res/edit-paste.png", shortcut="Ctrl+v")
# navegación
self.actionGoFirst = self.createAction(
text="Primer documento",
tip="Ir al primer documento",
icon=QIcon.fromTheme("go-first", QIcon(":/icons/res/go-first.png")),
slot=functools.partial(self.navigate, "first"),
)
self.actionGoPrevious = self.createAction(
text="Documento anterior",
tip="Ir al documento anterior",
icon=QIcon.fromTheme("go-previous", QIcon(":/icons/res/go-previous.png")),
slot=functools.partial(self.navigate, "previous"),
)
self.actionGoLast = self.createAction(
text="Ultimo documento",
tip="Ir al ultimo documento",
icon=QIcon.fromTheme("go-last", QIcon(":/icons/res/go-last.png")),
slot=functools.partial(self.navigate, "last"),
)
self.actionGoNext = self.createAction(
text="Documento siguiente",
tip="Ir al siguiente documento",
icon=QIcon.fromTheme("go-next", QIcon(":/icons/res/go-next.png")),
slot=functools.partial(self.navigate, "next"),
)
self.actionDeleteRow = self.createAction(
text="Borrar la fila",
icon=QIcon.fromTheme("edit-delete", QIcon(":/icons/res/edit-delete.png")),
slot=self.deleteRow,
)
self.addActionsToToolBar()
def addActionsToToolBar(self):
"""
Añade las acciones predefinidas a la barra de tareas
"""
self.toolBar.addActions(
[self.actionNew, self.actionPreview, self.actionPrint, self.actionSave, self.actionCancel]
)
self.toolBar.addSeparator()
self.toolBar.addActions(
[self.actionGoFirst, self.actionGoPrevious, self.actionGoLast, self.actionGoNext, self.actionGoLast]
)
@pyqtSlot()
@if_edit_model
def on_txtObservations_textChanged(self):
"""
Asignar las observaciones al editmodel
"""
self.editmodel.observations = self.txtObservations.toPlainText().strip()
def save(self, imgName, w=None, h=None, header=None, \
dpi=150, take_region=False):
ext = imgName.split(".")[-1].upper()
root = self.startNode
#aspect_ratio = root.fullRegion.height() / root.fullRegion.width()
aspect_ratio = self.i_height / self.i_width
# auto adjust size
if w is None and h is None and (ext == "PDF" or ext == "PS"):
w = dpi * 6.4
h = w * aspect_ratio
if h>dpi * 11:
h = dpi * 11
w = h / aspect_ratio
elif w is None and h is None:
w = self.i_width
h = self.i_height
elif h is None :
h = w * aspect_ratio
elif w is None:
w = h / aspect_ratio
if ext == "SVG":
svg = QtSvg.QSvgGenerator()
svg.setFileName(imgName)
svg.setSize(QtCore.QSize(w, h))
svg.setViewBox(QtCore.QRect(0, 0, w, h))
#svg.setTitle("SVG Generator Example Drawing")
#svg.setDescription("An SVG drawing created by the SVG Generator")
pp = QtGui.QPainter()
pp.begin(svg)
targetRect = QtCore.QRectF(0, 0, w, h)
self.render(pp, targetRect, self.sceneRect())
pp.end()
elif ext == "PDF" or ext == "PS":
format = QPrinter.PostScriptFormat if ext == "PS" else QPrinter.PdfFormat
printer = QPrinter(QPrinter.HighResolution)
printer.setResolution(dpi)
printer.setOutputFormat(format)
printer.setPageSize(QPrinter.A4)
pageTopLeft = printer.pageRect().topLeft()
paperTopLeft = printer.paperRect().topLeft()
# For PS -> problems with margins
# print paperTopLeft.x(), paperTopLeft.y()
# print pageTopLeft.x(), pageTopLeft.y()
# print printer.paperRect().height(), printer.pageRect().height()
topleft = pageTopLeft - paperTopLeft
printer.setFullPage(True);
printer.setOutputFileName(imgName);
pp = QtGui.QPainter(printer)
if header:
pp.setFont(QtGui.QFont("Verdana",12))
pp.drawText(topleft.x(),20, header)
targetRect = QtCore.QRectF(topleft.x(), 20 + (topleft.y()*2), w, h)
else:
targetRect = QtCore.QRectF(topleft.x(), topleft.y()*2, w, h)
if take_region:
self.selector.setVisible(False)
self.render(pp, targetRect, self.selector.rect())
self.selector.setVisible(True)
else:
self.render(pp, targetRect, self.sceneRect())
pp.end()
return
else:
targetRect = QtCore.QRectF(0, 0, w, h)
ii= QtGui.QImage(w, \
h, \
QtGui.QImage.Format_ARGB32)
pp = QtGui.QPainter(ii)
pp.setRenderHint(QtGui.QPainter.Antialiasing )
pp.setRenderHint(QtGui.QPainter.TextAntialiasing)
pp.setRenderHint(QtGui.QPainter.SmoothPixmapTransform)
if take_region:
self.selector.setVisible(False)
self.render(pp, targetRect, self.selector.rect())
self.selector.setVisible(True)
else:
self.render(pp, targetRect, self.sceneRect())
pp.end()
ii.save(imgName)
class WebPageToPdf(QObject):
threadPdfStatusBar = QtCore.pyqtSignal(object)
threadPdfWritingStatus = QtCore.pyqtSignal(object)
threadPdfWritingDone = QtCore.pyqtSignal(int)
def __init__(self):
QObject.__init__(self)
self.regex = Regex()
self.title = ''
self.webView = QWebView()
self.webView.settings().setAttribute(QWebSettings.AutoLoadImages, True)
self.webView.settings().setAttribute(QWebSettings.JavascriptEnabled, True)
self.webView.settings().setAttribute(QWebSettings.PluginsEnabled, True)
self.webView.settings().setAttribute(QWebSettings.DeveloperExtrasEnabled, True)
self.pdfPrinter = QPrinter()
self.webView.loadFinished.connect(self.convertToPdf)
def setupDefaultPdfPrinter(self, fileName):
self.pdfPrinter.setOrientation(QPrinter.Portrait)
self.pdfPrinter.setPageSize(QPrinter.A4)
self.pdfPrinter.setOutputFormat(QPrinter.PdfFormat)
self.pdfPrinter.setOutputFileName(fileName)
def printWebHtmlToPdf(self, url, filePath, fileName, groupType):
self.tempPdfFile = filePath + 'out.pdf'
self.filePath = filePath
self.fileName = fileName
self.url = url
self.groupType = groupType
self.setupDefaultPdfPrinter(self.tempPdfFile)
self.threadPdfStatusBar.emit('Fetching Data From Web. Please Wait...')
# self.threadPdfWritingStatus.emit(
# '<font size=4 color=green><b>Method "%s": </b></font><font color=green><b>Fetching Data From Web for</b> %s<b>.<br />Please Wait...</b></font>' % (
# self.groupType, self.url))
self.threadPdfWritingStatus.emit(
'<font color=green><b>Fetching Data From Web for</b> %s<b>.<br />Please Wait...</b></font>' % self.url)
self.webView.load(QUrl(url))
self.title = self.webView.title()
def convertToPdf(self):
print 'Generating Pdf'
# self.threadPdfWritingStatus.emit(
# '<font size=4><b>Method "%s": </b></font><b>Generating Pdf for</b> %s<b>. Please Wait...</b>' % (
# self.groupType, self.url))
self.threadPdfWritingStatus.emit(
'<b>Generating Pdf for</b> %s<b>. Please Wait...</b>' % self.url)
self.threadPdfStatusBar.emit('Generating Pdf. Please Wait...')
self.webView.print_(self.pdfPrinter)
print 'Generated Pdf'
# self.threadPdfWritingStatus.emit(
# '<font size=4><b>Method "%s": </b></font><b>Generated Pdf for</b> %s<b>. Please Wait...</b>' % (
# self.groupType, self.url))
self.threadPdfWritingStatus.emit(
'<b>Generated Pdf for</b> %s<b>. Please Wait...</b>' % self.url)
self.threadPdfStatusBar.emit('Generated Pdf.')
self.mergePdf()
self.threadPdfWritingDone.emit(True)
def mergePdf(self):
# self.threadPdfWritingStatus.emit(
# '<font size=4><b>Method "%s": </b></font><b>Setting Title for</b> %s<b>. Please Wait...</b><br />' % (
# self.groupType, self.url))
self.threadPdfWritingStatus.emit(
'<b>Setting Title for</b> %s<b>. Please Wait...</b><br />' % self.url)
packet = StringIO()
# create a new PDF with Reportlab
pdfCanvas = canvas.Canvas(packet, pagesize=A4)
pdfCanvas.setFont('Helvetica', 8)
if len(self.title) is 0:
self.title = str(self.url).split('/')[-1]
self.title = self.regex.getSearchedData('(?i)([a-zA-Z0-9-_ ]*?)\.[a-zA-Z0-9_]*$', self.title)
self.title = self.regex.replaceData('(?i)_', ' ', self.title)
title = unicode(self.title[:57] + '...') if (len(self.title) > 60) else unicode(self.title)
url = self.url[:57] + '...' if (len(self.title) > 60) else self.url
pdfCanvas.drawString(5, 830, title + ' ' + str(url).lower())
d = datetime.datetime.now()
strDate = str(d.strftime("%Y-%m-%d %H-%M-%S %p"))
pdfCanvas.drawString(420, 5, 'Created Date Time: ' + strDate)
pdfCanvas.save()
packet.seek(0)
newPdf = PdfFileReader(packet)
if not os.path.exists(self.tempPdfFile):
return self.printWebHtmlToPdf(self.url, self.filePath, self.fileName)
writer = PdfFileWriter()
tmpPdfFile = file(self.tempPdfFile, 'rb')
reader = PdfFileReader(tmpPdfFile)
for i in range(0, (reader.getNumPages())):
page = reader.getPage(i)
page.mergePage(newPdf.getPage(0))
# page = newPdf.getPage(0)
# page.mergePage(reader.getPage(i))
writer.addPage(page)
print 'Filename: ' + self.fileName
outputStream = file(self.filePath + self.fileName, "wb")
writer.write(outputStream)
outputStream.close()
tmpPdfFile.close()
os.remove(str(self.tempPdfFile))
def filePrintPdf(self, fn):
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fn)
printer.setOutputFormat(QPrinter.PdfFormat)
self.document().print_(printer)
class Level(QFrame):
def __init__(self, parent):
QFrame.__init__(self,parent)
self.filename = QString()
self.copiedItem = QByteArray()
self.pasteOffset = 5
self.prevPoint = QPoint()
self.addOffset = 5
self.screenSize = (320, 240)
self.bgColor = QColor(244,244,244)
'''0.Portrait 1.Landscape'''
self.orientation = 0
self.currentItem = None
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.Letter)
'''1.Header'''
self.levelBar = LevelBar(self)
'''3.Tiler'''
self.tiler = TileMapGrid(self)#Tiler(self)
self.tiler.setMinimumHeight(100)
#self.tiler.currentChanged.connect(self.closeDesigner)
#self.tiler.setTabsClosable(True)
#self.tiler.setTabShape(0)
#self.tiler.hide()
#self.levelLayout.addWidget(self.levelBar)
'''2.view'''
viewLayoutWidget = QFrame()
viewLayoutWidget.setFrameShape(QFrame.StyledPanel)
viewLayout = QHBoxLayout(viewLayoutWidget)
#viewLayout.setMargin(10)
self.view = LevelView(viewLayoutWidget)
'''scene'''
self.scene = QGraphicsScene(self)
#self.scene.selectionChanged.connect(self.setConnect)
#self.view.setStyleSheet("border: 1px solid red;")
self.setBackgroundColor(self.bgColor)
self.setScreenSize(self.screenSize)
self.view.setScene(self.scene)
self.view.setAlignment(Qt.AlignCenter)
self.scroll_off = 1
self.setScrollBar()
viewLayout.setMargin(0)
viewLayout.addWidget(self.view)
self.wrapped = [] # Needed to keep wrappers alive
layout = QVBoxLayout(self)
layout.addWidget(self.levelBar)
layout.addWidget(viewLayoutWidget)
layout.addWidget(self.tiler)
layout.setMargin(0)
self.setLayout(layout)
def setScrollBar(self):
if(self.scroll_off):
self.view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.scroll_off = 0
else:
self.view.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)
self.view.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
self.scroll_off = 1
def setBackgroundColor(self,color):
self.bgColor = color
self.scene.setBackgroundBrush(QBrush(color))
def setScreenSize(self,size):
self.screenSize = size
self.setOrientation(self.orientation)
def setOrientation(self,idx):
self.orientation = idx
if(idx == 0):
self.view.setMaximumSize(self.screenSize[1], self.screenSize[0])
self.scene.setSceneRect(0, 0, self.screenSize[1], self.screenSize[0])
else:
self.view.setMaximumSize(self.screenSize[0], self.screenSize[1])
self.scene.setSceneRect(0, 0, self.screenSize[0], self.screenSize[1])
def offerSave(self):
if (Dirty and QMessageBox.question(self,
"Designer - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes|QMessageBox.No) ==
QMessageBox.Yes):
self.save()
def position(self):
point = self.mapFromGlobal(QCursor.pos())
if not self.view.geometry().contains(point):
coord = random.randint(36, 144)
point = QPoint(coord, coord)
else:
if point == self.prevPoint:
point += QPoint(self.addOffset, self.addOffset)
self.addOffset += 5
else:
self.addOffset = 5
self.prevPoint = point
return self.view.mapToScene(point)
def selectedItem(self):
items = self.scene.selectedItems()
if len(items) == 1:
return items[0]
return None
def current(self,item):
self.scene.clearSelection()
sceneItems = self.scene.items()
for items in sceneItems:
if(items != item):
item.setSelected(False)
if(item.isConnected()):
self.propertyBar.disconnectText(item)
item.setConnected(False)
self.currentItem = item
self.currentItem.setConnected(True)
self.currentItem.setSelected(True)
self.propertyBar.connectText(self.currentItem)
self.propertyBar.initText(self.currentItem)
def addText(self):
item = TextItem("SomeText", self.position())
self.connect(item, SIGNAL("current"),self.current)
self.connect(item, SIGNAL("copy"),self.copy)
self.connect(item, SIGNAL("cut"),self.cut)
self.connect(item, SIGNAL("paste"),self.paste)
self.connect(item, SIGNAL("delete"),self.delete)
#self.current(item)
self.scene.addItem(item)
def copy(self):
item = self.selectedItem()
if item is None:
return
self.copiedItem.clear()
self.pasteOffset = 5
stream = QDataStream(self.copiedItem, QIODevice.WriteOnly)
self.writeItemToStream(stream, item)
def cut(self):
item = self.selectedItem()
if item is None:
return
self.copy()
self.scene.removeItem(item)
del item
def paste(self):
if self.copiedItem.isEmpty():
return
stream = QDataStream(self.copiedItem, QIODevice.ReadOnly)
item = self.readItemFromStream(stream, self.pasteOffset)
self.pasteOffset += 5
#self.scene.addItem(item)
def delete(self):
items = self.scene.selectedItems()
if (len(items) and QMessageBox.question(self,
"Designer - Delete",
"Delete {0} item{1}?".format(len(items),
"s" if len(items) != 1 else ""),
QMessageBox.Yes|QMessageBox.No) ==
QMessageBox.Yes):
while items:
item = items.pop()
self.scene.removeItem(item)
del item
def readItemFromStream(self, stream, offset=0):
type = QString()
position = QPointF()
matrix = QMatrix()
stream >> type >> position >> matrix
if offset:
position += QPointF(offset, offset)
if type == "Text":
text = QString()
font = QFont()
stream >> text >> font
self.scene.addItem(TextItem(text, position, font, matrix))
elif type == "Box":
rect = QRectF()
stream >> rect
style = Qt.PenStyle(stream.readInt16())
self.scene.addItem(BoxItem(position, style, matrix))
elif type == "Pixmap":
pixmap = QPixmap()
stream >> pixmap
self.scene.addItem(self.createPixmapItem(pixmap, position, matrix))
def writeItemToStream(self, stream, item):
if isinstance(item, QGraphicsTextItem):
stream << QString("Text") << item.pos() \
<< item.matrix() << item.toPlainText() << item.font()
elif isinstance(item, QGraphicsPixmapItem):
stream << QString("Pixmap") << item.pos() \
<< item.matrix() << item.pixmap()
elif isinstance(item, BoxItem):
stream << QString("Box") << item.pos() \
<< item.matrix() << item.rect
stream.writeInt16(item.style)
def rotate(self):
for item in self.scene.selectedItems():
item.rotate(30)
def print_(self):
dialog = QPrintDialog(self.printer)
if dialog.exec_():
painter = QPainter(self.printer)
painter.setRenderHint(QPainter.Antialiasing)
painter.setRenderHint(QPainter.TextAntialiasing)
self.scene.clearSelection()
#self.removeBorders()
self.scene.render(painter)
#self.addBorders()
def open(self):
self.offerSave()
path = (QFileInfo(self.filename).path()
if not self.filename.isEmpty() else ".")
fname = QFileDialog.getOpenFileName(self,
"Page Designer - Open", path,
"Page Designer Files (*.pgd)")
if fname.isEmpty():
return
self.filename = fname
fh = None
try:
fh = QFile(self.filename)
if not fh.open(QIODevice.ReadOnly):
raise IOError, unicode(fh.errorString())
items = self.scene.items()
while items:
item = items.pop()
self.scene.removeItem(item)
del item
self.addBorders()
stream = QDataStream(fh)
stream.setVersion(QDataStream.Qt_4_2)
magic = stream.readInt32()
if magic != MagicNumber:
raise IOError, "not a valid .pgd file"
fileVersion = stream.readInt16()
if fileVersion != FileVersion:
raise IOError, "unrecognised .pgd file version"
while not fh.atEnd():
self.readItemFromStream(stream)
except IOError, e:
QMessageBox.warning(self, "Page Designer -- Open Error",
"Failed to open {0}: {1}".format(self.filename, e))
finally:
class Window(QMainWindow):
def __init__(self, parent=None):
super(Window, self).__init__(parent)
self.image = QImage()
self.dirty = False
self.filename = None
self.dbFilename = None
self.mirroredvertically = False
self.mirroredhorizontally = False
self.printer = None
self.exampleForm= None
self.matlPopup= None
self.matlPropsHere= None
self.create_widgets()
self.create_actions()
self.load_settings()
self.setWindowTitle("Thermapythia")
self.updateFileMenu()
self.openStackupDB()
QTimer.singleShot(0, self.loadInitialFile)
def create_widgets(self):
self.imageLabel = QLabel()
self.imageLabel.setMinimumSize(200, 200)
self.imageLabel.setAlignment(Qt.AlignCenter)
self.imageLabel.setContextMenuPolicy(Qt.ActionsContextMenu)
self.setCentralWidget(self.imageLabel)
logDockWidget = QDockWidget("Log", self)
logDockWidget.setObjectName("LogDockWidget")
logDockWidget.setAllowedAreas(Qt.LeftDockWidgetArea|
Qt.RightDockWidgetArea)
self.listWidget = QListWidget()
logDockWidget.setWidget(self.listWidget)
self.addDockWidget(Qt.RightDockWidgetArea, logDockWidget)
self.sizeLabel = QLabel()
self.sizeLabel.setFrameStyle(QFrame.StyledPanel|QFrame.Sunken)
status = self.statusBar()
status.setSizeGripEnabled(False)
status.addPermanentWidget(self.sizeLabel)
status.showMessage("Ready", 5000)
def create_actions(self):
fileNewAction = self.createAction("&New...", self.fileNew,
QKeySequence.New, "filenew", "Create an image file")
fileOpenAction = self.createAction("&Open...", self.fileOpen,
QKeySequence.Open, "fileopen",
"Open an existing image file")
fileSaveAction = self.createAction("&Save", self.fileSave,
QKeySequence.Save, "filesave", "Save the image")
fileSaveAsAction = self.createAction("Save &As...",
self.fileSaveAs, icon="filesaveas",
tip="Save the image using a new name")
filePrintAction = self.createAction("&Print", self.filePrint,
QKeySequence.Print, "fileprint", "Print the image")
fileQuitAction = self.createAction("&Quit", self.close,
"Ctrl+Q", "filequit", "Close the application")
# createAction(String, slot, shortcut, icon, tip, checkable
matlEditAction = self.createAction("&Materials", self.matlEdit,
None, "matls", "Edit materials")
layerEditAction= self.createAction("&Layers", self.layerEdit,
None, "layers", "Edit layers")
viaEditAction= self.createAction("&Vias", self.viaEdit,
None, "vias", "Edit vias")
zzzEditAction= self.createAction("&Zzz", self.zzzEdit,
None, "zzz", "Edit zzz")
editInvertAction = self.createAction("&Invert", None, "Ctrl+I",
"editinvert", "Invert the image's colors", True)
editInvertAction.toggled.connect(self.editInvert)
editSwapRedAndBlueAction = self.createAction("Sw&ap Red and Blue",
None, "Ctrl+A", "editswap",
"Swap the image's red and blue color components", True)
editSwapRedAndBlueAction.toggled.connect(self.editSwapRedAndBlue)
editZoomAction = self.createAction("&Zoom...", self.editZoom,
"Alt+Z", "editzoom", "Zoom the image")
mirrorGroup = QActionGroup(self)
editUnMirrorAction = self.createAction("&Unmirror", None, "Ctrl+U",
"editunmirror", "Unmirror the image", True)
editUnMirrorAction.toggled.connect(self.editUnMirror)
mirrorGroup.addAction(editUnMirrorAction)
editMirrorHorizontalAction = self.createAction(
"Mirror &Horizontally", None, "Ctrl+H", "editmirrorhoriz",
"Horizontally mirror the image", True)
editMirrorHorizontalAction.toggled.connect(
self.editMirrorHorizontal)
mirrorGroup.addAction(editMirrorHorizontalAction)
editMirrorVerticalAction = self.createAction("Mirror &Vertically",
None, "Ctrl+V", "editmirrorvert",
"Vertically mirror the image", True)
editMirrorVerticalAction.toggled.connect(self.editMirrorVertical)
mirrorGroup.addAction(editMirrorVerticalAction)
editUnMirrorAction.setChecked(True)
helpAboutAction = self.createAction("&About Thermapythia",
self.helpAbout)
helpHelpAction = self.createAction("&Help", self.helpHelp,
QKeySequence.HelpContents)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenuActions = (fileNewAction, fileOpenAction,
fileSaveAction, fileSaveAsAction, None, filePrintAction,
fileQuitAction)
self.fileMenu.aboutToShow.connect(self.updateFileMenu)
stackupMenu = self.menuBar().addMenu("&Stackup")
self.addActions(stackupMenu, (matlEditAction, layerEditAction, viaEditAction))
editMenu = self.menuBar().addMenu("&Edit")
self.addActions(editMenu, (editInvertAction,
editSwapRedAndBlueAction, editZoomAction))
mirrorMenu = editMenu.addMenu(QIcon(":/editmirror.png"),
"&Mirror")
self.addActions(mirrorMenu, (editUnMirrorAction,
editMirrorHorizontalAction, editMirrorVerticalAction))
helpMenu = self.menuBar().addMenu("&Help")
self.addActions(helpMenu, (helpAboutAction, helpHelpAction))
fileToolbar = self.addToolBar("File")
fileToolbar.setObjectName("FileToolBar")
self.addActions(fileToolbar, (fileNewAction, fileOpenAction,
fileSaveAsAction))
stackupToolbar = self.addToolBar("Stackup")
stackupToolbar.setObjectName("StackupToolBar")
self.addActions(stackupToolbar, (matlEditAction, layerEditAction, viaEditAction))
editToolbar = self.addToolBar("Edit")
editToolbar.setObjectName("EditToolBar")
self.addActions(editToolbar, (editInvertAction,
editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.zoomSpinBox = QSpinBox()
self.zoomSpinBox.setRange(1, 400)
self.zoomSpinBox.setSuffix(" %")
self.zoomSpinBox.setValue(100)
self.zoomSpinBox.setToolTip("Zoom the image")
self.zoomSpinBox.setStatusTip(self.zoomSpinBox.toolTip())
self.zoomSpinBox.setFocusPolicy(Qt.NoFocus)
self.zoomSpinBox.valueChanged.connect(self.showImage)
editToolbar.addWidget(self.zoomSpinBox)
self.addActions(self.imageLabel, (editInvertAction,
editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.resetableActions = ((editInvertAction, False),
(editSwapRedAndBlueAction, False),
(editUnMirrorAction, True))
def load_settings(self):
settings = QSettings()
self.recentFiles = settings.value("RecentFiles").toStringList()
self.restoreGeometry(
settings.value("MainWindow/Geometry").toByteArray())
self.restoreState(settings.value("MainWindow/State").toByteArray())
def createAction(self, text, slot=None, shortcut=None, icon=None,
tip=None, checkable=False):
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(":/{0}.png".format(icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
action.triggered.connect(slot)
if checkable:
action.setCheckable(True)
return action
def addActions(self, target, actions):
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def closeEvent(self, event):
if self.okToContinue():
settings = QSettings()
filename = (QVariant(QString(self.filename))
if self.filename is not None else QVariant())
settings.setValue("LastFile", filename)
recentFiles = (QVariant(self.recentFiles)
if self.recentFiles else QVariant())
settings.setValue("RecentFiles", recentFiles)
settings.setValue("MainWindow/Geometry", QVariant(
self.saveGeometry()))
settings.setValue("MainWindow/State", QVariant(
self.saveState()))
else:
event.ignore()
def okToContinue(self):
if self.dirty:
reply = QMessageBox.question(self,
"Thermapythia - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes|QMessageBox.No|QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
return self.fileSave()
return True
def loadInitialFile(self):
settings = QSettings()
fname = unicode(settings.value("LastFile").toString())
if fname and QFile.exists(fname):
self.loadFile(fname)
def updateStatus(self, message):
self.statusBar().showMessage(message, 5000)
self.listWidget.addItem(message)
if self.filename is not None:
self.setWindowTitle("Thermapythia - {0}[*]".format(
os.path.basename(self.filename)))
elif not self.image.isNull():
self.setWindowTitle("Thermapythia - Unnamed[*]")
else:
self.setWindowTitle("Thermapythia[*]")
self.setWindowModified(self.dirty)
def updateFileMenu(self):
self.fileMenu.clear()
self.addActions(self.fileMenu, self.fileMenuActions[:-1])
current = (QString(self.filename)
if self.filename is not None else None)
recentFiles = []
for fname in self.recentFiles:
if fname != current and QFile.exists(fname):
recentFiles.append(fname)
if recentFiles:
self.fileMenu.addSeparator()
for i, fname in enumerate(recentFiles):
action = QAction(QIcon(":/icon.png"),
"&{0} {1}".format(i + 1, QFileInfo(
fname).fileName()), self)
action.setData(QVariant(fname))
action.triggered.connect(self.loadFile)
self.fileMenu.addAction(action)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.fileMenuActions[-1])
def fileNew(self):
if not self.okToContinue():
return
dialog = newimagedlg.NewImageDlg(self)
if dialog.exec_():
self.addRecentFile(self.filename)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = dialog.image()
self.filename = None
self.dirty = True
self.showImage()
self.sizeLabel.setText("{0} x {1}".format(self.image.width(),
self.image.height()))
self.updateStatus("Created new image")
def fileOpen(self):
if not self.okToContinue():
return
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageReader.supportedImageFormats()])
fname = unicode(QFileDialog.getOpenFileName(self,
"Thermapythia - Choose Image", dir,
"Image files ({0})".format(" ".join(formats))))
if fname:
self.loadFile(fname)
def loadFile(self, fname=None):
if fname is None:
action = self.sender()
if isinstance(action, QAction):
fname = unicode(action.data().toString())
if not self.okToContinue():
return
else:
return
if fname:
self.filename = None
image = QImage(fname)
if image.isNull():
message = "Failed to read {0}".format(fname)
else:
self.addRecentFile(fname)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = image
self.filename = fname
self.showImage()
self.dirty = False
self.sizeLabel.setText("{0} x {1}".format(
image.width(), image.height()))
message = "Loaded {0}".format(os.path.basename(fname))
self.updateStatus(message)
def addRecentFile(self, fname):
if fname is None:
return
if not self.recentFiles.contains(fname):
self.recentFiles.prepend(QString(fname))
while self.recentFiles.count() > 9:
self.recentFiles.takeLast()
def fileSave(self):
if self.image.isNull():
return True
if self.filename is None:
return self.fileSaveAs()
else:
if self.image.save(self.filename, None):
self.updateStatus("Saved as {0}".format(self.filename))
self.dirty = False
return True
else:
self.updateStatus("Failed to save {0}".format(
self.filename))
return False
def fileSaveAs(self):
if self.image.isNull():
return True
fname = self.filename if self.filename is not None else "."
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageWriter.supportedImageFormats()])
fname = unicode(QFileDialog.getSaveFileName(self,
"Thermapythia - Save Image", fname,
"Image files ({0})".format(" ".join(formats))))
if fname:
if "." not in fname:
fname += ".png"
self.addRecentFile(fname)
self.filename = fname
return self.fileSave()
return False
def filePrint(self):
if self.image.isNull():
return
if self.printer is None:
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.Letter)
form = QPrintDialog(self.printer, self)
if form.exec_():
painter = QPainter(self.printer)
rect = painter.viewport()
size = self.image.size()
size.scale(rect.size(), Qt.KeepAspectRatio)
painter.setViewport(rect.x(), rect.y(), size.width(),
size.height())
painter.drawImage(0, 0, self.image)
#
def openStackupDB(self):
if self.dbFilename != None:
return
self.dbFilename = os.path.join(os.path.dirname(__file__), "stackup.db")
create = not QFile.exists(self.dbFilename)
db = QSqlDatabase.addDatabase("QSQLITE")
db.setDatabaseName(self.dbFilename)
if not db.open():
QMessageBox.warning(None, "Can't open stackup database",
QString("Database Error: %1").arg(db.lastError().text()))
sys.exit(1)
if create == False:
self.updateStatus("Found stackup database file: " + str(self.dbFilename))
self.createDatabaseTables()
with open('test2.js', "r") as jsonHandle:
jsonContents= jsonHandle.read()
config= yaml.load(jsonContents)
matlCount= self.countDatabaseTableRows("matl")
if matlCount == 0:
self.updateStatus("material editor call goes here.")
#
# TODO: Put the guess-what-I'm thinking Qt table editor thing here.
#
# self.loadDefaultMatls()
matls = Matls.Matls(config['matl_config'])
self.loadDefaultData('matl', matls)
viaCount= self.countDatabaseTableRows("via")
if viaCount == 0:
vias = Vias.Vias(config['via_config'])
self.loadDefaultData('via', vias)
layerCount= self.countDatabaseTableRows("layer")
if layerCount == 0:
layers = Layers.Layers(config['layer_config'])
self.loadDefaultData('layer', layers)
# initialize zzz new content here, perhaps
def createDatabaseTables(self):
query = QSqlQuery()
self.updateStatus("Creating database tables")
# "id" INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,
statusMatl = query.exec_("""CREATE TABLE IF NOT EXISTS "matl" (
"name" TEXT,
"type" TEXT,
"density" real,
"color" TEXT,
"specific_heat" real,
"conductivity" real,
"conductivityXX" real,
"conductivityYY" real,
"conductivityZZ" real,
"reflection_coeff" real,
"emissivity" real,
"max_height" real,
"thickness" real
);""")
if statusMatl == False:
self.updateStatus("Could not create material property database table 'matl'")
self.updateStatus("Database error message: " + query.lastError().text())
statusPhylayer = query.exec_("""CREATE TABLE IF NOT EXISTS "layer" (
"id" INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,
"name" TEXT,
"matl" TEXT,
"type" TEXT,
"thickness" real,
"displaces" TEXT,
"coverage" real,
"z_bottom" real,
"z_top" real,
"adheres_to" TEXT);""")
if statusPhylayer == False:
self.updateStatus("Could not create material property database table 'layer'")
self.updateStatus("Database error message: " + query.lastError().text())
statusVia = query.exec_("""CREATE TABLE IF NOT EXISTS "via" (
"id" INTEGER PRIMARY KEY AUTOINCREMENT UNIQUE NOT NULL,
"name" TEXT,
"matl" TEXT,
"to" TEXT,
"from" TEXT);""")
if statusVia == False:
self.updateStatus("Could not create material property database table 'via'")
self.updateStatus("Database error message: " + query.lastError().text())
def countDatabaseTableRows(self, tablename):
query = QSqlQuery()
self.updateStatus("Loading database tables")
query.prepare("select count(*) from " + tablename + ";")
status = query.exec_()
if status == False:
self.updateStatus("Could not count entries in table '" + tablename + "'")
self.updateStatus("Database error message: " + query.lastError().text())
query.first()
count, ok = query.value(0).toInt()
# self.updateStatus("There are " + str(count) + " rows in the " + tablename + " table")
return count
def loadDefaultData(self, table, data):
query = QSqlQuery()
insertList= "', '".join(data.tableCols)
insertList= "'" + insertList + "'"
colCount= len(data.tableCols)
quest= ""
for i in range(0, colCount):
quest = quest + "?,"
quest= quest[:-1]
sql= "INSERT INTO " + table + "(" + insertList + ") VALUES (" + quest + ");"
# self.updateStatus(sql)
status= query.prepare(sql)
if status == False:
self.updateStatus("Could not prepare material property database table " + table)
self.updateStatus("Database error message: " + query.lastError().text())
for row in data.propDict:
for prop in data.tableCols:
propval= data.propDict[row][prop]
if data.tableColSQLTypes[prop] == 'TEXT':
query.addBindValue(QVariant(QString(propval)))
# self.updateStatus("Setting TEXT property " + prop + " to value " + str(propval) + " in row " + str(row))
elif data.tableColSQLTypes[prop] == 'real':
if (propval == '-'):
propreal= -999.9
else:
propreal= float(propval)
query.addBindValue(QVariant(propreal))
# self.updateStatus("Setting real property " + prop + " to value " + str(propreal) + " in row " + str(row))
status= query.exec_()
if status == False:
self.updateStatus("Could not load property database table " + table + " with " + str(row))
self.updateStatus("Database error message: " + query.lastError().text())
def matlEdit(self):
self.dirty= True
self.updateStatus("Call edit material code here")
# This ends up in a little child window
self.matlPopup= QLabel("<font color=green size=72><b> Material Editor </b></font>")
# self.matlPopup.setWindowFlags(Qt.SplashScreen)
self.matlPopup.setWindowTitle("Material edit popup")
self.matlPropsHere= QLabel("<font color=red size=24><b> Data goes here </b></font>")
# self.matlPopup.setWindowFlags(Qt.SplashScreen)
if 1 == 0:
dataLayout= QVBoxLayout()
dataLayout.addWidget(self.matlPopup)
dataLayout.addWidget(self.matlPropsHere)
self.setLayout(dataLayout)
self.matlPopup.show()
self.matlPropsHere.show()
# self.exampleForm= MatlForm()
# First need a layout manager, and add the widget to that.
# Then attach the layout manager to a window.
# addWidget(self.exampleForm)
# self.exampleForm.show()
QTimer.singleShot(5000, self.update)
self.updateStatus("Created stackup db")
def layerEdit(self):
self.dirty= True
self.updateStatus("Call edit layer code here")
# This way of doing it is modal - window doesn't go away and covers up other windows until it is dismissed.
form= MatlForm()
form.exec_()
def viaEdit(self):
self.dirty= True
self.updateStatus("Call edit via code here")
def zzzEdit(self):
self.dirty= True
self.updateStatus("Call code to be triggered by zzz button here")
def editInvert(self, on):
if self.image.isNull():
return
self.image.invertPixels()
self.showImage()
self.dirty = True
self.updateStatus("Inverted" if on else "Uninverted")
def editSwapRedAndBlue(self, on):
if self.image.isNull():
return
self.image = self.image.rgbSwapped()
self.showImage()
self.dirty = True
self.updateStatus(("Swapped Red and Blue"
if on else "Unswapped Red and Blue"))
def editUnMirror(self, on):
if self.image.isNull():
return
if self.mirroredhorizontally:
self.editMirrorHorizontal(False)
if self.mirroredvertically:
self.editMirrorVertical(False)
def editMirrorHorizontal(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(True, False)
self.showImage()
self.mirroredhorizontally = not self.mirroredhorizontally
self.dirty = True
self.updateStatus(("Mirrored Horizontally"
if on else "Unmirrored Horizontally"))
def editMirrorVertical(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(False, True)
self.showImage()
self.mirroredvertically = not self.mirroredvertically
self.dirty = True
self.updateStatus(("Mirrored Vertically"
if on else "Unmirrored Vertically"))
def editZoom(self):
if self.image.isNull():
return
percent, ok = QInputDialog.getInteger(self,
"Thermapythia - Zoom", "Percent:",
self.zoomSpinBox.value(), 1, 400)
if ok:
self.zoomSpinBox.setValue(percent)
def showImage(self, percent=None):
if self.image.isNull():
return
if percent is None:
percent = self.zoomSpinBox.value()
factor = percent / 100.0
width = self.image.width() * factor
height = self.image.height() * factor
image = self.image.scaled(width, height, Qt.KeepAspectRatio)
self.imageLabel.setPixmap(QPixmap.fromImage(image))
def helpAbout(self):
QMessageBox.about(self, "About Thermapythia",
"""<b>Thermapythia</b> v {0}
<p>Thermal Analyzer in Python predicts circuit board temperature
<p>Python {1} - Qt {2} - PyQt {3} on {4}""".format(
__version__, platform.python_version(),
QT_VERSION_STR, PYQT_VERSION_STR,
platform.system()))
def helpHelp(self):
form = HelpForm.HelpForm("index.html", self)
form.show()
class Ventana_Imprimir(QtGui.QDialog):
def __init__(self, parent=None, datos=None):
self.parent = parent
self.transito = datos
super(Ventana_Imprimir, self).__init__(parent=self.parent)
self.setGeometry(30, 40, 800, 500)
self.setWindowTitle("Imprimir albaran")
self.init_controles()
#self.navegador.setUrl(QUrl('http://www.google.es'))
self.navegador.setHtml(self.leer_plantilla_y_poner_datos())
#self.navegador.loadFinished.connect(lambda: self.LoadEnd())
def leer_plantilla_y_poner_datos(self):
html = open(PLANTILLA_ALBARAN).read()
if self.transito is None:
html = """
<!DOCTYPE html>
<html lang="es">
<head>
<meta charset="UTF-8">
<title></title>
</head>
<body >
<h1 style='color:red;'>NO HAY DATOS QUE PRESENTAR</h1>
</body>
</html>
"""
else:
#pongamos los datos
#self.transito = Historico()
html = html.replace("@num_albaran", str(self.transito.num_albaran))
html = html.replace("@mat_cabina", self.transito.mat_cabina)
html = html.replace("@mat_remolque", self.transito.mat_remolque)
html = html.replace("@cliente", self.transito.razon_social_cliente)
html = html.replace("@producto", self.transito.nombre_producto)
html = html.replace("@proveedor",
self.transito.razon_social_proveedor)
html = html.replace("@poseedor",
self.transito.razon_social_poseedor)
html = html.replace("@agencia", self.transito.razon_social_agencia)
html = html.replace("@conductor", self.transito.nombre_conductor)
html = html.replace("@origen", self.transito.origen)
html = html.replace("@destino", self.transito.destino)
html = html.replace("@fech_entrada", self.transito.fecha_entrada)
html = html.replace("@fech_salida", self.transito.fecha_salida)
html = html.replace("@peso_entrada", str(self.transito.bruto))
html = html.replace("@peso_salida", str(self.transito.tara))
html = html.replace("@neto", str(self.transito.neto))
return html
def printPreview(self):
dialog = QtGui.QPrintPreviewDialog(self.printer)
dialog.setWindowState(Qt.WindowMaximized)
#dialog.paintRequested.connect(self.print_)
dialog.setWindowFlags(Qt.CustomizeWindowHint | Qt.WindowTitleHint
| Qt.WindowMinMaxButtonsHint
| Qt.WindowCloseButtonHint
| Qt.WindowContextHelpButtonHint)
dialog.exec()
def imprimir(self):
print('Prepandando impresion')
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.A4)
self.printer.setOrientation(QPrinter.Portrait)
self.printer.setPageMargins(10, 10, 10, 10, QPrinter.Millimeter)
self.navegador.print_(self.printer)
def LoadEnd(self):
#self.printer = QPrinter( QPrinter.HighResolution )
#self.printer.setOutputFormat( QPrinter.PdfFormat )
#self.printer.setOutputFileName( "out.pdf" )
#self.printPreview()
#self.navegador.print_( self.printer )
pass
def init_controles(self):
vbox = QtGui.QVBoxLayout()
self.navegador = QtWebKit.QWebView(self)
self.btnImprimir = QtGui.QPushButton(IMPRIMIR, self)
self.btnImprimir.clicked.connect(lambda: self.imprimir())
vbox.addWidget(self.btnImprimir)
vbox.addWidget(self.navegador)
self.setLayout(vbox)
class Ventana_Imprimir(QtGui.QDialog):
def __init__(self, parent=None, datos=None):
self.parent = parent
self.transito = datos
super(Ventana_Imprimir, self).__init__(parent=self.parent)
self.setGeometry(30, 40, 800, 500)
self.setWindowTitle("Imprimir albaran")
self.init_controles()
#self.navegador.setUrl(QUrl('http://www.google.es'))
self.navegador.setHtml(self.leer_plantilla_y_poner_datos())
#self.navegador.loadFinished.connect(lambda: self.LoadEnd())
def leer_plantilla_y_poner_datos(self):
html = open(PLANTILLA_ALBARAN).read()
if self.transito is None:
html = """
<!DOCTYPE html>
<html lang="es">
<head>
<meta charset="UTF-8">
<title></title>
</head>
<body >
<h1 style='color:red;'>NO HAY DATOS QUE PRESENTAR</h1>
</body>
</html>
"""
else:
#pongamos los datos
#self.transito = Historico()
html = html.replace("@num_albaran", str(self.transito.num_albaran))
html = html.replace("@mat_cabina", self.transito.mat_cabina)
html = html.replace("@mat_remolque", self.transito.mat_remolque)
html = html.replace("@cliente", self.transito.razon_social_cliente)
html = html.replace("@producto", self.transito.nombre_producto)
html = html.replace("@proveedor", self.transito.razon_social_proveedor)
html = html.replace("@poseedor", self.transito.razon_social_poseedor)
html = html.replace("@agencia", self.transito.razon_social_agencia)
html = html.replace("@conductor", self.transito.nombre_conductor)
html = html.replace("@origen", self.transito.origen)
html = html.replace("@destino", self.transito.destino)
html = html.replace("@fech_entrada", self.transito.fecha_entrada)
html = html.replace("@fech_salida", self.transito.fecha_salida)
html = html.replace("@peso_entrada", str(self.transito.bruto))
html = html.replace("@peso_salida", str(self.transito.tara))
html = html.replace("@neto", str(self.transito.neto))
return html
def printPreview(self):
dialog = QtGui.QPrintPreviewDialog(self.printer)
dialog.setWindowState(Qt.WindowMaximized)
#dialog.paintRequested.connect(self.print_)
dialog.setWindowFlags(Qt.CustomizeWindowHint |
Qt.WindowTitleHint |
Qt.WindowMinMaxButtonsHint |
Qt.WindowCloseButtonHint |
Qt.WindowContextHelpButtonHint)
dialog.exec()
def imprimir(self):
print('Prepandando impresion')
self.printer = QPrinter(QPrinter.HighResolution);
self.printer.setPageSize(QPrinter.A4);
self.printer.setOrientation(QPrinter.Portrait);
self.printer.setPageMargins(10, 10, 10, 10, QPrinter.Millimeter)
self.navegador.print_(self.printer)
def LoadEnd(self):
#self.printer = QPrinter( QPrinter.HighResolution )
#self.printer.setOutputFormat( QPrinter.PdfFormat )
#self.printer.setOutputFileName( "out.pdf" )
#self.printPreview()
#self.navegador.print_( self.printer )
pass
def init_controles(self):
vbox = QtGui.QVBoxLayout()
self.navegador = QtWebKit.QWebView(self)
self.btnImprimir = QtGui.QPushButton(IMPRIMIR, self)
self.btnImprimir.clicked.connect(lambda: self.imprimir())
vbox.addWidget(self.btnImprimir)
vbox.addWidget(self.navegador)
self.setLayout(vbox)
from PyQt4.QtGui import QTextDocument, QPrinter, QApplication
import sys
app = QApplication(sys.argv)
doc = QTextDocument()
location = "https://translate.google.cn/"
html = open(location).read()
doc.setHtml(html)
printer = QPrinter()
printer.setOutputFileName("foo.pdf")
printer.setOutputFormat(QPrinter.PdfFormat)
printer.setPageSize(QPrinter.A4)
printer.setPageMargins (15,15,15,15,QPrinter.Millimeter)
doc.print_(printer)
print ("done!")
from PyQt4.QtGui import QTextDocument, QPrinter, QApplication import sys app = QApplication(sys.argv) doc = QTextDocument() location = sys.argv[1] html = open(location).read() doc.setHtml(html) printer = QPrinter() printer.setOutputFileName(sys.argv[2]) printer.setOutputFormat(QPrinter.PdfFormat) printer.setPageSize(QPrinter.A4) printer.setPageMargins(15,15,15,15,QPrinter.Millimeter); doc.print_(printer) print "\n" + sys.argv[2] + " was created successfully.\n"
class Window(QMainWindow):
def __init__(self, parent=None):
super(Window, self).__init__(parent)
self.image = QImage()
self.dirty = False
self.filename = None
self.mirroredvertically = False
self.mirroredhorizontally = False
self.printer = None
self.create_widgets()
self.create_actions()
self.load_settings()
self.setWindowTitle("Image Changer")
self.updateFileMenu()
QTimer.singleShot(0, self.loadInitialFile)
def create_widgets(self):
self.imageLabel = QLabel()
self.imageLabel.setMinimumSize(200, 200)
self.imageLabel.setAlignment(Qt.AlignCenter)
self.imageLabel.setContextMenuPolicy(Qt.ActionsContextMenu)
self.setCentralWidget(self.imageLabel)
logDockWidget = QDockWidget("Log", self)
logDockWidget.setObjectName("LogDockWidget")
logDockWidget.setAllowedAreas(Qt.LeftDockWidgetArea|
Qt.RightDockWidgetArea)
self.listWidget = QListWidget()
logDockWidget.setWidget(self.listWidget)
self.addDockWidget(Qt.RightDockWidgetArea, logDockWidget)
self.sizeLabel = QLabel()
self.sizeLabel.setFrameStyle(QFrame.StyledPanel|QFrame.Sunken)
status = self.statusBar()
status.setSizeGripEnabled(False)
status.addPermanentWidget(self.sizeLabel)
status.showMessage("Ready", 5000)
def create_actions(self):
fileNewAction = self.createAction("&New...", self.fileNew,
QKeySequence.New, "filenew", "Create an image file")
fileOpenAction = self.createAction("&Open...", self.fileOpen,
QKeySequence.Open, "fileopen",
"Open an existing image file")
fileSaveAction = self.createAction("&Save", self.fileSave,
QKeySequence.Save, "filesave", "Save the image")
fileSaveAsAction = self.createAction("Save &As...",
self.fileSaveAs, icon="filesaveas",
tip="Save the image using a new name")
filePrintAction = self.createAction("&Print", self.filePrint,
QKeySequence.Print, "fileprint", "Print the image")
fileQuitAction = self.createAction("&Quit", self.close,
"Ctrl+Q", "filequit", "Close the application")
editInvertAction = self.createAction("&Invert", None, "Ctrl+I",
"editinvert", "Invert the image's colors", True)
editInvertAction.toggled.connect(self.editInvert)
editSwapRedAndBlueAction = self.createAction("Sw&ap Red and Blue",
None, "Ctrl+A", "editswap",
"Swap the image's red and blue color components", True)
editSwapRedAndBlueAction.toggled.connect(self.editSwapRedAndBlue)
editZoomAction = self.createAction("&Zoom...", self.editZoom,
"Alt+Z", "editzoom", "Zoom the image")
editResizeAction = self.createAction("&Resize...",
self.editResize, "Ctrl+R", "editresize",
"Resize the image")
mirrorGroup = QActionGroup(self)
editUnMirrorAction = self.createAction("&Unmirror", None, "Ctrl+U",
"editunmirror", "Unmirror the image", True)
editUnMirrorAction.toggled.connect(self.editUnMirror)
mirrorGroup.addAction(editUnMirrorAction)
editMirrorHorizontalAction = self.createAction(
"Mirror &Horizontally", None, "Ctrl+H", "editmirrorhoriz",
"Horizontally mirror the image", True)
editMirrorHorizontalAction.toggled.connect(
self.editMirrorHorizontal)
mirrorGroup.addAction(editMirrorHorizontalAction)
editMirrorVerticalAction = self.createAction( "Mirror &Vertically",
None, "Ctrl+V", "editmirrorvert",
"Vertically mirror the image", True)
editMirrorVerticalAction.toggled.connect(self.editMirrorVertical)
mirrorGroup.addAction(editMirrorVerticalAction)
editUnMirrorAction.setChecked(True)
helpAboutAction = self.createAction("&About Image Changer",
self.helpAbout)
helpHelpAction = self.createAction("&Help", self.helpHelp,
QKeySequence.HelpContents)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenuActions = (fileNewAction, fileOpenAction,
fileSaveAction, fileSaveAsAction, None, filePrintAction,
fileQuitAction)
self.fileMenu.aboutToShow.connect(self.updateFileMenu)
editMenu = self.menuBar().addMenu("&Edit")
self.addActions(editMenu, (editInvertAction,
editSwapRedAndBlueAction, editZoomAction,
editResizeAction))
mirrorMenu = editMenu.addMenu(QIcon(":/editmirror.png"),
"&Mirror")
self.addActions(mirrorMenu, (editUnMirrorAction,
editMirrorHorizontalAction, editMirrorVerticalAction))
helpMenu = self.menuBar().addMenu("&Help")
self.addActions(helpMenu, (helpAboutAction, helpHelpAction))
fileToolbar = self.addToolBar("File")
fileToolbar.setObjectName("FileToolBar")
self.addActions(fileToolbar, (fileNewAction, fileOpenAction,
fileSaveAsAction))
editToolbar = self.addToolBar("Edit")
editToolbar.setObjectName("EditToolBar")
self.addActions(editToolbar, (editInvertAction,
editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.zoomSpinBox = QSpinBox()
self.zoomSpinBox.setRange(1, 400)
self.zoomSpinBox.setSuffix(" %")
self.zoomSpinBox.setValue(100)
self.zoomSpinBox.setToolTip("Zoom the image")
self.zoomSpinBox.setStatusTip(self.zoomSpinBox.toolTip())
self.zoomSpinBox.setFocusPolicy(Qt.NoFocus)
self.zoomSpinBox.valueChanged.connect(self.showImage)
editToolbar.addWidget(self.zoomSpinBox)
self.addActions(self.imageLabel, (editInvertAction,
editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.resetableActions = ((editInvertAction, False),
(editSwapRedAndBlueAction, False),
(editUnMirrorAction, True))
def load_settings(self):
settings = QSettings()
self.recentFiles = settings.value("RecentFiles").toStringList()
self.restoreGeometry(
settings.value("MainWindow/Geometry").toByteArray())
self.restoreState(settings.value("MainWindow/State").toByteArray())
def createAction(self, text, slot=None, shortcut=None, icon=None,
tip=None, checkable=False):
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(":/{0}.png".format(icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
action.triggered.connect(slot)
if checkable:
action.setCheckable(True)
return action
def addActions(self, target, actions):
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def closeEvent(self, event):
if self.okToContinue():
settings = QSettings()
filename = (QVariant(QString(self.filename))
if self.filename is not None else QVariant())
settings.setValue("LastFile", filename)
recentFiles = (QVariant(self.recentFiles)
if self.recentFiles else QVariant())
settings.setValue("RecentFiles", recentFiles)
settings.setValue("MainWindow/Geometry", QVariant(
self.saveGeometry()))
settings.setValue("MainWindow/State", QVariant(
self.saveState()))
else:
event.ignore()
def okToContinue(self):
if self.dirty:
reply = QMessageBox.question(self,
"Image Changer - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes|QMessageBox.No|QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
return self.fileSave()
return True
def loadInitialFile(self):
settings = QSettings()
fname = unicode(settings.value("LastFile").toString())
if fname and QFile.exists(fname):
self.loadFile(fname)
def updateStatus(self, message):
self.statusBar().showMessage(message, 5000)
self.listWidget.addItem(message)
if self.filename is not None:
self.setWindowTitle("Image Changer - {0}[*]".format(
os.path.basename(self.filename)))
elif not self.image.isNull():
self.setWindowTitle("Image Changer - Unnamed[*]")
else:
self.setWindowTitle("Image Changer[*]")
self.setWindowModified(self.dirty)
def updateFileMenu(self):
self.fileMenu.clear()
self.addActions(self.fileMenu, self.fileMenuActions[:-1])
current = (QString(self.filename)
if self.filename is not None else None)
recentFiles = []
for fname in self.recentFiles:
if fname != current and QFile.exists(fname):
recentFiles.append(fname)
if recentFiles:
self.fileMenu.addSeparator()
for i, fname in enumerate(recentFiles):
action = QAction(QIcon(":/icon.png"),
"&{0} {1}".format(i + 1, QFileInfo(
fname).fileName()), self)
action.setData(QVariant(fname))
action.triggered.connect(self.loadFile)
self.fileMenu.addAction(action)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.fileMenuActions[-1])
def fileNew(self):
if not self.okToContinue():
return
dialog = newimagedlg.NewImageDlg(self)
if dialog.exec_():
self.addRecentFile(self.filename)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = dialog.image()
self.filename = None
self.dirty = True
self.showImage()
self.sizeLabel.setText("{0} x {1}".format(self.image.width(),
self.image.height()))
self.updateStatus("Created new image")
def fileOpen(self):
if not self.okToContinue():
return
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageReader.supportedImageFormats()])
fname = unicode(QFileDialog.getOpenFileName(self,
"Image Changer - Choose Image", dir,
"Image files ({0})".format(" ".join(formats))))
if fname:
self.loadFile(fname)
def loadFile(self, fname=None):
if fname is None:
action = self.sender()
if isinstance(action, QAction):
fname = unicode(action.data().toString())
if not self.okToContinue():
return
else:
return
if fname:
self.filename = None
image = QImage(fname)
if image.isNull():
message = "Failed to read {0}".format(fname)
else:
self.addRecentFile(fname)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = image
self.filename = fname
self.showImage()
self.dirty = False
self.sizeLabel.setText("{0} x {1}".format(
image.width(), image.height()))
message = "Loaded {0}".format(os.path.basename(fname))
self.updateStatus(message)
def addRecentFile(self, fname):
if fname is None:
return
if not self.recentFiles.contains(fname):
self.recentFiles.prepend(QString(fname))
while self.recentFiles.count() > 9:
self.recentFiles.takeLast()
def fileSave(self):
if self.image.isNull():
return True
if self.filename is None:
return self.fileSaveAs()
else:
if self.image.save(self.filename, None):
self.updateStatus("Saved as {0}".format(self.filename))
self.dirty = False
return True
else:
self.updateStatus("Failed to save {0}".format(
self.filename))
return False
def fileSaveAs(self):
if self.image.isNull():
return True
fname = self.filename if self.filename is not None else "."
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageWriter.supportedImageFormats()])
fname = unicode(QFileDialog.getSaveFileName(self,
"Image Changer - Save Image", fname,
"Image files ({0})".format(" ".join(formats))))
if fname:
if "." not in fname:
fname += ".png"
self.addRecentFile(fname)
self.filename = fname
return self.fileSave()
return False
def filePrint(self):
if self.image.isNull():
return
if self.printer is None:
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.Letter)
form = QPrintDialog(self.printer, self)
if form.exec_():
painter = QPainter(self.printer)
rect = painter.viewport()
size = self.image.size()
size.scale(rect.size(), Qt.KeepAspectRatio)
painter.setViewport(rect.x(), rect.y(), size.width(),
size.height())
painter.drawImage(0, 0, self.image)
def editInvert(self, on):
if self.image.isNull():
return
self.image.invertPixels()
self.showImage()
self.dirty = True
self.updateStatus("Inverted" if on else "Uninverted")
def editSwapRedAndBlue(self, on):
if self.image.isNull():
return
self.image = self.image.rgbSwapped()
self.showImage()
self.dirty = True
self.updateStatus(("Swapped Red and Blue"
if on else "Unswapped Red and Blue"))
def editUnMirror(self, on):
if self.image.isNull():
return
if self.mirroredhorizontally:
self.editMirrorHorizontal(False)
if self.mirroredvertically:
self.editMirrorVertical(False)
def editMirrorHorizontal(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(True, False)
self.showImage()
self.mirroredhorizontally = not self.mirroredhorizontally
self.dirty = True
self.updateStatus(("Mirrored Horizontally"
if on else "Unmirrored Horizontally"))
def editMirrorVertical(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(False, True)
self.showImage()
self.mirroredvertically = not self.mirroredvertically
self.dirty = True
self.updateStatus(("Mirrored Vertically"
if on else "Unmirrored Vertically"))
def editZoom(self):
if self.image.isNull():
return
percent, ok = QInputDialog.getInteger(self,
"Image Changer - Zoom", "Percent:",
self.zoomSpinBox.value(), 1, 400)
if ok:
self.zoomSpinBox.setValue(percent)
def editResize(self):
if self.image.isNull():
return
form = resizedlg.ResizeDlg(self.image.width(),
self.image.height(), self)
if form.exec_():
width, height = form.result()
if (width == self.image.width() and
height == self.image.height()):
self.statusBar().showMessage("Resized to the same size",
5000)
else:
self.image = self.image.scaled(width, height)
self.showImage()
self.dirty = True
size = "{0} x {1}".format(self.image.width(),
self.image.height())
self.sizeLabel.setText(size)
self.updateStatus("Resized to {0}".format(size))
def showImage(self, percent=None):
if self.image.isNull():
return
if percent is None:
percent = self.zoomSpinBox.value()
factor = percent / 100.0
width = self.image.width() * factor
height = self.image.height() * factor
image = self.image.scaled(width, height, Qt.KeepAspectRatio)
self.imageLabel.setPixmap(QPixmap.fromImage(image))
def helpAbout(self):
QMessageBox.about(self, "About Image Changer",
"""<b>Image Changer</b> v {0}
<p>Copyright © 2008 Qtrac Ltd.
All rights reserved.
<p>This application can be used to perform
simple image manipulations.
<p>Python {1} - Qt {2} - PyQt {3} on {4}""".format(
__version__, platform.python_version(),
QT_VERSION_STR, PYQT_VERSION_STR,
platform.system()))
def helpHelp(self):
form = helpform.HelpForm("index.html", self)
form.show()
def filePrintPdf(self, fn):
printer = QPrinter(QPrinter.HighResolution)
printer.setPageSize(QPrinter.A4)
printer.setOutputFileName(fn)
printer.setOutputFormat(QPrinter.PdfFormat)
self.document().print_(printer)
class Window(QMainWindow):
def __init__(self, parent=None):
super(Window, self).__init__(parent)
self.image = QImage()
self.dirty = False
self.filename = None
self.mirroredvertically = False
self.mirroredhorizontally = False
self.printer = None
self.create_widgets()
self.create_actions()
self.load_settings()
self.setWindowTitle("Image Changer")
self.updateFileMenu()
QTimer.singleShot(0, self.loadInitialFile)
def create_widgets(self):
self.imageLabel = QLabel()
self.imageLabel.setMinimumSize(200, 200)
self.imageLabel.setAlignment(Qt.AlignCenter)
self.imageLabel.setContextMenuPolicy(Qt.ActionsContextMenu)
self.setCentralWidget(self.imageLabel)
logDockWidget = QDockWidget("Log", self)
logDockWidget.setObjectName("LogDockWidget")
logDockWidget.setAllowedAreas(Qt.LeftDockWidgetArea
| Qt.RightDockWidgetArea)
self.listWidget = QListWidget()
logDockWidget.setWidget(self.listWidget)
self.addDockWidget(Qt.RightDockWidgetArea, logDockWidget)
self.sizeLabel = QLabel()
self.sizeLabel.setFrameStyle(QFrame.StyledPanel | QFrame.Sunken)
status = self.statusBar()
status.setSizeGripEnabled(False)
status.addPermanentWidget(self.sizeLabel)
status.showMessage("Ready", 5000)
def create_actions(self):
fileNewAction = self.createAction("&New...", self.fileNew,
QKeySequence.New, "filenew",
"Create an image file")
fileOpenAction = self.createAction("&Open...", self.fileOpen,
QKeySequence.Open, "fileopen",
"Open an existing image file")
fileSaveAction = self.createAction("&Save", self.fileSave,
QKeySequence.Save, "filesave",
"Save the image")
fileSaveAsAction = self.createAction(
"Save &As...",
self.fileSaveAs,
icon="filesaveas",
tip="Save the image using a new name")
filePrintAction = self.createAction("&Print", self.filePrint,
QKeySequence.Print, "fileprint",
"Print the image")
fileQuitAction = self.createAction("&Quit", self.close, "Ctrl+Q",
"filequit", "Close the application")
editInvertAction = self.createAction("&Invert", None, "Ctrl+I",
"editinvert",
"Invert the image's colors", True)
editInvertAction.toggled.connect(self.editInvert)
editSwapRedAndBlueAction = self.createAction(
"Sw&ap Red and Blue", None, "Ctrl+A", "editswap",
"Swap the image's red and blue color components", True)
editSwapRedAndBlueAction.toggled.connect(self.editSwapRedAndBlue)
editZoomAction = self.createAction("&Zoom...", self.editZoom, "Alt+Z",
"editzoom", "Zoom the image")
mirrorGroup = QActionGroup(self)
editUnMirrorAction = self.createAction("&Unmirror", None, "Ctrl+U",
"editunmirror",
"Unmirror the image", True)
editUnMirrorAction.toggled.connect(self.editUnMirror)
mirrorGroup.addAction(editUnMirrorAction)
editMirrorHorizontalAction = self.createAction(
"Mirror &Horizontally", None, "Ctrl+H", "editmirrorhoriz",
"Horizontally mirror the image", True)
editMirrorHorizontalAction.toggled.connect(self.editMirrorHorizontal)
mirrorGroup.addAction(editMirrorHorizontalAction)
editMirrorVerticalAction = self.createAction(
"Mirror &Vertically", None, "Ctrl+V", "editmirrorvert",
"Vertically mirror the image", True)
editMirrorVerticalAction.toggled.connect(self.editMirrorVertical)
mirrorGroup.addAction(editMirrorVerticalAction)
editUnMirrorAction.setChecked(True)
helpAboutAction = self.createAction("&About Image Changer",
self.helpAbout)
helpHelpAction = self.createAction("&Help", self.helpHelp,
QKeySequence.HelpContents)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenuActions = (fileNewAction, fileOpenAction, fileSaveAction,
fileSaveAsAction, None, filePrintAction,
fileQuitAction)
self.fileMenu.aboutToShow.connect(self.updateFileMenu)
editMenu = self.menuBar().addMenu("&Edit")
self.addActions(
editMenu,
(editInvertAction, editSwapRedAndBlueAction, editZoomAction))
mirrorMenu = editMenu.addMenu(QIcon(":/editmirror.png"), "&Mirror")
self.addActions(mirrorMenu,
(editUnMirrorAction, editMirrorHorizontalAction,
editMirrorVerticalAction))
helpMenu = self.menuBar().addMenu("&Help")
self.addActions(helpMenu, (helpAboutAction, helpHelpAction))
fileToolbar = self.addToolBar("File")
fileToolbar.setObjectName("FileToolBar")
self.addActions(fileToolbar,
(fileNewAction, fileOpenAction, fileSaveAsAction))
editToolbar = self.addToolBar("Edit")
editToolbar.setObjectName("EditToolBar")
self.addActions(
editToolbar,
(editInvertAction, editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.zoomSpinBox = QSpinBox()
self.zoomSpinBox.setRange(1, 400)
self.zoomSpinBox.setSuffix(" %")
self.zoomSpinBox.setValue(100)
self.zoomSpinBox.setToolTip("Zoom the image")
self.zoomSpinBox.setStatusTip(self.zoomSpinBox.toolTip())
self.zoomSpinBox.setFocusPolicy(Qt.NoFocus)
self.zoomSpinBox.valueChanged.connect(self.showImage)
editToolbar.addWidget(self.zoomSpinBox)
self.addActions(
self.imageLabel,
(editInvertAction, editSwapRedAndBlueAction, editUnMirrorAction,
editMirrorVerticalAction, editMirrorHorizontalAction))
self.resetableActions = ((editInvertAction,
False), (editSwapRedAndBlueAction, False),
(editUnMirrorAction, True))
def load_settings(self):
settings = QSettings()
self.recentFiles = settings.value("RecentFiles").toStringList()
self.restoreGeometry(
settings.value("MainWindow/Geometry").toByteArray())
self.restoreState(settings.value("MainWindow/State").toByteArray())
def createAction(self,
text,
slot=None,
shortcut=None,
icon=None,
tip=None,
checkable=False):
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(":/{0}.png".format(icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
action.triggered.connect(slot)
if checkable:
action.setCheckable(True)
return action
def addActions(self, target, actions):
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def closeEvent(self, event):
if self.okToContinue():
settings = QSettings()
filename = (QVariant(QString(self.filename))
if self.filename is not None else QVariant())
settings.setValue("LastFile", filename)
recentFiles = (QVariant(self.recentFiles)
if self.recentFiles else QVariant())
settings.setValue("RecentFiles", recentFiles)
settings.setValue("MainWindow/Geometry",
QVariant(self.saveGeometry()))
settings.setValue("MainWindow/State", QVariant(self.saveState()))
else:
event.ignore()
def okToContinue(self):
if self.dirty:
reply = QMessageBox.question(
self, "Image Changer - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
return self.fileSave()
return True
def loadInitialFile(self):
settings = QSettings()
fname = unicode(settings.value("LastFile").toString())
if fname and QFile.exists(fname):
self.loadFile(fname)
def updateStatus(self, message):
self.statusBar().showMessage(message, 5000)
self.listWidget.addItem(message)
if self.filename is not None:
self.setWindowTitle("Image Changer - {0}[*]".format(
os.path.basename(self.filename)))
elif not self.image.isNull():
self.setWindowTitle("Image Changer - Unnamed[*]")
else:
self.setWindowTitle("Image Changer[*]")
self.setWindowModified(self.dirty)
def updateFileMenu(self):
self.fileMenu.clear()
self.addActions(self.fileMenu, self.fileMenuActions[:-1])
current = (QString(self.filename)
if self.filename is not None else None)
recentFiles = []
for fname in self.recentFiles:
if fname != current and QFile.exists(fname):
recentFiles.append(fname)
if recentFiles:
self.fileMenu.addSeparator()
for i, fname in enumerate(recentFiles):
action = QAction(
QIcon(":/icon.png"),
"&{0} {1}".format(i + 1,
QFileInfo(fname).fileName()), self)
action.setData(QVariant(fname))
action.triggered.connect(self.loadFile)
self.fileMenu.addAction(action)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.fileMenuActions[-1])
def fileNew(self):
if not self.okToContinue():
return
dialog = newimagedlg.NewImageDlg(self)
if dialog.exec_():
self.addRecentFile(self.filename)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = dialog.image()
self.filename = None
self.dirty = True
self.showImage()
self.sizeLabel.setText("{0} x {1}".format(self.image.width(),
self.image.height()))
self.updateStatus("Created new image")
def fileOpen(self):
if not self.okToContinue():
return
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
formats = ([
"*.{0}".format(unicode(format).lower())
for format in QImageReader.supportedImageFormats()
])
fname = unicode(
QFileDialog.getOpenFileName(
self, "Image Changer - Choose Image", dir,
"Image files ({0})".format(" ".join(formats))))
if fname:
self.loadFile(fname)
def loadFile(self, fname=None):
if fname is None:
action = self.sender()
if isinstance(action, QAction):
fname = unicode(action.data().toString())
if not self.okToContinue():
return
else:
return
if fname:
self.filename = None
image = QImage(fname)
if image.isNull():
message = "Failed to read {0}".format(fname)
else:
self.addRecentFile(fname)
self.image = QImage()
for action, check in self.resetableActions:
action.setChecked(check)
self.image = image
self.filename = fname
self.showImage()
self.dirty = False
self.sizeLabel.setText("{0} x {1}".format(
image.width(), image.height()))
message = "Loaded {0}".format(os.path.basename(fname))
self.updateStatus(message)
def addRecentFile(self, fname):
if fname is None:
return
if not self.recentFiles.contains(fname):
self.recentFiles.prepend(QString(fname))
while self.recentFiles.count() > 9:
self.recentFiles.takeLast()
def fileSave(self):
if self.image.isNull():
return True
if self.filename is None:
return self.fileSaveAs()
else:
if self.image.save(self.filename, None):
self.updateStatus("Saved as {0}".format(self.filename))
self.dirty = False
return True
else:
self.updateStatus("Failed to save {0}".format(self.filename))
return False
def fileSaveAs(self):
if self.image.isNull():
return True
fname = self.filename if self.filename is not None else "."
formats = ([
"*.{0}".format(unicode(format).lower())
for format in QImageWriter.supportedImageFormats()
])
fname = unicode(
QFileDialog.getSaveFileName(
self, "Image Changer - Save Image", fname,
"Image files ({0})".format(" ".join(formats))))
if fname:
if "." not in fname:
fname += ".png"
self.addRecentFile(fname)
self.filename = fname
return self.fileSave()
return False
def filePrint(self):
if self.image.isNull():
return
if self.printer is None:
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.Letter)
form = QPrintDialog(self.printer, self)
if form.exec_():
painter = QPainter(self.printer)
rect = painter.viewport()
size = self.image.size()
size.scale(rect.size(), Qt.KeepAspectRatio)
painter.setViewport(rect.x(), rect.y(), size.width(),
size.height())
painter.drawImage(0, 0, self.image)
def editInvert(self, on):
if self.image.isNull():
return
self.image.invertPixels()
self.showImage()
self.dirty = True
self.updateStatus("Inverted" if on else "Uninverted")
def editSwapRedAndBlue(self, on):
if self.image.isNull():
return
self.image = self.image.rgbSwapped()
self.showImage()
self.dirty = True
self.updateStatus(
("Swapped Red and Blue" if on else "Unswapped Red and Blue"))
def editUnMirror(self, on):
if self.image.isNull():
return
if self.mirroredhorizontally:
self.editMirrorHorizontal(False)
if self.mirroredvertically:
self.editMirrorVertical(False)
def editMirrorHorizontal(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(True, False)
self.showImage()
self.mirroredhorizontally = not self.mirroredhorizontally
self.dirty = True
self.updateStatus(
("Mirrored Horizontally" if on else "Unmirrored Horizontally"))
def editMirrorVertical(self, on):
if self.image.isNull():
return
self.image = self.image.mirrored(False, True)
self.showImage()
self.mirroredvertically = not self.mirroredvertically
self.dirty = True
self.updateStatus(
("Mirrored Vertically" if on else "Unmirrored Vertically"))
def editZoom(self):
if self.image.isNull():
return
percent, ok = QInputDialog.getInteger(self, "Image Changer - Zoom",
"Percent:",
self.zoomSpinBox.value(), 1, 400)
if ok:
self.zoomSpinBox.setValue(percent)
def showImage(self, percent=None):
if self.image.isNull():
return
if percent is None:
percent = self.zoomSpinBox.value()
factor = percent / 100.0
width = self.image.width() * factor
height = self.image.height() * factor
image = self.image.scaled(width, height, Qt.KeepAspectRatio)
self.imageLabel.setPixmap(QPixmap.fromImage(image))
def helpAbout(self):
QMessageBox.about(
self, "About Image Changer", """<b>Image Changer</b> v {0}
<p>Copyright © 2008-14 Qtrac Ltd.
All rights reserved.
<p>This application can be used to perform
simple image manipulations.
<p>Python {1} - Qt {2} - PyQt {3} on {4}""".format(
__version__, platform.python_version(), QT_VERSION_STR,
PYQT_VERSION_STR, platform.system()))
def helpHelp(self):
form = helpform.HelpForm("index.html", self)
form.show()
class MyWindow(QtGui.QMainWindow,Ui_MainWindow):
def __init__(self):
super(MyWindow, self).__init__()
self.filename = None
self.image = None
self.printer = None
self.style_image = None
self.style = None
self.saved = True
self.setupUi(self)
self.init()
def init(self):
self.setWindowTitle(_translate("MainWindow", "Style Transfer - chicago.jpg", None))
self.filename = 'images/chicago.jpg'
self.image = QImage(self.filename)
self.showImage()
@Slot()
def on_actionOpen_File_triggered(self):
dir = (os.path.dirname(self.filename)
if self.filename is not None else ".")
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageReader.supportedImageFormats()])
fname = unicode(QFileDialog.getOpenFileName(self,
"Image Changer - Choose Image",
dir,
"Image files ({0})".format(" ".join(formats))))
if fname:
self.loadFile(fname)
self.changeTitleOriginal()
def loadFile(self, fname=None):
if fname:
self.filename = fname
image = QImage(fname)
if image.isNull():
message = "Failed to read {0}".format(fname)
else:
self.image = image
self.filename = fname
self.showImage()
message = "Loaded {0}".format(os.path.basename(fname))
def showImage(self):
if self.image.isNull():
return
width = self.imageLabel.geometry().width()
height = self.imageLabel.geometry().height()
image = self.image.scaled(width, height, Qt.KeepAspectRatio)
self.imageLabel.setPixmap(QPixmap.fromImage(image))
def showStyleImage(self):
if self.style_image == None:
msg_box = QMessageBox(QMessageBox.Warning, "Alert", "Please configure the baseline!")
msg_box.show()
return
# nimage = numpy2qimage(self.style_image)
scipy.misc.imsave("images/temporary.jpg", np.clip(self.style_image, 0, 255).astype(np.uint8))
nimage = QImage("images/temporary.jpg")
if nimage.isNull():
return
self.image = nimage
self.showImage()
def defaultIcon(self):
icon_lamuse_clicked = QtGui.QIcon()
icon_lamuse_clicked.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/la_muse.jpg")), QtGui.QIcon.Normal,QtGui.QIcon.Off)
self.btn_la_muse.setIcon(icon_lamuse_clicked)
self.btn_la_muse.setIconSize(QtCore.QSize(80, 80))
icon_rainprincess = QtGui.QIcon()
icon_rainprincess.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/rain_princess.jpg")), QtGui.QIcon.Normal,QtGui.QIcon.Off)
self.btn_rain_princess.setIcon(icon_rainprincess)
self.btn_rain_princess.setIconSize(QtCore.QSize(80, 80))
icon_scream = QtGui.QIcon()
icon_scream.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/scream.jpg")), QtGui.QIcon.Normal,QtGui.QIcon.Off)
self.btn_scream.setIcon(icon_scream)
self.btn_scream.setIconSize(QtCore.QSize(80, 80))
icon_udnie = QtGui.QIcon()
icon_udnie.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/udnie.jpg")), QtGui.QIcon.Normal,QtGui.QIcon.Off)
self.btn_udnie.setIcon(icon_udnie)
self.btn_udnie.setIconSize(QtCore.QSize(80, 80))
icon_wave = QtGui.QIcon()
icon_wave.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/wave.jpg")), QtGui.QIcon.Normal,QtGui.QIcon.Off)
self.btn_wave.setIcon(icon_wave)
self.btn_wave.setIconSize(QtCore.QSize(80, 80))
icon_wreck = QtGui.QIcon()
icon_wreck.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/wreck.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_wreck.setIcon(icon_wreck)
self.btn_wreck.setIconSize(QtCore.QSize(80, 80))
@Slot()
def on_actionSave_triggered(self):
fname = self.filename if self.filename is not None else "."
formats = (["*.{0}".format(unicode(format).lower())
for format in QImageWriter.supportedImageFormats()])
fname = unicode(QFileDialog.getSaveFileName(self,
"Image Changer - Save Image", fname,
"Image files ({0})".format(" ".join(formats))))
if fname:
if "." not in fname:
fname += ".png"
self.filename = fname
if self.image.save(self.filename, None):
self.saved = True
self.changeTitleOriginal()
return True
else:
return False
return False
@Slot()
def on_actionPrint_triggered(self):
if self.printer is None:
self.printer = QPrinter(QPrinter.HighResolution)
self.printer.setPageSize(QPrinter.Letter)
form = QPrintDialog(self.printer, self)
if form.exec_():
painter = QPainter(self.printer)
rect = painter.viewport()
size = self.image.size()
size.scale(rect.size(), Qt.KeepAspectRatio)
painter.setViewport(rect.x(), rect.y(), size.width(),
size.height())
painter.drawImage(0, 0, self.image)
@Slot()
def on_actionQuit_triggered(self):
if self.saved == False:
reply = QMessageBox.question(self,
"Image Changer - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel)
if reply == QMessageBox.Cancel:
return False
elif reply == QMessageBox.Yes:
return self.on_actionSave_triggered()
self.image = None
self.filename=None
quit()
def closeEvent(self, QCloseEvent):
self.on_actionQuit_triggered()
def changeTitle(self):
name = "Style Transfer - " + self.filename.split("/")[-1] + "*"
self.setWindowTitle(_translate("MainWindow", name, None))
def changeTitleOriginal(self):
name = "Style Transfer - " + self.filename.split("/")[-1]
self.setWindowTitle(_translate("MainWindow", name, None))
@Slot()
def on_actionla_muse_triggered(self):
#set icon
self.defaultIcon()
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/la_muse_clicked.jpg")), QtGui.QIcon.Normal,
QtGui.QIcon.Off)
self.btn_la_muse.setIcon(icon)
self.btn_la_muse.setIconSize(QtCore.QSize(80, 80))
#style-transfer
self.style = 'la_muse.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_actionrain_princess_triggered(self):
#set icon
self.defaultIcon()
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/rain_princess_clicked.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_rain_princess.setIcon(icon)
self.btn_rain_princess.setIconSize(QtCore.QSize(80, 80))
#style-transfer
self.style = 'rain_princess.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_actionscream_triggered(self):
#set icon
self.defaultIcon()
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/scream_clicked.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_scream.setIcon(icon2)
self.btn_scream.setIconSize(QtCore.QSize(80, 80))
#style_transfer
self.style = 'scream.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_actionudine_triggered(self):
#set icon
self.defaultIcon()
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/udnie_clicked.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_udnie.setIcon(icon3)
self.btn_udnie.setIconSize(QtCore.QSize(80, 80))
#style-transfer
self.style = 'udnie.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_actionwave_triggered(self):
#set icon
self.defaultIcon()
icon4 = QtGui.QIcon()
icon4.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/wave_clicked.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_wave.setIcon(icon4)
self.btn_wave.setIconSize(QtCore.QSize(80, 80))
#style-transfer
self.style = 'wave.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_actionwreck_triggered(self):
#set icon
self.defaultIcon()
icon5 = QtGui.QIcon()
icon5.addPixmap(QtGui.QPixmap(QtCore.QString.fromUtf8("images/wreck_clicked.jpg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.btn_wreck.setIcon(icon5)
self.btn_wreck.setIconSize(QtCore.QSize(80, 80))
#style-transfer
self.style = 'wreck.ckpt'
self.style_image = style_transfer(self.style, self.filename)
self.saved = False
self.showStyleImage()
self.changeTitle()
@Slot()
def on_btn_la_muse_clicked(self):
self.on_actionla_muse_triggered()
@Slot()
def on_btn_rain_princess_clicked(self):
self.on_actionrain_princess_triggered()
@Slot()
def on_btn_scream_clicked(self):
self.on_actionscream_triggered()
@Slot()
def on_btn_udnie_clicked(self):
self.on_actionudine_triggered()
@Slot()
def on_btn_wave_clicked(self):
self.on_actionwave_triggered()
@Slot()
def on_btn_wreck_clicked(self):
self.on_actionwreck_triggered()
@Slot()
def on_action_About_triggered(self):
QMessageBox.about(self, "About Image Style Transfer",
"""<b>Image Style Transfer</b> v {0}
<p>Copyright © 2017-6 GTS.
All rights reserved.
<p>This application can be used to image style transfer.
<p>Python {1} - Qt {2} - PyQt {3} on {4}""".format(
__version__, platform.python_version(),
QT_VERSION_STR, PYQT_VERSION_STR,
platform.system()))
@Slot()
def on_action_Help_F1_triggered(self):
QMessageBox.about(self, "Help about this application",
"""<b>Image Style Transfer</b>
<p>Chose one kind of style,click.
<p>Then the style image will show.""")
