mirror of
https://github.com/Kilian/Trimage.git
synced 2026-01-26 10:08:40 -05:00
Drag and drop now works. PEP-8 compliance a bit closer.
This commit is contained in:
Paul Chaplin
parent
b5c2d73eec
commit
089e8beb4e
3 changed files with 144 additions and 110 deletions
214
trimage.py
214
trimage.py
|
|
@ -2,139 +2,144 @@ import sys
|
|||
from os import system
|
||||
from PyQt4.QtCore import *
|
||||
from PyQt4.QtGui import *
|
||||
|
||||
from hurry.filesize import *
|
||||
|
||||
from ui import Ui_trimage
|
||||
|
||||
|
||||
DEBUG = True
|
||||
|
||||
|
||||
class StartQT4(QMainWindow):
|
||||
def __init__(self, parent=None):
|
||||
QWidget.__init__(self, parent)
|
||||
self.ui = Ui_trimage()
|
||||
self.ui.setupUi(self)
|
||||
self.quit_shortcut = QShortcut(QKeySequence("Ctrl+Q"), self); # todo use standardKey Quit.
|
||||
# set recompress to false
|
||||
self.ui.recompress.setEnabled(False)
|
||||
self.imagelist = []
|
||||
|
||||
# connect signals with slots
|
||||
QObject.connect(self.ui.addfiles, SIGNAL("clicked()"), self.file_dialog)
|
||||
QObject.connect(self.ui.recompress, SIGNAL("clicked()"), self.recompress_files)
|
||||
QObject.connect(self.quit_shortcut, SIGNAL("activated()"), qApp, SLOT('quit()'))
|
||||
QObject.connect(self.ui.processedfiles, SIGNAL("dragEnterEvent()"), self.file_drop)
|
||||
def __init__(self, parent=None):
|
||||
QWidget.__init__(self, parent)
|
||||
self.ui = Ui_trimage()
|
||||
self.ui.setupUi(self)
|
||||
# todo use standardKey Quit.
|
||||
self.quit_shortcut = QShortcut(QKeySequence("Ctrl+Q"), self)
|
||||
# set recompress to false
|
||||
self.ui.recompress.setEnabled(False)
|
||||
self.imagelist = []
|
||||
|
||||
# connect signals with slots
|
||||
QObject.connect(
|
||||
self.ui.addfiles, SIGNAL("clicked()"), self.file_dialog)
|
||||
QObject.connect(
|
||||
self.ui.recompress, SIGNAL("clicked()"), self.recompress_files)
|
||||
QObject.connect(
|
||||
self.quit_shortcut, SIGNAL("activated()"), qApp, SLOT('quit()'))
|
||||
QObject.connect(
|
||||
self.ui.processedfiles, SIGNAL("dragEnterEvent()"), self.file_drop)
|
||||
|
||||
def file_drop(self):
|
||||
print "booya"
|
||||
def file_drop(self):
|
||||
print "booya"
|
||||
|
||||
def file_dialog(self):
|
||||
fd = QFileDialog(self)
|
||||
images = fd.getOpenFileNames(
|
||||
self,
|
||||
"Select one or more image files to compress",
|
||||
"", # directory
|
||||
"Image files (*.png *.jpg)")
|
||||
for image in images:
|
||||
self.compress_file(image)
|
||||
|
||||
def file_dialog(self):
|
||||
fd = QFileDialog(self)
|
||||
images = fd.getOpenFileNames(self,
|
||||
"Select one or more image files to compress",
|
||||
"", # directory
|
||||
"Image files (*.png *.jpg)")
|
||||
for image in images:
|
||||
self.compress_file(image)
|
||||
def enable_recompress(self):
|
||||
self.ui.recompress.setEnabled(True)
|
||||
|
||||
def recompress_files(self):
|
||||
imagelistcopy = self.imagelist
|
||||
self.imagelist = []
|
||||
for image in imagelistcopy:
|
||||
self.compress_file(image[-1])
|
||||
|
||||
def enable_recompress(self):
|
||||
self.ui.recompress.setEnabled(True)
|
||||
def compress_file(self, filename):
|
||||
print filename
|
||||
oldfile = QFileInfo(filename)
|
||||
name = oldfile.fileName()
|
||||
oldfilesize = oldfile.size()
|
||||
oldfilesizestr = size(oldfilesize, system=alternative)
|
||||
|
||||
if name.endsWith("jpg"):
|
||||
runstr = 'jpegoptim --strip-all -f "' + str(filename) + '"'
|
||||
runfile = system(runstr)
|
||||
|
||||
def recompress_files(self):
|
||||
imagelistcopy = self.imagelist
|
||||
self.imagelist = []
|
||||
for image in imagelistcopy:
|
||||
self.compress_file(image[-1])
|
||||
elif name.endsWith("png"):
|
||||
#runstr = ('optipng -force -o7 "' + str(filename)
|
||||
#+ '"; advpng -z4 "' + str(filename) + '"') ## don't do advpng yet
|
||||
runstr = 'optipng -force -o7 "' + str(filename) + '"'
|
||||
runfile = system(runstr)
|
||||
|
||||
if runfile == 0:
|
||||
newfile = QFile(filename)
|
||||
newfilesize = newfile.size()
|
||||
newfilesizestr = size(newfilesize, system=alternative)
|
||||
|
||||
def compress_file(self, filename):
|
||||
print filename
|
||||
oldfile = QFileInfo(filename);
|
||||
name = oldfile.fileName()
|
||||
oldfilesize = oldfile.size()
|
||||
oldfilesizestr = size(oldfilesize, system=alternative)
|
||||
ratio = 100 - (float(newfilesize) / float(oldfilesize) * 100)
|
||||
ratiostr = "%.1f%%" % ratio
|
||||
|
||||
if name.endsWith("jpg"):
|
||||
runstr = 'jpegoptim --strip-all -f "' + str(filename) + '"'
|
||||
runfile = system(runstr)
|
||||
self.imagelist.append(
|
||||
(name, oldfilesizestr, newfilesizestr, ratiostr, filename))
|
||||
self.update_table()
|
||||
|
||||
elif name.endsWith("png"):
|
||||
#runstr = 'optipng -force -o7 "' + str(filename) + '"; advpng -z4 "' + str(filename) + '"' ## don't do advpng yet
|
||||
runstr = 'optipng -force -o7 "' + str(filename) + '"'
|
||||
runfile = system(runstr)
|
||||
else:
|
||||
print "uh. not good" #throw dialogbox error or something?
|
||||
|
||||
if runfile == 0:
|
||||
newfile = QFile(filename)
|
||||
newfilesize = newfile.size()
|
||||
newfilesizestr = size(newfilesize, system=alternative)
|
||||
def update_table(self):
|
||||
tview = self.ui.processedfiles
|
||||
|
||||
ratio = 100 - (float(newfilesize) / float(oldfilesize) * 100)
|
||||
ratiostr = "%.1f%%" % ratio
|
||||
# set table model
|
||||
tmodel = tri_table_model(self, self.imagelist,
|
||||
['Filename', 'Old Size', 'New Size', 'Compressed'])
|
||||
tview.setModel(tmodel)
|
||||
|
||||
self.imagelist.append((name, oldfilesizestr, newfilesizestr, ratiostr, filename))
|
||||
self.update_table()
|
||||
# set minimum size of table
|
||||
vh = tview.verticalHeader()
|
||||
vh.setVisible(False)
|
||||
|
||||
else:
|
||||
print "uh. not good" #throw dialogbox error or something?
|
||||
# set horizontal header properties
|
||||
hh = tview.horizontalHeader()
|
||||
hh.setStretchLastSection(True)
|
||||
|
||||
|
||||
def update_table(self):
|
||||
tview = self.ui.processedfiles
|
||||
|
||||
# set table model
|
||||
tmodel = tri_table_model(self,
|
||||
self.imagelist,
|
||||
['Filename', 'Old Size', 'New Size', 'Compressed'])
|
||||
tview.setModel(tmodel)
|
||||
|
||||
# set minimum size of table
|
||||
vh = tview.verticalHeader()
|
||||
vh.setVisible(False)
|
||||
|
||||
# set horizontal header properties
|
||||
hh = tview.horizontalHeader()
|
||||
hh.setStretchLastSection(True)
|
||||
|
||||
# set all row heights
|
||||
nrows = len(self.imagelist)
|
||||
for row in range(nrows):
|
||||
tview.setRowHeight(row, 25)
|
||||
tview.setColumnWidth(0,300)
|
||||
tview.setDragDropMode(QAbstractItemView.DropOnly)
|
||||
tview.setAcceptDrops(True)
|
||||
self.enable_recompress()
|
||||
# set all row heights
|
||||
nrows = len(self.imagelist)
|
||||
for row in range(nrows):
|
||||
tview.setRowHeight(row, 25)
|
||||
tview.setColumnWidth(0, 300)
|
||||
#tview.setDragDropMode(QAbstractItemView.DropOnly)
|
||||
#tview.setAcceptDrops(True)
|
||||
self.enable_recompress()
|
||||
|
||||
|
||||
class tri_table_model(QAbstractTableModel):
|
||||
def __init__(self, parent, imagelist, header, *args):
|
||||
"""
|
||||
mydata is list of tuples
|
||||
header is list of strings
|
||||
tuple length has to match header length
|
||||
"""
|
||||
QAbstractTableModel.__init__(self, parent, *args)
|
||||
self.imagelist = imagelist
|
||||
self.header = header
|
||||
|
||||
def __init__(self, parent, imagelist, header, *args):
|
||||
"""
|
||||
mydata is list of tuples
|
||||
header is list of strings
|
||||
tuple length has to match header length
|
||||
"""
|
||||
QAbstractTableModel.__init__(self, parent, *args)
|
||||
self.imagelist = imagelist
|
||||
self.header = header
|
||||
|
||||
def rowCount(self, parent):
|
||||
return len(self.imagelist)
|
||||
def rowCount(self, parent):
|
||||
return len(self.imagelist)
|
||||
|
||||
def columnCount(self, parent):
|
||||
return len(self.header)
|
||||
def columnCount(self, parent):
|
||||
return len(self.header)
|
||||
|
||||
def data(self, index, role):
|
||||
if not index.isValid():
|
||||
return QVariant()
|
||||
elif role != Qt.DisplayRole:
|
||||
return QVariant()
|
||||
return QVariant(self.imagelist[index.row()][index.column()])
|
||||
def data(self, index, role):
|
||||
if not index.isValid():
|
||||
return QVariant()
|
||||
elif role != Qt.DisplayRole:
|
||||
return QVariant()
|
||||
return QVariant(self.imagelist[index.row()][index.column()])
|
||||
|
||||
def headerData(self, col, orientation, role):
|
||||
if orientation == Qt.Horizontal and role == Qt.DisplayRole:
|
||||
return QVariant(self.header[col])
|
||||
return QVariant()
|
||||
def headerData(self, col, orientation, role):
|
||||
if orientation == Qt.Horizontal and role == Qt.DisplayRole:
|
||||
return QVariant(self.header[col])
|
||||
return QVariant()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
@ -142,4 +147,3 @@ if __name__ == "__main__":
|
|||
myapp = StartQT4()
|
||||
myapp.show()
|
||||
sys.exit(app.exec_())
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue