class ImageFileListModel(MyBaseListModel):
delete_repeat = False
def __init__(self, context):
super().__init__()
self._base_dir = ""
self.__image_extension_list = ['.jpg', '.jpeg', '.bmp', '.png', 'gif', '.dib', '.pcp', '.dif', '.wmf', '.tif',
'.eps', '.psd', '.cdr', '.iff', '.tga', '.pcd', '.mpi', '.icon', '.ico']
self._data_list_in_database = []
self.__db_helper = DBHelper(context)
def data(self, index: QModelIndex, role: int = ...):
if index.isValid() or (0 <= index.row() < len(self._data_list)):
if role == Qt.DisplayRole:
return QVariant(self._data_list[index.row()].name)
elif role == Qt.StatusTipRole:
return QVariant(self._data_list[index.row()].full_path)
elif role == Qt.BackgroundRole:
if self._data_list[index.row()].id != 0:
return QBrush(QColor(84, 255, 159))
else:
return QBrush(QColor(255, 255, 255))
else:
return QVariant()
def rowCount(self, parent: QModelIndex = ...) -> int:
return len(self._data_list)
def get_item(self, row) -> ImageFile:
"""
自定义。获取数据
:param row: 索引
:return:
"""
if -1 < row < len(self._data_list):
return self._data_list[row]
def __add_dir(self, dir_path):
self._base_dir = os.path.basename(dir_path)
for filename in os.listdir(dir_path):
file_path = "%s/%s" % (dir_path, filename)
if os.path.isdir(file_path):
self.__add_children_dir(file_path)
continue
relative_path = "%s/%s" % (self._base_dir, filename)
self.__add_image_data(relative_path, file_path, filename)
def __add_children_dir(self, dir_path):
for filename in os.listdir(dir_path):
file_path = "%s/%s" % (dir_path, filename)
dir_name = os.path.basename(dir_path)
if os.path.isdir(file_path):
self.__add_children_dir(file_path)
continue
relative_path = "%s/%s/%s" % (self._base_dir, dir_name, filename)
self.__add_image_data(relative_path, file_path, filename)
def __add_image_data(self, relative_path, full_path, filename):
if not self.__is_image(filename):
return
image = self.__db_helper.search_by_file_path(full_path)
if not image:
# 根据md5再做1次判断
md5 = FileHelper.get_md5(full_path)
image = self.__db_helper.search_by_md5(md5)
if image and image.path != full_path:
# 已有图片存在且删除重复时删除当前图片
if os.path.exists(image.path) and self.delete_repeat:
print(f'删除重复图片: {full_path}')
os.remove(full_path)
return
image.path = full_path
self.__db_helper.update_image(image)
if image:
image_id = image.id
self._data_list_in_database.append(image)
else:
image_id = 0
item_data = ImageFile(image_id, relative_path, full_path)
self.add_item(item_data)
def add_path(self, path):
if os.path.isdir(path):
self.__add_dir(path)
elif os.path.isfile(path):
self.__add_file(path)
def __add_file(self, file_path):
filename = os.path.basename(file_path)
if not self.__is_image(filename):
return
relative_path = filename
self.__add_image_data(relative_path, file_path, filename)
def set_image_id(self, index, image_id):
self._data_list[index.row()].id = image_id
self.dataChanged.emit(index, index, [Qt.BackgroundRole])
def __is_image(self, filename):
extension = FileHelper.get_file_extension(filename).lower()
return extension in self.__image_extension_list
def clear(self):
super().clear()
self._data_list_in_database.clear()
def get_database_item(self, image_id) -> MyImage:
for image in self._data_list_in_database:
if image.id == image_id:
return image
return None
def set_images(self, image_sql_list, image_file_list):
self.beginResetModel()
self._data_list_in_database = image_sql_list
self._data_list = image_file_list
self.endResetModel()
class ImageManager(QMainWindow, Ui_Manager):
_signal_update_image_id = pyqtSignal(QModelIndex, int)
def __init__(self, parent=None):
super(ImageManager, self).__init__(parent)
self.setupUi(self)
self.__db_helper = DBHelper(self.db_error_handler) # 数据库操作
# 下拉列表设置
self.__type_model = MyBaseListModel()
self.comboBox_type.setModel(self.__type_model)
self.__type_model.add_items(self.__db_helper.get_model_data_list('type'))
self.comboBox_type.setCurrentIndex(0)
self.__level_model = MyBaseListModel()
self.comboBox_level.setModel(self.__level_model)
levels = self.__db_helper.get_model_data_list('level')
for i in range(len(levels)):
level = levels[i]
if level.id == 10:
levels.remove(level)
levels.insert(4, level)
self.__level_model.add_items(levels)
self.comboBox_level.setCurrentIndex(0)
# 图片信息
self.__image_model = ImageFileListModel(self)
self.__image_model.delete_repeat = self.checkBox_delete_repeat.isChecked()
self.__config = ConfigHelper(self)
threading.Thread(target=self._load_default_images, daemon=True).start()
self.lineEdit_sql_where.setText(self.__config.get_config_key('history', 'sqlWhere'))
self.lineEdit_export_dir.setText(self.__config.get_config_key('history', 'lastExportDir'))
self.listView.setModel(self.__image_model)
# 关联事件
self.listView.selectionModel().currentChanged.connect(self.__on_list_view_current_row_change)
self.listView.set_key_press_delegate(self.key_press_delegate)
self.listView.set_action_show_file_directory_delegate(self.open_file_directory)
self.pushButton_classify.clicked.connect(self.__classify)
self.pushButton_search.clicked.connect(self.__search)
self.pushButton_clean.clicked.connect(self.__clean_not_exist_images)
self.checkBox_delete_repeat.clicked.connect(self._on_check_box_delete_repeat_click)
self.actionOpen.triggered.connect(self.__open_files)
self.lineEdit_sql_where.returnPressed.connect(self.__search)
self.pushButton_export_dir.clicked.connect(self.__choose_export_dir)
self.pushButton_export.clicked.connect(self.__choose_export)
self.lineEdit_desc.returnPressed.connect(self.__classify)
self.lineEdit_tag.returnPressed.connect(self.__classify)
self.lineEdit_role.returnPressed.connect(self.__classify)
self.lineEdit_works.returnPressed.connect(self.__classify)
self.lineEdit_series.returnPressed.connect(self.__classify)
self.lineEdit_source.returnPressed.connect(self.__classify)
self.lineEdit_uploader.returnPressed.connect(self.__classify)
self.lineEdit_author.returnPressed.connect(self.__classify)
self._signal_update_image_id.connect(self._update_image_id)
# 设置 tab 切换顺序
self.setTabOrder(self.lineEdit_desc, self.lineEdit_tag)
self.setTabOrder(self.lineEdit_tag, self.lineEdit_path)
self.setTabOrder(self.lineEdit_path, self.comboBox_type)
self.setTabOrder(self.comboBox_type, self.comboBox_level)
self.setTabOrder(self.comboBox_level, self.lineEdit_role)
self.setTabOrder(self.lineEdit_role, self.lineEdit_works)
self.setTabOrder(self.lineEdit_works, self.lineEdit_series)
self.setTabOrder(self.lineEdit_series, self.lineEdit_source)
self.setTabOrder(self.lineEdit_source, self.lineEdit_uploader)
self.setTabOrder(self.lineEdit_uploader, self.lineEdit_author)
self.setTabOrder(self.lineEdit_author, self.pushButton_classify)
self.setTabOrder(self.pushButton_classify, self.pushButton_search)
# 自动补全
self.__completer_list = []
self.__completer_filename = 'works.txt'
if not os.path.exists(self.__completer_filename):
f = open(self.__completer_filename, 'w', encoding='utf-8')
f.close()
with open(self.__completer_filename, 'r+', encoding='utf-8') as f:
self.__completer_list = list(map(lambda x: x.replace("\n", "").replace("\r", ""), f.readlines()))
self.completer = QCompleter(self.__completer_list)
self.completer.setCompletionMode(QCompleter.InlineCompletion)
self.completer.setFilterMode(Qt.MatchContains)
self.lineEdit_works.setCompleter(self.completer)
self.lineEdit_works.editingFinished.connect(self.__add_complete)
# Image.MAX_IMAGE_PIXELS = 1882320000
self.listView.setFocus()
# 预加载图片
threading.Thread(target=self.__preload, daemon=True).start()
def __add_complete(self):
"""
添加自动补全作品
:return:
"""
cur_completion = self.completer.currentCompletion()
if cur_completion == "":
self.__completer_list.append(self.lineEdit_works.text())
self.completer = QCompleter(self.__completer_list)
self.completer.setCompletionMode(QCompleter.InlineCompletion)
self.completer.setFilterMode(Qt.MatchContains)
self.lineEdit_works.setCompleter(self.completer)
print(self.__completer_list)
def _load_default_images(self):
last_dir = self.__config.get_config_key('history', 'lastDir')
if os.path.isdir(last_dir) and os.path.exists(last_dir):
self.__image_model.add_path(last_dir)
if self.__image_model.rowCount() > 0:
self.listView.setCurrentIndex(self.__image_model.index(0, 0))
def __open_files(self):
"""
打开图片文件
:return:
"""
path_list = \
QtWidgets.QFileDialog.getOpenFileNames(self, "选择文件", filter='图片(*.jpg *.png *.gif *.jpeg *.bmp)')[0]
# 生成List使用的Model
for path in path_list:
tp_lists = path.split('/')
item_data = ImageFile(
name="%s/%s" % (tp_lists[-2], tp_lists[-1]),
full_path=path
)
self.__image_model.addItem(item_data)
def __choose_export_dir(self):
"""
选择保存文件夹
:return:
"""
dir_path = QtWidgets.QFileDialog.getExistingDirectory(self, "选择保存的文件夹", "E:/图片")
self.lineEdit_export_dir.setText(dir_path)
def __show_image(self, index):
"""
显示指定索引文件名对应的图片
:param index: 文件索引
:return:
"""
path = self.__image_model.get_item(index).full_path
start_time = time.time()
status = f"[{index + 1}/{self.__image_model.rowCount()}] {path}"
try:
# 填充缩放
pixmap, is_preload = self.__get_image(path)
cur_time = time.time()
status += f"\t是否预加载：{is_preload}\t图片读取：${round((cur_time - start_time) * 1000, 2)}ms"
start_time = time.time()
# 加载图片
item = QtWidgets.QGraphicsPixmapItem(pixmap)
scene = QtWidgets.QGraphicsScene()
scene.addItem(item)
self.graphicsView.setScene(scene)
cur_time = time.time()
status += f"\t图片加载：${round((cur_time - start_time) * 1000, 2)}ms"
except Exception as e:
print(e)
QMessageBox.information(self, "提示", str(e), QMessageBox.Ok)
self.__analysis_file_info(path)
self.statusbar.showMessage(status)
def __on_list_view_current_row_change(self, current: QModelIndex, previous: QModelIndex):
"""
图片列表当前行变化事件
:param current: 当前行索引
:param previous:
:return:
"""
self.__show_image(current.row())
def __analysis_file_info(self, path):
info = self.__db_helper.search_by_file_path(path)
if not info:
# 分析图片信息
self.lineEdit_path.setText(path)
info = ImageHelper.analyze_image_info(path)
self.lineEdit_size.setText(f"{info.size} MB")
self.dateTimeEdit_file_create.setDateTime(info.create_time)
self.lineEdit_desc.setText(info.desc)
self.lineEdit_tag.setText(info.tags)
if info.source:
self.lineEdit_source.setText(info.source)
if info.uploader:
self.lineEdit_uploader.setText(info.uploader)
if info.author:
self.lineEdit_author.setText(info.author)
return
# 显示已有记录
self.lineEdit_desc.setText(info.desc)
self.lineEdit_tag.setText(info.tags)
self.lineEdit_path.setText(info.path)
self.lineEdit_works.setText(info.works)
self.lineEdit_source.setText(info.source)
self.lineEdit_role.setText(info.role)
self.lineEdit_author.setText(info.author)
self.lineEdit_series.setText(info.series)
self.lineEdit_uploader.setText(info.uploader)
self.lineEdit_size.setText(f"{info.size} MB")
self.comboBox_type.setCurrentIndex(self.__type_model.get_index(info.type_id))
self.comboBox_level.setCurrentIndex(self.__level_model.get_index(info.level_id))
self.dateTimeEdit_file_create.setDateTime(info.file_create_time)
self.dateTimeEdit_create.setDateTime(info.create_time)
self.dateTimeEdit_update.setDateTime(info.update_time)
def __classify(self):
"""
分类图片
:return:
"""
select_rows = self.listView.selectionModel().selectedRows()
select_rows = [x for x in select_rows]
th = threading.Thread(target=self.__insert_or_update_db, args=(select_rows,), daemon=True)
th.start()
end_index = select_rows[-1]
self.__select_index(self.__image_model.index(end_index.row() + 1, end_index.column()))
def __insert_or_update_db(self, select_rows):
index = self.comboBox_type.currentIndex()
type_id = self.__type_model.get_item(index).id
index = self.comboBox_level.currentIndex()
level_id = self.__level_model.get_item(index).id
desc = self.lineEdit_desc.text()
author = self.lineEdit_author.text()
tags = self.lineEdit_tag.text()
works = self.lineEdit_works.text()
role = self.lineEdit_role.text()
source = self.lineEdit_source.text()
series = self.lineEdit_series.text()
uploader = self.lineEdit_uploader.text()
for i in range(len(select_rows)):
item = self.__image_model.get_item(select_rows[i].row())
path = item.full_path
image = MyImage(id=item.id, desc=desc, author=author, type_id=type_id, level_id=level_id, tags=tags,
works=works, role=role, source=source, width=self.lineEdit_width.text(),
height=self.lineEdit_height.text(), size=FileHelper.get_file_size_in_mb(path),
filename=item.name, path=path, md5=FileHelper.get_md5(path),
file_create_time=FileHelper.get_create_time_str(path), series=series, uploader=uploader)
if image.id == 0:
self.__db_helper.insert_image(image)
image_id = self.__db_helper.get_id_by_path(path)
self._signal_update_image_id.emit(select_rows[i], image_id)
self.dateTimeEdit_create.setDateTime(datetime.datetime.now())
self.dateTimeEdit_update.setDateTime(datetime.datetime.now())
# message = f"{item.name} 创建完成！"
else:
# 批量更新时，保持原来的描述、作者、等级、标签、作品
old_image = self.__image_model.get_database_item(image.id)
if old_image and len(select_rows) > 1:
image.desc = old_image.desc
image.author = old_image.author
image.level_id = old_image.level_id
image.tags = old_image.tags
image.works = old_image.works
self.__db_helper.update_image(image)
self.dateTimeEdit_update.setDateTime(datetime.datetime.now())
message = f"{item.name} 更新完成！"
self.statusbar.showMessage(f"[{i + 1}/{len(select_rows)}] {message}")
# end_index = select_rows[-1]
def _update_image_id(self, index: QModelIndex, image_id: int):
self.__image_model.set_image_id(index,image_id)
def __select_index(self, index: QModelIndex):
if 0 < index.row() < self.__image_model.rowCount():
self.listView.setCurrentIndex(index)
self.listView.setFocus()
else:
self.listView.clearFocus()
self.listView.setFocus()
def __del_select_rows(self):
"""
删除选中行
:return:
"""
select_rows = self.listView.selectionModel().selectedRows()
if len(select_rows) == 0:
return
first_index = select_rows[0]
for i in range(len(select_rows)):
index = select_rows[i]
item = self.__image_model.get_item(index.row() - i)
if item.id != 0:
self.__db_helper.delete(item.id)
os.remove(item.full_path)
self.__image_model.delete_item(index.row() - i)
self.statusbar.showMessage(f"[{i + 1}/{len(select_rows)}] {item.name} 删除成功！")
if len(select_rows) > 1:
self.listView.clearSelection()
# 如果删除到了最后一行，则刷新上一个
if first_index.row() >= self.__image_model.rowCount():
if first_index.row() == 0:
return
else:
self.listView.setCurrentIndex(self.listView.model().index(first_index.row() - 1, first_index.column()))
else:
if len(select_rows) > 1:
self.listView.setCurrentIndex(first_index)
else:
self.__show_image(first_index.row())
def __search(self):
sql_where = self.lineEdit_sql_where.text()
if not sql_where:
sql_where = ""
if len(self.lineEdit_desc.text()):
sql_where += f" `desc` like '%{self.lineEdit_desc.text()}%'"
if len(self.lineEdit_role.text()):
sql_where += f" `role` like '%{self.lineEdit_role.text()}%'"
if len(self.lineEdit_works.text()):
sql_where += f" `works` like '%{self.lineEdit_works.text()}%'"
if len(self.lineEdit_series.text()):
sql_where += f" `series` like '%{self.lineEdit_series.text()}%'"
if len(self.lineEdit_source.text()):
sql_where += f" `source` like '%{self.lineEdit_source.text()}%'"
if len(self.lineEdit_uploader.text()):
sql_where += f" `uploader` like '%{self.lineEdit_uploader.text()}%'"
if len(self.lineEdit_author.text()):
sql_where += f" `author` like '%{self.lineEdit_author.text()}%'"
image_sql_list, image_file_list = self.__db_helper.search_by_where(sql_where)
if len(image_sql_list) > 0:
self.__image_model.set_images(image_sql_list, image_file_list)
self.listView.setFocus()
self.listView.scrollToTop()
def __choose_export(self):
dir_path = self.lineEdit_export_dir.text()
if not os.path.exists(dir_path):
os.makedirs(dir_path)
if not os.path.isdir(dir_path):
return
for i in range(self.__image_model.rowCount()):
image = self.__image_model.get_item(i)
if image.id:
image_sql = self.__image_model.get_database_item(image.id)
if not os.path.exists(image_sql.path):
continue
try:
FileHelper.copyfile_without_override(image_sql.path, dir_path)
except Exception as e:
print(e)
else:
FileHelper.copyfile_without_override(image.full_path, dir_path)
self.statusbar.showMessage(f"[{i + 1}/{self.__image_model.rowCount()}] {image.name} 复制成功！")
def _on_check_box_delete_repeat_click(self):
print(f'是否删除重复：{self.checkBox_delete_repeat.isChecked()}')
self.__image_model.delete_repeat = self.checkBox_delete_repeat.isChecked()
# region 重写 Qt 控件方法
def keyPressEvent(self, event: QtGui.QKeyEvent) -> None:
# 键盘快捷键事件
if event.key() == Qt.Key_R and QApplication.keyboardModifiers() == Qt.ControlModifier:
self.__classify()
self.listView.setFocus()
if event.key() == Qt.Key_E and QApplication.keyboardModifiers() == Qt.ControlModifier:
self.comboBox_level.setFocus()
if event.key() == Qt.Key_W and QApplication.keyboardModifiers() == Qt.ControlModifier:
self.lineEdit_works.setText("")
# if event.key() == Qt.Key_Delete:
# self.__del_select_rows()
def key_press_delegate(self, event: QtGui.QKeyEvent):
level_index = None
if event.key() == Qt.Key_1:
level_index = 1
if event.key() == Qt.Key_2:
level_index = 2
if event.key() == Qt.Key_3:
level_index = 3
if event.key() == Qt.Key_4:
level_index = 4
if event.key() == Qt.Key_5:
level_index = 5
if event.key() == Qt.Key_6:
level_index = 6
if event.key() == Qt.Key_7:
level_index = 7
if event.key() == Qt.Key_8:
level_index = 8
if event.key() == Qt.Key_9:
level_index = 9
if level_index and self.__level_model.rowCount() >= level_index:
self.comboBox_level.setCurrentIndex(level_index - 1)
return True
if event.key() == Qt.Key_R:
self.__classify()
return True
if event.key() == Qt.Key_E:
self.lineEdit_role.setFocus()
return True
if event.key() == Qt.Key_C:
self.lineEdit_works.setText("")
return True
if event.key() == Qt.Key_D:
self.__del_select_rows()
return True
if event.key() == Qt.Key_W:
current_index = self.listView.currentIndex()
if current_index.row() > 0:
self.listView.setCurrentIndex(self.__image_model.index(current_index.row() - 1, current_index.column()))
return True
if event.key() == Qt.Key_S:
current_index = self.listView.currentIndex()
if current_index.row() < self.__image_model.rowCount() - 1:
self.listView.setCurrentIndex(self.__image_model.index(current_index.row() + 1, current_index.column()))
return True
return False
def dragEnterEvent(self, e: QtGui.QDragEnterEvent) -> None:
# 设置允许接收
e.accept()
def dropEvent(self, e: QtGui.QDropEvent) -> None:
# 接收文件夹和文件以刷新图片列表
urls = e.mimeData().urls()
th = threading.Thread(target=self.__load_list_data, args=(urls,), daemon=True)
th.start()
def __load_list_data(self, urls):
self.__image_model.clear()
for url in urls:
self.__image_model.add_path(url.toLocalFile())
if self.__image_model.rowCount() > 0:
self.listView.setCurrentIndex(self.__image_model.index(0, 0))
if not os.path.isdir(urls[0].toLocalFile()):
return
self.__config.add_config_key('history', 'lastDir', urls[0].toLocalFile())
def closeEvent(self, event: QtGui.QCloseEvent) -> None:
self.__config.add_config_key('history', 'lastExportDir', self.lineEdit_export_dir.text())
self.__config.add_config_key('history', 'sqlWhere', self.lineEdit_sql_where.text())
# 关闭时保存自动填充作品列表的配置文件
with open(self.__completer_filename, 'w+', encoding='utf-8') as f:
f.writelines(list(map(lambda x: x + "\n", self.__completer_list)))
# endregion
# region 预加载图片
__preload_count = 5
__preload_image_queue = queue.Queue(__preload_count)
__preload_image_size = queue.Queue(__preload_count)
__preload_lock = threading.Lock()
def __preload(self):
while True:
if self.__preload_image_queue.qsize() == 5:
time.sleep(1)
continue
index = self.listView.currentIndex().row()
preload_index = index + self.__preload_image_queue.qsize() + 1
image_file = self.__image_model.get_item(preload_index)
if not image_file:
time.sleep(1)
continue
full_path = image_file.full_path
try:
pixmap, width, height = ImageHelper.get_image_from_file(full_path, self.graphicsView.width(),
self.graphicsView.height())
self.__preload_image_queue.put(PreloadImage(full_path, pixmap))
self.__preload_image_size.put((width, height))
print(f"预加载成功：{full_path}")
except Exception as e:
print(e)
print(f"预加载失败：{full_path}")
time.sleep(1)
def __get_image(self, path):
# 优先从队列中获取
while self.__preload_image_queue.qsize() > 0:
image = self.__preload_image_queue.get()
size = self.__preload_image_size.get()
if isinstance(image, PreloadImage) and image.full_path == path:
print("从预载中读取")
self.lineEdit_width.setText(str(size[0]))
self.lineEdit_height.setText(str(size[1]))
return image.pixmap, True
print("从文件中读取")
image, width, height = ImageHelper.get_image_from_file(path, self.graphicsView.width(),
self.graphicsView.height())
self.lineEdit_width.setText(str(width))
self.lineEdit_height.setText(str(height))
return image, False
# endregion
def __clean_not_exist_images(self):
"""
清理不存在的图片
:return:
"""
th = threading.Thread(
target=ImageHelper.refresh_recode_info,
args=(self.db_error_handler,self.show_status_message,),
daemon=True
)
th.start()
def show_status_message(self, message):
self.statusbar.showMessage(message)
def open_file_directory(self):
"""
打开文件所在目录并选中文件
:return:
"""
select_rows = self.listView.selectionModel().selectedRows()
if not len(select_rows):
return
file_path = self.__image_model.get_item(select_rows[0].row()).full_path
FileHelper.open_file_directory(file_path)
def db_error_handler(self, error_str):
QMessageBox.information(self, "提示", error_str, QMessageBox.Ok)