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
|
# -*- coding: utf-8 -*-
#
# Licensed under the terms of the MIT License
# Copyright (c) 2015 Pierre Raybaut
# (see LICENSE file for more details)
"""Qt helpers"""
import os
from qtpy import QtCore as QC
from qtpy import QtGui as QG
from qtpy import QtWidgets as QW
QT_API = os.environ["QT_API"]
def qcolor_from_str(color, default):
"""Return QColor object from str
:param color: Input color
:type color: QColor or str or None
:param QColor default: Default color (returned if color is None)
If color is already a QColor instance, simply return color.
If color is None, return default color.
If color is neither an str nor a QColor instance nor None, raise TypeError.
"""
if color is None:
return default
elif isinstance(color, str):
try:
return getattr(QC.Qt, color)
except AttributeError:
raise ValueError("Unknown Qt color %r" % color)
else:
try:
return QG.QColor(color)
except TypeError:
raise TypeError("Invalid color %r" % color)
def take_screenshot(widget, path, size=None, quit=True):
"""Take screenshot of widget"""
if size is not None:
widget.resize(*size)
widget.show()
QW.QApplication.processEvents()
pixmap = widget.grab()
pixmap.save(path)
if quit:
QC.QTimer.singleShot(0, QW.QApplication.instance().quit)
|
