1658986556

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/feeluown/FeelUOwn/commit/<a class=hub.com/feeluown/FeelUOwn/commit/956e1866a924ad8164f9c6b8cbcec805c43ffd8b">956e1866a<a href="https://github.com/feeluown/FeelUOwn/commit/956e1866a924ad8164f9c6b8cbcec805c43ffd8b">">Merge </a><a class="double-link" href="https://github.com/feeluown/FeelUOwn/commit/<a class="double-link" href="https://github.com/feeluown/FeelUOwn/commit/69e789634dd650b5de2f25527a4dc380d9305e30">69e789634</a>">69e789634</a><a href="https://github.com/feeluown/FeelUOwn/commit/956e1866a924ad8164f9c6b8cbcec805c43ffd8b"> into be0c8cf4c">be0c8cf4c</a>

Pull Request Pull Request #530: gui: fix crash due to 'float' is incompatible with 'int'

Run Details

11 of 11 new or added lines in 5 files covered. (100.0%)

4662 of 11349 relevant lines covered (41.08%)

0.41 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

15.34

/feeluown/gui/widgets/imglist.py

"""
ImgList model and delegate

- ImgListDelegate
- ImgListModel

By default, we think the proper width of a cover is about 160px,
the margin between two cover should be about 20px. When the view is
resized, the cover width and the margin should make a few adjustment.
"""

# pylint: disable=unused-argument
import logging
import random

from PyQt5.QtCore import (
    QAbstractListModel, QModelIndex, Qt,
    QRectF, QRect, QSize, QSortFilterProxyModel
)
from PyQt5.QtGui import (
    QImage, QPixmap, QColor,
    QBrush, QPainter, QTextOption, QFontMetrics, QPalette
)
from PyQt5.QtWidgets import (
    QAbstractItemDelegate, QListView, QFrame,
)

from feeluown.utils import aio
from feeluown.utils.reader import wrap
from feeluown.models.uri import reverse
from feeluown.gui.helpers import ItemViewNoScrollMixin, resize_font, ReaderFetchMoreMixin

logger = logging.getLogger(__name__)


COLORS = {
    'yellow':    '#b58900',
    'orange':    '#cb4b16',
    'red':       '#dc322f',
    'magenta':   '#d33682',
    'violet':    '#6c71c4',
    'blue':      '#268bd2',
    'cyan':      '#2aa198',
    'green':     '#859900',
}


class ImgListModel(QAbstractListModel, ReaderFetchMoreMixin):
    def __init__(self, reader, fetch_image, source_name_map=None, parent=None):
        """

        :param reader: objects in reader should have `name` property
        :param fetch_image: func(item, cb, uid)
        :type reader: Iterable
        """
        super().__init__(parent)

        self.reader = self._reader = wrap(reader)
        self._fetch_more_step = 10
        self._items = []
        self._is_fetching = False

        self.source_name_map = source_name_map or {}
        self.fetch_image = fetch_image
        self.colors = []
        self.pixmaps = {}  # {uri: QPixmap}

    def rowCount(self, parent=QModelIndex()):
        return len(self._items)

    def _fetch_more_cb(self, items):
        self._is_fetching = False
        if items is not None and not items:
            self.no_more_item.emit()
            return
        items_len = len(items)
        colors = [random.choice(list(COLORS.values())) for _ in range(0, items_len)]
        self.colors.extend(colors)
        self.on_items_fetched(items)
        for item in items:
            aio.create_task(self.fetch_image(
                item,
                self._fetch_image_callback(item),
                uid=reverse(item) + '/cover'))

    def _fetch_image_callback(self, item):
        def cb(content):
            img = QImage()
            img.loadFromData(content)
            pixmap = QPixmap(img)
            uri = reverse(item)
            self.pixmaps[uri] = pixmap
            row = self._items.index(item)
            top_left = self.createIndex(row, 0)
            bottom_right = self.createIndex(row, 0)
            self.dataChanged.emit(top_left, bottom_right)
        return cb

    def data(self, index, role):
        offset = index.row()
        if not index.isValid() or offset >= len(self._items):
            return None
        item = self._items[offset]
        if role == Qt.DecorationRole:
            uri = reverse(item)
            pixmap = self.pixmaps.get(uri)
            if pixmap is not None:
                return pixmap
            color_str = self.colors[offset]
            color = QColor(color_str)
            color.setAlphaF(0.8)
            return color
        elif role == Qt.DisplayRole:
            return item.name_display
        elif role == Qt.UserRole:
            return item
        elif role == Qt.WhatsThisRole:
            return self.source_name_map.get(item.source, item.source)
        return None


class ImgListDelegate(QAbstractItemDelegate):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.view = parent
        # TODO: move as_circle/w_h_ratio attribute to view
        self.as_circle = True
        self.w_h_ratio = 1.0

    def paint(self, painter, option, index):
        painter.save()
        painter.setRenderHint(QPainter.Antialiasing)
        rect = option.rect
        text_rect_height = 30
        img_text_height = self.view.img_text_height
        source_rect_height = img_text_height - text_rect_height
        text_y = rect.y() + rect.height() - img_text_height
        cover_height = rect.height() - img_text_height
        cover_width = rect.width()
        text_rect = QRectF(rect.x(), text_y, rect.width(), text_rect_height)
        source_rect = QRectF(rect.x(), text_y + text_rect_height - 5,
                             rect.width(), source_rect_height + 5)
        obj = index.data(Qt.DecorationRole)
        if obj is None:
            painter.restore()
            return

        text_color = option.palette.color(QPalette.Text)
        if text_color.lightness() > 150:
            non_text_color = text_color.darker(140)
        else:
            non_text_color = text_color.lighter(150)
        non_text_color.setAlpha(100)
        painter.save()
        pen = painter.pen()
        pen.setColor(non_text_color)
        painter.setPen(pen)
        painter.translate(rect.x(), rect.y())
        if isinstance(obj, QColor):
            color = obj
            brush = QBrush(color)
            painter.setBrush(brush)
        else:
            if obj.height() < obj.width():
                pixmap = obj.scaledToHeight(cover_height, Qt.SmoothTransformation)
            else:
                pixmap = obj.scaledToWidth(cover_width, Qt.SmoothTransformation)
            brush = QBrush(pixmap)
            painter.setBrush(brush)
        border_radius = 3
        if self.as_circle:
            border_radius = cover_width // 2
        cover_rect = QRect(0, 0, cover_width, cover_height)
        painter.drawRoundedRect(cover_rect, border_radius, border_radius)
        painter.restore()
        option = QTextOption()
        source_option = QTextOption()
        if self.as_circle:
            option.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
            source_option.setAlignment(Qt.AlignHCenter | Qt.AlignTop)
        else:
            option.setAlignment(Qt.AlignLeft | Qt.AlignVCenter)
            source_option.setAlignment(Qt.AlignLeft | Qt.AlignTop)
        name = index.data(Qt.DisplayRole)
        fm = QFontMetrics(painter.font())
        elided_name = fm.elidedText(name, Qt.ElideRight, int(text_rect.width()))
        source = index.data(Qt.WhatsThisRole)
        painter.drawText(text_rect, elided_name, option)
        painter.restore()
        painter.save()
        pen = painter.pen()
        font = painter.font()
        resize_font(font, -2)
        painter.setFont(font)
        pen.setColor(non_text_color)
        painter.setPen(non_text_color)
        painter.drawText(source_rect, source, source_option)
        painter.restore()

    def sizeHint(self, option, index):
        width = self.view.img_sizehint()[0]
        if index.isValid():
            return QSize(width, int(width / self.w_h_ratio) + self.view.img_text_height)
        return super().sizeHint(option, index)


class ImgFilterProxyModel(QSortFilterProxyModel):
    def __init__(self, parent=None, types=None):
        super().__init__(parent)

        self.text = ''

    def filter_by_text(self, text):
        if text == self.text:
            return
        self.text = text
        self.invalidateFilter()

    def filterAcceptsRow(self, source_row, source_parent):
        accepted = True
        source_model = self.sourceModel()
        index = source_model.index(source_row, parent=source_parent)
        artist = index.data(Qt.UserRole)
        if self.text:
            accepted = self.text.lower() in artist.name_display.lower()
        return accepted


class ImgListView(ItemViewNoScrollMixin, QListView):
    """
    .. versionadded:: 3.7.7
       The *img_min_width*, *img_spacing*, *img_text_height* parameter were added.
    """
    def __init__(self, parent=None,
                 img_min_width=150, img_spacing=20, img_text_height=40,
                 **kwargs):
        super().__init__(**kwargs)
        QListView.__init__(self, parent=parent)

        self.img_min_width = img_min_width
        self.img_spacing = img_spacing
        self.img_text_height = img_text_height

        # override ItemViewNoScrollMixin variables
        self._least_row_count = 1
        self._row_height = img_min_width + img_spacing + img_text_height

        self.setViewMode(QListView.IconMode)
        self.setResizeMode(QListView.Adjust)
        self.setWrapping(True)
        self.setFrameShape(QFrame.NoFrame)
        self.setSpacing(self.img_spacing)
        self.initialize()

    def resizeEvent(self, e):
        super().resizeEvent(e)

        if self._no_scroll_v is True:
            self._row_height = self.img_sizehint()[1] + self.img_spacing
            self.adjust_height()

    def img_sizehint(self) -> tuple:
        """

        .. versionadded:: 3.7.7
        """
        # HELP: listview needs about 20 spacing left on macOS
        width = self.width() - 20

        img_spacing = self.img_spacing
        img_min_width = self.img_min_width
        img_text_height = self.img_text_height

        # according to our algorithm, when the widget width is:
        #   2(img_min_width + img_spacing) + img_spacing - 1,
        # the cover width can take the maximum width, it will be:
        #   CoverMaxWidth = 2 * img_min_width + img_spacing - 1

        # calculate max column count
        count = (width - img_spacing) // (img_min_width + img_spacing)
        count = max(count, 1)
        img_height = img_width = (width - ((count + 1) * img_spacing)) // count
        img_height = img_height + img_text_height
        return img_width, img_height

1	"""
2	ImgList model and delegate
3
4	- ImgListDelegate
5	- ImgListModel
6
7	By default, we think the proper width of a cover is about 160px,
8	the margin between two cover should be about 20px. When the view is
9	resized, the cover width and the margin should make a few adjustment.
10	"""
11
12	# pylint: disable=unused-argument
13	import logging	1✔
14	import random	1✔
15
16	from PyQt5.QtCore import (	1✔
17	QAbstractListModel, QModelIndex, Qt,
18	QRectF, QRect, QSize, QSortFilterProxyModel
19	)
20	from PyQt5.QtGui import (	1✔
21	QImage, QPixmap, QColor,
22	QBrush, QPainter, QTextOption, QFontMetrics, QPalette
23	)
24	from PyQt5.QtWidgets import (	1✔
25	QAbstractItemDelegate, QListView, QFrame,
26	)
27
28	from feeluown.utils import aio	1✔
29	from feeluown.utils.reader import wrap	1✔
30	from feeluown.models.uri import reverse	1✔
31	from feeluown.gui.helpers import ItemViewNoScrollMixin, resize_font, ReaderFetchMoreMixin	1✔
32
33	logger = logging.getLogger(__name__)	1✔
34
35
36	COLORS = {	1✔
37	'yellow': '#b58900',
38	'orange': '#cb4b16',
39	'red': '#dc322f',
40	'magenta': '#d33682',
41	'violet': '#6c71c4',
42	'blue': '#268bd2',
43	'cyan': '#2aa198',
44	'green': '#859900',
45	}
46
47
48	class ImgListModel(QAbstractListModel, ReaderFetchMoreMixin):	1✔
49	def __init__(self, reader, fetch_image, source_name_map=None, parent=None):	1✔
50	"""
51
52	:param reader: objects in reader should have `name` property
53	:param fetch_image: func(item, cb, uid)
54	:type reader: Iterable
55	"""
56	super().__init__(parent)	×
57
58	self.reader = self._reader = wrap(reader)	×
59	self._fetch_more_step = 10	×
60	self._items = []	×
61	self._is_fetching = False	×
62
63	self.source_name_map = source_name_map or {}	×
64	self.fetch_image = fetch_image	×
65	self.colors = []	×
66	self.pixmaps = {} # {uri: QPixmap}	×
67
68	def rowCount(self, parent=QModelIndex()):	1✔
69	return len(self._items)	×
70
71	def _fetch_more_cb(self, items):	1✔
72	self._is_fetching = False	×
73	if items is not None and not items:	×
74	self.no_more_item.emit()	×
75	return	×
76	items_len = len(items)	×
77	colors = [random.choice(list(COLORS.values())) for _ in range(0, items_len)]	×
78	self.colors.extend(colors)	×
79	self.on_items_fetched(items)	×
80	for item in items:	×
81	aio.create_task(self.fetch_image(	×
82	item,
83	self._fetch_image_callback(item),
84	uid=reverse(item) + '/cover'))
85
86	def _fetch_image_callback(self, item):	1✔
87	def cb(content):	×
88	img = QImage()	×
89	img.loadFromData(content)	×
90	pixmap = QPixmap(img)	×
91	uri = reverse(item)	×
92	self.pixmaps[uri] = pixmap	×
93	row = self._items.index(item)	×
94	top_left = self.createIndex(row, 0)	×
95	bottom_right = self.createIndex(row, 0)	×
96	self.dataChanged.emit(top_left, bottom_right)	×
97	return cb	×
98
99	def data(self, index, role):	1✔
100	offset = index.row()	×
101	if not index.isValid() or offset >= len(self._items):	×
102	return None	×
103	item = self._items[offset]	×
104	if role == Qt.DecorationRole:	×
105	uri = reverse(item)	×
106	pixmap = self.pixmaps.get(uri)	×
107	if pixmap is not None:	×
108	return pixmap	×
109	color_str = self.colors[offset]	×
110	color = QColor(color_str)	×
111	color.setAlphaF(0.8)	×
112	return color	×
113	elif role == Qt.DisplayRole:	×
114	return item.name_display	×
115	elif role == Qt.UserRole:	×
116	return item	×
117	elif role == Qt.WhatsThisRole:	×
118	return self.source_name_map.get(item.source, item.source)	×
119	return None	×
120
121
122	class ImgListDelegate(QAbstractItemDelegate):	1✔
123	def __init__(self, parent=None):	1✔
124	super().__init__(parent)	×
125
126	self.view = parent	×
127	# TODO: move as_circle/w_h_ratio attribute to view
128	self.as_circle = True	×
129	self.w_h_ratio = 1.0	×
130
131	def paint(self, painter, option, index):	1✔
132	painter.save()	×
133	painter.setRenderHint(QPainter.Antialiasing)	×
134	rect = option.rect	×
135	text_rect_height = 30	×
136	img_text_height = self.view.img_text_height	×
137	source_rect_height = img_text_height - text_rect_height	×
138	text_y = rect.y() + rect.height() - img_text_height	×
139	cover_height = rect.height() - img_text_height	×
140	cover_width = rect.width()	×
141	text_rect = QRectF(rect.x(), text_y, rect.width(), text_rect_height)	×
142	source_rect = QRectF(rect.x(), text_y + text_rect_height - 5,	×
143	rect.width(), source_rect_height + 5)
144	obj = index.data(Qt.DecorationRole)	×
145	if obj is None:	×
146	painter.restore()	×
147	return	×
148
149	text_color = option.palette.color(QPalette.Text)	×
150	if text_color.lightness() > 150:	×
151	non_text_color = text_color.darker(140)	×
152	else:
153	non_text_color = text_color.lighter(150)	×
154	non_text_color.setAlpha(100)	×
155	painter.save()	×
156	pen = painter.pen()	×
157	pen.setColor(non_text_color)	×
158	painter.setPen(pen)	×
159	painter.translate(rect.x(), rect.y())	×
160	if isinstance(obj, QColor):	×
161	color = obj	×
162	brush = QBrush(color)	×
163	painter.setBrush(brush)	×
164	else:
165	if obj.height() < obj.width():	×
166	pixmap = obj.scaledToHeight(cover_height, Qt.SmoothTransformation)	×
167	else:
168	pixmap = obj.scaledToWidth(cover_width, Qt.SmoothTransformation)	×
169	brush = QBrush(pixmap)	×
170	painter.setBrush(brush)	×
171	border_radius = 3	×
172	if self.as_circle:	×
173	border_radius = cover_width // 2	×
174	cover_rect = QRect(0, 0, cover_width, cover_height)	×
175	painter.drawRoundedRect(cover_rect, border_radius, border_radius)	×
176	painter.restore()	×
177	option = QTextOption()	×
178	source_option = QTextOption()	×
179	if self.as_circle:	×
180	option.setAlignment(Qt.AlignHCenter \| Qt.AlignVCenter)	×
181	source_option.setAlignment(Qt.AlignHCenter \| Qt.AlignTop)	×
182	else:
183	option.setAlignment(Qt.AlignLeft \| Qt.AlignVCenter)	×
184	source_option.setAlignment(Qt.AlignLeft \| Qt.AlignTop)	×
185	name = index.data(Qt.DisplayRole)	×
186	fm = QFontMetrics(painter.font())	×
187	elided_name = fm.elidedText(name, Qt.ElideRight, int(text_rect.width()))	×
188	source = index.data(Qt.WhatsThisRole)	×
189	painter.drawText(text_rect, elided_name, option)	×
190	painter.restore()	×
191	painter.save()	×
192	pen = painter.pen()	×
193	font = painter.font()	×
194	resize_font(font, -2)	×
195	painter.setFont(font)	×
196	pen.setColor(non_text_color)	×
197	painter.setPen(non_text_color)	×
198	painter.drawText(source_rect, source, source_option)	×
199	painter.restore()	×
200
201	def sizeHint(self, option, index):	1✔
202	width = self.view.img_sizehint()[0]	×
203	if index.isValid():	×
204	return QSize(width, int(width / self.w_h_ratio) + self.view.img_text_height)	×
205	return super().sizeHint(option, index)	×
206
207
208	class ImgFilterProxyModel(QSortFilterProxyModel):	1✔
209	def __init__(self, parent=None, types=None):	1✔
210	super().__init__(parent)	×
211
212	self.text = ''	×
213
214	def filter_by_text(self, text):	1✔
215	if text == self.text:	×
216	return	×
217	self.text = text	×
218	self.invalidateFilter()	×
219
220	def filterAcceptsRow(self, source_row, source_parent):	1✔
221	accepted = True	×
222	source_model = self.sourceModel()	×
223	index = source_model.index(source_row, parent=source_parent)	×
224	artist = index.data(Qt.UserRole)	×
225	if self.text:	×
226	accepted = self.text.lower() in artist.name_display.lower()	×
227	return accepted	×
228
229
230	class ImgListView(ItemViewNoScrollMixin, QListView):	1✔
231	"""
232	.. versionadded:: 3.7.7
233	The img_min_width, img_spacing, img_text_height parameter were added.
234	"""
235	def __init__(self, parent=None,	1✔
236	img_min_width=150, img_spacing=20, img_text_height=40,
237	**kwargs):
238	super().__init__(**kwargs)	×
239	QListView.__init__(self, parent=parent)	×
240
241	self.img_min_width = img_min_width	×
242	self.img_spacing = img_spacing	×
243	self.img_text_height = img_text_height	×
244
245	# override ItemViewNoScrollMixin variables
246	self._least_row_count = 1	×
247	self._row_height = img_min_width + img_spacing + img_text_height	×
248
249	self.setViewMode(QListView.IconMode)	×
250	self.setResizeMode(QListView.Adjust)	×
251	self.setWrapping(True)	×
252	self.setFrameShape(QFrame.NoFrame)	×
253	self.setSpacing(self.img_spacing)	×
254	self.initialize()	×
255
256	def resizeEvent(self, e):	1✔
257	super().resizeEvent(e)	×
258
259	if self._no_scroll_v is True:	×
260	self._row_height = self.img_sizehint()[1] + self.img_spacing	×
261	self.adjust_height()	×
262
263	def img_sizehint(self) -> tuple:	1✔
264	"""
265
266	.. versionadded:: 3.7.7
267	"""
268	# HELP: listview needs about 20 spacing left on macOS
269	width = self.width() - 20	×
270
271	img_spacing = self.img_spacing	×
272	img_min_width = self.img_min_width	×
273	img_text_height = self.img_text_height	×
274
275	# according to our algorithm, when the widget width is:
276	# 2(img_min_width + img_spacing) + img_spacing - 1,
277	# the cover width can take the maximum width, it will be:
278	# CoverMaxWidth = 2 * img_min_width + img_spacing - 1
279
280	# calculate max column count
281	count = (width - img_spacing) // (img_min_width + img_spacing)	×
282	count = max(count, 1)	×
283	img_height = img_width = (width - ((count + 1) * img_spacing)) // count	×
284	img_height = img_height + img_text_height	×
285	return img_width, img_height	×

feeluown / FeelUOwn / 1658986556

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous