Veronica Berglyd Olsen
117 lines
3.3 KiB
Python
117 lines
3.3 KiB
Python
"""
|
||
novelWriter – Custom Widget: Novel Selector
|
||
===========================================
|
||
|
||
File History:
|
||
Created: 2022-11-17 [2.0]
|
||
|
||
This file is a part of novelWriter
|
||
Copyright 2018–2023, Veronica Berglyd Olsen
|
||
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program is distributed in the hope that it will be useful, but
|
||
WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from PyQt5.QtCore import pyqtSignal, pyqtSlot
|
||
from PyQt5.QtWidgets import QComboBox, QWidget
|
||
|
||
from novelwriter import SHARED
|
||
from novelwriter.enum import nwItemClass
|
||
from novelwriter.constants import nwLabels
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class NovelSelector(QComboBox):
|
||
|
||
novelSelectionChanged = pyqtSignal(str)
|
||
|
||
def __init__(self, parent: QWidget) -> None:
|
||
super().__init__(parent=parent)
|
||
self._blockSignal = False
|
||
self._firstHandle = None
|
||
self.currentIndexChanged.connect(self._indexChanged)
|
||
return
|
||
|
||
##
|
||
# Properties
|
||
##
|
||
|
||
@property
|
||
def handle(self) -> str:
|
||
return self.currentData()
|
||
|
||
@property
|
||
def firstHandle(self) -> str | None:
|
||
return self._firstHandle
|
||
|
||
##
|
||
# Methods
|
||
##
|
||
|
||
def setHandle(self, tHandle: str, blockSignal: bool = True) -> None:
|
||
"""Set the currently selected handle."""
|
||
self._blockSignal = blockSignal
|
||
if tHandle is None:
|
||
index = self.count() - 1
|
||
else:
|
||
index = self.findData(tHandle)
|
||
if index >= 0:
|
||
self.setCurrentIndex(index)
|
||
self._blockSignal = False
|
||
return
|
||
|
||
def updateList(self, includeAll: bool = False, prefix: str | None = None) -> None:
|
||
"""Rebuild the list of novel items."""
|
||
self._blockSignal = True
|
||
self._firstHandle = None
|
||
self.clear()
|
||
|
||
icon = SHARED.theme.getIcon(nwLabels.CLASS_ICON[nwItemClass.NOVEL])
|
||
handle = self.currentData()
|
||
for tHandle, nwItem in SHARED.project.tree.iterRoots(nwItemClass.NOVEL):
|
||
if prefix:
|
||
name = prefix.format(nwItem.itemName)
|
||
self.addItem(name, tHandle)
|
||
else:
|
||
name = nwItem.itemName
|
||
self.addItem(icon, nwItem.itemName, tHandle)
|
||
if self._firstHandle is None:
|
||
self._firstHandle = tHandle
|
||
|
||
if includeAll:
|
||
self.insertSeparator(self.count())
|
||
self.addItem(icon, self.tr("All Novel Folders"), "")
|
||
|
||
self.setHandle(handle)
|
||
self.setEnabled(self.count() > 1)
|
||
self._blockSignal = False
|
||
|
||
return
|
||
|
||
##
|
||
# Private Slots
|
||
##
|
||
|
||
@pyqtSlot(int)
|
||
def _indexChanged(self, index: int) -> None:
|
||
"""Re-emit the change of selection signal, unless blocked."""
|
||
if not self._blockSignal:
|
||
self.novelSelectionChanged.emit(self.currentData())
|
||
return
|
||
|
||
# END Class NovelSelector
|