import sys
from qt import *

GOOD = 0

class PenTest(QLabel):
    def __init__(self, parent, name):
        QLabel.__init__(self, parent, name)

    def paintEvent(self, e):
        p = QPainter(self)
        if GOOD:
            pn = QPen()
        else:
            pn = p.pen()
        pn.setColor(Qt.red)
        pn.setWidth(3)
        p.setPen(pn)
        p.drawRect(40, 30, 200, 100)    # draw a rectangle
        p.setPen(Qt.blue)               # set blue pen, 0 pixel width
        p.drawLine(40, 30, 240, 130)    # draw a diagonal in rectangle

    def sizeHint(self):
        return QSize(280, 160)


class Test(QWidget):
    def __init__(self):
        QWidget.__init__(self)
        layout = QVBoxLayout(self, 6, 6, "layout")
        label = PenTest(self, "label")
        layout.addWidget(label)
        button = QPushButton("Done", self, "button")
        layout.addWidget(button)
        app.connect(button, SIGNAL("clicked()"), qApp, SLOT("quit()"))


app = QApplication(sys.argv)
test = Test()
app.setMainWidget(test)
test.show()
app.exec_loop()