1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#****************************************************************************
# icondict.py, provides a class to load and store icons
#
# Copyright (C) 2014, Douglas W. Bell
#
# This is free software; you can redistribute it and/or modify it under the
# terms of the GNU General Public License, either Version 2 or any later
# version. This program is distributed in the hope that it will be useful,
# but WITTHOUT ANY WARRANTY. See the included LICENSE file for details.
#*****************************************************************************
import os.path
from PyQt4 import QtCore, QtGui
class IconDict(dict):
"""Stores icons by name, loads on demand.
"""
iconExt = ['.png', '.bmp']
def __init__(self):
dict.__init__(self, {})
self.pathList = []
def addIconPath(self, potentialPaths):
"""Add first good path from potentialPaths.
"""
for path in potentialPaths:
try:
for name in os.listdir(path):
pixmap = QtGui.QPixmap(os.path.join(path, name))
if not pixmap.isNull():
self.pathList.append(path)
return
except OSError:
pass
def __getitem__(self, name):
"""Return icon, loading if necessary.
"""
try:
return dict.__getitem__(self, name)
except KeyError:
icon = self.loadIcon(name)
if not icon:
raise
return icon
def loadAllIcons(self):
"""Load all icons available in self.pathList.
"""
self.clear()
for path in self.pathList:
try:
for name in os.listdir(path):
pixmap = QtGui.QPixmap(os.path.join(path, name))
if not pixmap.isNull():
name = os.path.splitext(name)[0]
try:
icon = self[name]
except KeyError:
icon = QtGui.QIcon()
self[name] = icon
icon.addPixmap(pixmap)
except OSError:
pass
def loadIcon(self, iconName):
"""Load icon from iconPath, add to dictionary and return the icon.
"""
icon = QtGui.QIcon()
for path in self.pathList:
for ext in IconDict.iconExt:
fileName = iconName + ext
pixmap = QtGui.QPixmap(os.path.join(path, fileName))
if not pixmap.isNull():
icon.addPixmap(pixmap)
if not icon.isNull():
self[iconName] = icon
return icon
return None
|
