94 lines
2.4 KiB
Python
94 lines
2.4 KiB
Python
from math import sqrt
|
|
import hashlib
|
|
try:
|
|
from PyQt5.QtGui import *
|
|
from PyQt5.QtCore import *
|
|
from PyQt5.QtWidgets import *
|
|
except ImportError:
|
|
from PyQt4.QtGui import *
|
|
from PyQt4.QtCore import *
|
|
|
|
|
|
def newIcon(icon):
|
|
return QIcon(':/' + icon)
|
|
|
|
|
|
def newButton(text, icon=None, slot=None):
|
|
b = QPushButton(text)
|
|
if icon is not None:
|
|
b.setIcon(newIcon(icon))
|
|
if slot is not None:
|
|
b.clicked.connect(slot)
|
|
return b
|
|
|
|
|
|
def newAction(parent, text, slot=None, shortcut=None, icon=None,
|
|
tip=None, checkable=False, enabled=True):
|
|
"""Create a new action and assign callbacks, shortcuts, etc."""
|
|
a = QAction(text, parent)
|
|
if icon is not None:
|
|
a.setIcon(newIcon(icon))
|
|
if shortcut is not None:
|
|
if isinstance(shortcut, (list, tuple)):
|
|
a.setShortcuts(shortcut)
|
|
else:
|
|
a.setShortcut(shortcut)
|
|
if tip is not None:
|
|
a.setToolTip(tip)
|
|
a.setStatusTip(tip)
|
|
if slot is not None:
|
|
a.triggered.connect(slot)
|
|
if checkable:
|
|
a.setCheckable(True)
|
|
a.setEnabled(enabled)
|
|
return a
|
|
|
|
|
|
def addActions(widget, actions):
|
|
for action in actions:
|
|
if action is None:
|
|
widget.addSeparator()
|
|
elif isinstance(action, QMenu):
|
|
widget.addMenu(action)
|
|
else:
|
|
widget.addAction(action)
|
|
|
|
|
|
def labelValidator():
|
|
return QRegExpValidator(QRegExp(r'^[^ \t].+'), None)
|
|
|
|
|
|
class struct(object):
|
|
|
|
def __init__(self, **kwargs):
|
|
self.__dict__.update(kwargs)
|
|
|
|
|
|
def distance(p):
|
|
return sqrt(p.x() * p.x() + p.y() * p.y())
|
|
|
|
|
|
def fmtShortcut(text):
|
|
mod, key = text.split('+', 1)
|
|
return '<b>%s</b>+<b>%s</b>' % (mod, key)
|
|
|
|
|
|
def generateColorByText(text):
|
|
color_table = []
|
|
color_table.append(QColor(0, 0, 50))
|
|
color_table.append(QColor(0, 0, 255))
|
|
color_table.append(QColor(0, 50, 0))
|
|
color_table.append(QColor(0, 255, 0))
|
|
color_table.append(QColor(50, 0, 0))
|
|
color_table.append(QColor(255, 0, 0))
|
|
color_table.append(QColor(0, 50, 50))
|
|
color_table.append(QColor(0, 255, 255))
|
|
color_table.append(QColor(50, 50, 0))
|
|
color_table.append(QColor(255, 255, 0))
|
|
color_table.append(QColor(50, 0, 50))
|
|
color_table.append(QColor(255, 0, 255))
|
|
color_table.append(QColor(50, 50, 50))
|
|
color_table.append(QColor(255, 255, 255))
|
|
colorInd = int(hashlib.sha1(text.encode('utf-8')).hexdigest(), 16) % len(color_table)
|
|
return color_table[colorInd]
|