Source code for TermTk.TTkWidgets.listwidget

# MIT License
#
# Copyright (c) 2021 Eugenio Parodi <ceccopierangiolieugenio AT googlemail DOT com>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

__all__ = ['TTkAbstractListItem', 'TTkListWidget']

from dataclasses import dataclass

from TermTk.TTkCore.cfg import TTkCfg
from TermTk.TTkCore.constant import TTkK
from TermTk.TTkCore.log import TTkLog
from TermTk.TTkCore.signal import pyTTkSlot, pyTTkSignal
from TermTk.TTkCore.color import TTkColor
from TermTk.TTkCore.canvas import TTkCanvas
from TermTk.TTkCore.string import TTkString
from TermTk.TTkGui.drag import TTkDrag
from TermTk.TTkWidgets.widget import TTkWidget
from TermTk.TTkAbstract.abstractscrollview import TTkAbstractScrollView

[docs] class TTkAbstractListItem(TTkWidget): '''TTkAbstractListItem''' classStyle = TTkWidget.classStyle | { 'default': {'color': TTkColor.RST}, 'highlighted': {'color': TTkColor.bg('#008855')+TTkColor.UNDERLINE}, 'hover': {'color': TTkColor.bg('#0088FF')}, 'selected': {'color': TTkColor.bg('#0055FF')}, 'clicked': {'color': TTkColor.fg('#FFFF00')}, 'disabled': {'color': TTkColor.fg('#888888')}, } __slots__ = ('_text', '_selected', '_highlighted', '_data', 'listItemClicked') def __init__(self, *, text='', data=None, **kwargs) -> None: self.listItemClicked = pyTTkSignal(TTkAbstractListItem) self._selected = False self._highlighted = False self._text = TTkString(text) self._data = data super().__init__(**kwargs) self.setFocusPolicy(TTkK.ParentFocus)
[docs] def text(self): return self._text
[docs] def setText(self, text): self._text = TTkString(text) self.update()
[docs] def data(self): '''data''' return self._data
[docs] def setData(self, data): '''setData''' if self._data == data: return self._data = data self.update()
def mousePressEvent(self, evt) -> bool: self.listItemClicked.emit(self) return True def _setSelected(self, selected): if self._selected == selected: return self._selected = selected self._highlighted = not selected self.update() def _setHighlighted(self, highlighted): if self._highlighted == highlighted: return self._highlighted = highlighted self.update() def paintEvent(self, canvas): color = (style:=self.currentStyle())['color'] if self._highlighted: color = color+self.style()['highlighted']['color'] if self._selected: color = color+self.style()['selected']['color'] if style==self.style()['hover']: color = color+self.style()['hover']['color'] w = self.width() canvas.drawTTkString(pos=(0,0), width=w, color=color ,text=self._text)
[docs] class TTkListWidget(TTkAbstractScrollView): ''' A :py:class:`TTkListWidget` implements a list view that displays a set of selectable items. :: ╔════════════════════════════════╗ ║S-0) --Zero-3- officia ▲║ ║S-1) ad ipsum ▓║ ║S-2) irure nisi ┊║ ║S-3) minim --Zero-3- ┊║ ║S-4) ea sunt ┊║ ║S-5) qui mollit ┊║ ║S-6) magna sunt ┊║ ║S-7) sunt officia ▼║ ╚════════════════════════════════╝ Quickstart: .. code-block:: python import TermTk as ttk root = ttk.TTk(layout=ttk.TTkHBoxLayout(), mouseTrack=True) ttk.TTkList(parent=root, items=["Item 1","Item 2","Item 3"]) ttk.TTkList(parent=root, items=[f"Item 0x{i:03X}" for i in range(100)]) ttkList = ttk.TTkList(parent=root) ttkList.addItems([f"Item 0x{i:04X}" for i in range(200)]) root.mainloop() ''' itemClicked:pyTTkSignal ''' This signal is emitted whenever an item is clicked. :param item: the item selected :type item: :py:class:`TTkAbstractListItem` ''' textClicked:pyTTkSignal ''' This signal is emitted whenever an item is clicked. :param text: the text of the item selected :type text: str ''' @dataclass(frozen=True) class _DropListData: widget: TTkAbstractScrollView items: list __slots__ = ('itemClicked', 'textClicked', '_selectedItems', '_selectionMode', '_highlighted', '_items', '_dragPos', '_dndMode') def __init__(self, *, items:list[str]=[], selectionMode:int=TTkK.SelectionMode.SingleSelection, dragDropMode:TTkK.DragDropMode=TTkK.DragDropMode.NoDragDrop, **kwargs) -> None: ''' :param items: Use this field to intialize the :py:class:`TTkListWidget` with the entries in the items list, defaults to "[]" :type items: list[str], optional :param selectionMode: This property controls whether the user can select one or many items, defaults to :py:class:`TTkK.SelectionMode.SingleSelection`. :type selectionMode: :py:class:`TTkK.SelectionMode`, optional :param dragDropMode: This property holds the drag and drop event the view will act upon, defaults to :py:class:`TTkK.DragDropMode.NoDragDrop`. :type dragDropMode: :py:class:`TTkK.DragDropMode`, optional ''' # Signals self.itemClicked = pyTTkSignal(TTkAbstractListItem) self.textClicked = pyTTkSignal(str) # Default Class Specific Values self._selectionMode = selectionMode self._selectedItems = [] self._items = [] self._highlighted = None self._dragPos = None self._dndMode = dragDropMode # Init Super super().__init__(**kwargs) self.addItems(items) self.viewChanged.connect(self._viewChangedHandler) self.setFocusPolicy(TTkK.ClickFocus + TTkK.TabFocus) @pyTTkSlot() def _viewChangedHandler(self): x,y = self.getViewOffsets() self.layout().setOffset(-x,-y) @pyTTkSlot(TTkAbstractListItem) def _labelSelectedHandler(self, label:TTkAbstractListItem): if self._selectionMode == TTkK.SingleSelection: for item in self._selectedItems: item._setSelected(False) item._setHighlighted(False) self._selectedItems = [label] label._setSelected(True) elif self._selectionMode == TTkK.MultiSelection: for item in self._selectedItems: item._setHighlighted(False) label._setSelected(not label._selected) if label._selected: self._selectedItems.append(label) else: self._selectedItems.remove(label) if self._highlighted: self._highlighted._setHighlighted(False) label._setHighlighted(True) self._highlighted = label self.itemClicked.emit(label) self.textClicked.emit(label.text())
[docs] def dragDropMode(self): '''dragDropMode''' return self._dndMode
[docs] def setDragDropMode(self, dndMode): '''setDragDropMode''' self._dndMode = dndMode
[docs] def selectionMode(self): '''selectionMode''' return self._selectionMode
[docs] def setSelectionMode(self, mode): '''setSelectionMode''' self._selectionMode = mode
[docs] def selectedItems(self): '''selectedItems''' return self._selectedItems
[docs] def selectedLabels(self): '''selectedLabels''' return [i.text() for i in self._selectedItems]
[docs] def items(self): '''items''' return self._items
def resizeEvent(self, w, h): maxw = 0 for item in self.layout().children(): maxw = max(maxw,item.minimumWidth()) maxw = max(self.width(),maxw) for item in self.layout().children(): x,y,_,h = item.geometry() item.setGeometry(x,y,maxw,h) TTkAbstractScrollView.resizeEvent(self, w, h)
[docs] def viewFullAreaSize(self) -> tuple[int,int]: _,_,w,h = self.layout().fullWidgetAreaGeometry() return w, h
[docs] def addItem(self, item, data=None): '''addItem''' self.addItemAt(item, len(self._items), data)
[docs] def addItems(self, items): '''addItems''' for item in items: self.addItem(item)
def _placeItems(self): minw = self.width() for item in self._items: minw = max(minw,item.minimumWidth()) for y,item in enumerate(self._items): item.setGeometry(0,y,minw,1) self.viewChanged.emit() self.update()
[docs] def addItemAt(self, item, pos, data=None): '''addItemAt''' if isinstance(item, str) or isinstance(item, TTkString): item = TTkAbstractListItem(text=item, data=data) return self.addItemsAt([item],pos)
[docs] def addItemsAt(self, items, pos): '''addItemsAt''' for item in items: if not issubclass(type(item),TTkAbstractListItem): TTkLog.error(f"{item=} is not an TTkAbstractListItem") return for item in items: item.listItemClicked.connect(self._labelSelectedHandler) self._items[pos:pos] = items self.layout().addWidgets(items) self._placeItems()
[docs] def indexOf(self, item): '''indexOf''' for i, it in enumerate(self._items): if it == item: return i return -1
[docs] def itemAt(self, pos): '''itemAt''' return self._items[pos]
[docs] def moveItem(self, fr, to): '''moveItem''' fr = max(min(fr,len(self._items)-1),0) to = max(min(to,len(self._items)-1),0) # Swap self._items[to] , self._items[fr] = self._items[fr] , self._items[to] self._placeItems()
[docs] def removeItem(self, item): '''removeItem''' self.removeItems([item])
[docs] def removeItems(self, items): '''removeItems''' self.layout().removeWidgets(items) for item in items.copy(): item.listItemClicked.disconnect(self._labelSelectedHandler) item._setSelected(False) item._setHighlighted(False) self._items.remove(item) if item in self._selectedItems: self._selectedItems.remove(item) if item == self._highlighted: self._highlighted = None self._placeItems()
[docs] def removeAt(self, pos): '''removeAt''' self.removeItem(self._items[pos])
[docs] def setCurrentRow(self, row): '''setCurrentRow''' if row<len(self._items): item = self._items[row] self.setCurrentItem(item)
[docs] def setCurrentItem(self, item): '''setCurrentItem''' item.listItemClicked.emit(item)
def _moveToHighlighted(self): index = self._items.index(self._highlighted) h = self.height() offx,offy = self.getViewOffsets() if index >= h+offy-1: TTkLog.debug(f"{index} {h} {offy}") self.viewMoveTo(offx, index-h+1) elif index <= offy: self.viewMoveTo(offx, index) def mouseDragEvent(self, evt) -> bool: if not(self._dndMode & TTkK.DragDropMode.AllowDrag): return False if not (items:=self._selectedItems.copy()): return True drag = TTkDrag() data =TTkListWidget._DropListData(widget=self,items=items) h = min(3,ih:=len(items)) + 2 + (1 if ih>3 else 0) w = min(20,iw:=max([it.text().termWidth() for it in items[:3]])) + 2 pm = TTkCanvas(width=w,height=h) for y,it in enumerate(items[:3],1): txt = it.text() if txt.termWidth() < 20: pm.drawText(pos=(1,y), text=it.text()) else: pm.drawText(pos=(1,y), text=it.text(), width=17) pm.drawText(pos=(18,y), text='...') if ih>3: pm.drawText(pos=(1,4), text='...') pm.drawBox(pos=(0,0),size=(w,h)) drag.setPixmap(pm) drag.setData(data) drag.exec() return True def dragEnterEvent(self, evt): if not(self._dndMode & TTkK.DragDropMode.AllowDrop): return False if issubclass(type(evt.data()),TTkListWidget._DropListData): return self.dragMoveEvent(evt) return False def dragMoveEvent(self, evt): offx,offy = self.getViewOffsets() y=min(evt.y+offy,len(self._items)) self._dragPos = (offx+evt.x, y) self.update() return True def dragLeaveEvent(self, evt): self._dragPos = None self.update() return True def dropEvent(self, evt) -> bool: if not(self._dndMode & TTkK.DragDropMode.AllowDrop): return False self._dragPos = None if not issubclass(type(evt.data()) ,TTkListWidget._DropListData): return False offx,offy = self.getViewOffsets() wid = evt.data().widget items = evt.data().items if wid and items: wid.removeItems(items) for it in items: it.setCurrentStyle(it.style()['default']) self.addItemsAt(items,offy+evt.y) return True return False def keyEvent(self, evt): if not self._highlighted: return False if ( evt.type == TTkK.Character and evt.key==" " ) or \ ( evt.type == TTkK.SpecialKey and evt.key == TTkK.Key_Enter ): if self._highlighted: # TTkLog.debug(self._highlighted) self._highlighted.listItemClicked.emit(self._highlighted) return True elif evt.type == TTkK.SpecialKey: if evt.key == TTkK.Key_Tab: return False index = self._items.index(self._highlighted) offx,offy = self.getViewOffsets() h = self.height() if evt.key == TTkK.Key_Up: index = max(0, index-1) elif evt.key == TTkK.Key_Down: index = min(len(self._items)-1, index+1) elif evt.key == TTkK.Key_PageUp: index = max(0, index-h) elif evt.key == TTkK.Key_PageDown: index = min(len(self._items)-1, index+h) elif evt.key == TTkK.Key_Right: self.viewMoveTo(offx+1, offy) elif evt.key == TTkK.Key_Left: self.viewMoveTo(offx-1, offy) elif evt.key == TTkK.Key_Home: self.viewMoveTo(0, offy) elif evt.key == TTkK.Key_End: self.viewMoveTo(0x10000, offy) self._highlighted._setHighlighted(False) self._highlighted = self._items[index] self._highlighted._setHighlighted(True) self._moveToHighlighted() return True return False def focusInEvent(self): if not self._items: return if not self._highlighted: self._highlighted = self._items[0] self._highlighted._setHighlighted(True) self._moveToHighlighted() def focusOutEvent(self): if self._highlighted: self._highlighted._setHighlighted(False) self._dragPos = None # Stupid hack to paint on top of the child widgets def paintChildCanvas(self): super().paintChildCanvas() if self._dragPos: canvas = self.getCanvas() x,y = self._dragPos offx,offy = self.getViewOffsets() p1 = (0,y-offy-1) p2 = (0,y-offy) canvas.drawText(pos=p1,text="╙─╼", color=TTkColor.fg("#FFFF00")+TTkColor.bg("#008855")) canvas.drawText(pos=p2,text="╓─╼", color=TTkColor.fg("#FFFF00")+TTkColor.bg("#008855"))