PyQt写的图片浏览器
# -*- coding: utf-8 -*-
# ! /usr/bin/env python
import sys
from PyQt5 import QtGui, QtCore,QtWidgets
from PyQt5.QtCore import QObject, pyqtSignal
import os, string
import math
THUMB_WIDTH = 128
THUMB_HEIGHT = 128
THUMB_MIN = 64
THUMB_MAX = 256
FILE_TYPE = ['png','jpg', 'jpeg', 'tif', 'bmp', 'gif']
class ImageWidget(QtWidgets.QWidget):
# 单选,上一个被选择的对象
prevSelected = None
myclicked = QtCore.pyqtSignal(int)
"""
Use this widget to display image.
"""
def __init__(self):
super(ImageWidget, self).__init__()
self.id = 0
self.displayText = '' # 显示的文字
self.version = ''
self.status = 0
self.path = ''
self.showStatus = True
self.selected = False
self.isHightlight = False
self.thumb = QtGui.QImage()
self.initAttrib()
def initAttrib(self):
self.name_font = QtGui.QFont()
self.bg_color = QtGui.QColor(50, 50, 50)
self.hightlight = QtGui.QColor(255, 255, 255, 100)
self.edge_size = 5
self.pen_selected = QtGui.QPen(QtGui.QColor(255, 255, 0))
self.pen_selected.setWidth(self.edge_size)
self.pen_selected.setJoinStyle(QtCore.Qt.MiterJoin)
# self.setToolTip('aaaa\nbbbb\ncccc')
def assetFile(self):
return self.path + "_asset_.txt"
def thumbFile(self):
return self.path + "_thumb_.png"
def informationFile(self):
return self.path + "_information_.txt"
def getPublishPath(self):
current_version = self.version
if not current_version:
current_version = '000'
new_version = int((current_version)) + 1
return '%s/%03d' % (self.path, new_version)
def getVersionPath(self, version):
return '%s/%s' % (self.path, version)
def getCurrentVersionPath(self):
return self.getVersionPath(self.version)
def setThumb(self, thumb=None):
if not thumb:
thumb = self.thumbFile()
if os.path.isfile(thumb):
self.thumb.load(thumb)
self.repaint()
return True
def paintAsThumb(self, painter):
name_height = max(self.height() * 0.15, 20)
name_ty = self.height() - self.edge_size * 2
# draw background
painter.fillRect(self.rect(), self.bg_color)
painter.drawImage(self.rect(), self.thumb)
# draw hightlight
if self.isHightlight and not self.selected:
painter.fillRect(self.rect(), self.hightlight)
# draw name
painter.setPen(QtGui.QPen(QtGui.QColor(255, 255, 255)))
self.name_font.setPixelSize(name_height)
painter.setFont(self.name_font)
# 脚标字符
painter.drawText(self.edge_size, name_ty, str(self.displayText))
if self.status:
title_height = self.edge_size + name_height
p1 = QtCore.QPoint(0, 0)
p2 = QtCore.QPoint(0, title_height)
p3 = QtCore.QPoint(title_height, 0)
painter.setPen(QtCore.Qt.NoPen)
painter.fillRect(0, 0, self.width(), title_height, QtGui.QColor(40, 40, 40, 40))
if self.status == 1:
painter.setBrush(QtGui.QBrush(QtGui.QColor(255, 0, 0)))
elif self.status == 2:
painter.setBrush(QtGui.QBrush(QtGui.QColor(0, 255, 0)))
elif self.status == 3:
painter.setBrush(QtGui.QBrush(QtGui.QColor(0, 0, 255)))
painter.drawConvexPolygon(p1, p2, p3)
if self.version:
version_x = self.width() - self.edge_size - name_height * 1.5
version_y = name_height
painter.setPen(QtGui.QPen(QtGui.QColor(255, 255, 255)))
painter.drawText(version_x, version_y, '%s' % self.version)
# draw selected
if self.selected:
painter.setPen(self.pen_selected)
painter.setBrush(QtCore.Qt.NoBrush)
painter.drawRect(self.edge_size / 2, self.edge_size / 2, \
self.width() - self.edge_size, self.height() - self.edge_size)
def paintEvent(self, event):
painter = QtGui.QPainter(self)
self.paintAsThumb(painter)
def mouseReleaseEvent(self, event):
if event.button() == QtCore.Qt.LeftButton:
self.setSelected()
def mouseDoubleClickEvent(self, event):
self.emit(pyqtSignal('doubleClick'))
def enterEvent(self, event):
self.isHightlight = True
self.repaint()
def leaveEvent(self, event):
self.isHightlight = False
self.repaint()
# 设定当前为选中状态
def setSelected(self):
# 取消其他缩略图的选择状态, 当前设为选择状态
if ImageWidget.prevSelected != None:
ImageWidget.prevSelected.selected = False
ImageWidget.prevSelected.repaint()
self.selected = True
self.repaint()
ImageWidget.prevSelected = self
self.onWidgetClicked()
self.sender(pyqtSignal("click"), self.id)
def onWidgetClicked(self):
print('on widget clicked')
class ImageContainer(QtWidgets.QFrame):
def __init__(self, widgets=None):
super(ImageContainer, self).__init__()
containerLayout = QtWidgets.QVBoxLayout()
# 初始化Slider
self.zoomSlider = QtWidgets.QSlider()
self.zoomSlider.setOrientation(QtCore.Qt.Horizontal)
self.zoomSlider.setMinimum(THUMB_MIN)
self.zoomSlider.setMaximum(THUMB_MAX)
self.zoomSlider.setValue(THUMB_WIDTH)
self.zoomSlider.setFixedWidth(128)
self.zoomSlider.setFixedHeight(10)
# Slider设定
# QtCore.QObject.connect(self.zoomSlider, QtCore.SIGNAL('valueChanged(int)'), self.setItemSize)
self.zoomSlider.valueChanged.connect(self.setItemSize)
self.item_scrollarea = QtWidgets.QScrollArea()
self.item_area = QtWidgets.QWidget()
self.item_scrollarea.setWidget(self.item_area)
containerLayout.addWidget(self.zoomSlider)
containerLayout.addWidget(self.item_scrollarea)
self.widget_w = THUMB_WIDTH
self.widget_h = THUMB_HEIGHT
self.min_width = THUMB_MIN
self.max_height = THUMB_MAX
self.asset_space = 2
self.auto_space = False
self.setWindowOpacity(0.0)
self.setLayout(containerLayout)
# 缩略图对象列表
self.ImageWidgetList = {}
def addWidget(self, widget):
widget.setParent(self.item_area)
widget.resize(self.widget_w, self.widget_h)
widget.show()
# 添加到列表
self.ImageWidgetList[str(widget.id)] = widget
def addWidgets(self, widgets):
for widget in widgets:
self.addWidget(widget)
self.layout()
def clearAll(self):
widgets = self.item_area.children()
if widgets:
for widget in widgets:
widget.setParent(None)
self.ImageWidgetList.clear()
def layout(self):
w = self.width() - 20
widgets = self.item_area.children()
num_x = max(math.ceil(w / (self.widget_w + self.asset_space)), 1) # Can do -1
num_y = math.ceil(len(widgets) / num_x)
self.item_area.resize(w, num_y * (self.widget_h + self.asset_space) + 50)
main_w = self.item_area.width()
main_h = self.item_area.height()
num_x = max(math.ceil(main_w / (self.widget_w + self.asset_space)), 1) # Can do -1
x = 0
y = 0
for i in range(len(widgets)):
space_x = 0
if self.auto_space:
space_x = (main_w - self.asset_space * 2 - num_x * (self.widget_w + self.asset_space)) / num_x
widgets[i].move(self.asset_space * 2 + x * (self.widget_w + self.asset_space + space_x), \
self.asset_space * 2 + y * (self.widget_h + self.asset_space))
x += 1
if x >= num_x:
x = 0
y += 1
def resizeEvent(self, event):
self.layout()
def changeItemSize(self, mount):
widgets = self.item_area.children()
self.widget_w += mount
if self.widget_w > self.max_height:
self.widget_w = self.max_height
elif self.widget_w < self.min_width:
self.widget_w = self.min_width
self.widget_h += mount
if self.widget_h > self.max_height:
self.widget_h = self.max_height
elif self.widget_h < self.min_width:
self.widget_h = self.min_width
for a in widgets:
a.resize(self.widget_w, self.widget_h)
self.layout()
def setItemSize(self, size):
widgets = self.item_area.children()
self.widget_w = size
self.widget_h = size
for a in widgets:
a.resize(size, size)
self.layout()
# 设定指定id为选中状态
def setSelected(self, id):
print('ImageContainer -> setSelected ', id)
self.ImageWidgetList[str(id)].setSelected()
class MainWindow(QtWidgets.QWidget):
def __init__(self):
app = QtWidgets.QApplication(sys.argv)
super(MainWindow, self).__init__()
self.setWindowTitle("Image Viewer")
self.resize(1280, 800)
# 屏幕居中
self.screen = QtWidgets.QDesktopWidget().screenGeometry()
self.size = self.geometry()
self.move((self.screen.width() - self.size.width()) / 2, (self.screen.height() - self.size.height()) / 2)
self.show()
mainSpliter = QtWidgets.QSplitter(QtCore.Qt.Horizontal)
# 文件夹列表model
self.dirModel = QtWidgets.QDirModel(self)
# 只显示文件夹
self.dirModel.setFilter(QtCore.QDir.Dirs | QtCore.QDir.NoDotAndDotDot)
# 文件夹列表view
self.dirTreeView = QtWidgets.QTreeView()
# 绑定model
self.dirTreeView.setModel(self.dirModel)
self.dirTreeView.hideColumn(1)
self.dirTreeView.hideColumn(2)
self.dirTreeView.hideColumn(3)
# DirTree事件响应
self.dirTreeView.selectionModel().selectionChanged.connect(self.dirTreeClicked)
mainLayout = QtWidgets.QVBoxLayout()
mainSpliter.addWidget(self.dirTreeView)
self.imageContainer = ImageContainer(mainSpliter)
# self.imageContainer.setGeometry(self.imageContainer.geometry().x(), self.imageContainer.geometry().y(), 100, self.imageContainer.geometry().height())
self.imageContainer.setMinimumWidth(self.geometry().width() * 0.7)
mainSpliter.addWidget(self.imageContainer)
mainLayout.addWidget(mainSpliter)
self.setLayout(mainLayout)
sys.exit(app.exec_())
def dirTreeClicked(self):
print('dirTreeClicked')
self.imageContainer.clearAll()
# 获取选择的路径
pathSelected = self.dirModel.filePath(self.dirTreeView.selectedIndexes()[0])
print('pathSelected ', pathSelected)
# 遍历路径下的媒体文件
for item in os.listdir(pathSelected):
if item.split('.')[-1] in FILE_TYPE:
print(item)
# 添加widget
try:
widget = ImageWidget()
widget.displayText = item
widget.setThumb(str(pathSelected + '/' + item))
self.imageContainer.addWidget(widget)
except:
print(u'出错了')
pass
# self.imageContainer.layout()
def main():
MainWindow()
if __name__ == '__main__':
main()效果图：