#!/bin/env python

 # -*- coding: utf-8 -*-

 # ------------------------------------------------------------

 # notification-dialog

 # 

 # show the user an opensecurity specific message box

 #

 # Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>

 #

 # Copyright (C) 2013 AIT Austrian Institute of Technology

 # AIT Austrian Institute of Technology GmbH

 # Donau-City-Strasse 1 | 1220 Vienna | Austria

 # http://www.ait.ac.at

 #

 # This program is free software; you can redistribute it and/or

 # modify it under the terms of the GNU General Public License

 # as published by the Free Software Foundation version 2.

 # 

 # This program is distributed in the hope that it will be useful,

 # but WITHOUT ANY WARRANTY; without even the implied warranty of

 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 # GNU General Public License for more details.

 # 

 # You should have received a copy of the GNU General Public License

 # along with this program; if not, write to the Free Software

 # Foundation, Inc., 51 Franklin Street, Fifth Floor, 

 # Boston, MA  02110-1301, USA.

 # ------------------------------------------------------------

 # ------------------------------------------------------------

 # imports

 import sys

 from PyQt4 import QtCore

 from PyQt4 import QtGui

 # local

 from about import About

 # ------------------------------------------------------------

 # code

 class Notification(QtGui.QDialog):

     """Show the user an OpenSecurity specific notification."""

     TYPES = ['information', 'warning', 'critical']

     def __init__(self, msgtype, text, parent = None, flags = QtCore.Qt.WindowFlags(0)):

         # super call and widget init

         super(Notification, self).__init__(parent, flags)

         self.setWindowTitle('OpenSecuirty Notification')

         self.setup_ui(msgtype)

         # positionate ourself central

         screen = QtGui.QDesktopWidget().screenGeometry()

         self.resize(self.geometry().width() * 1.25, self.geometry().height())

         size = self.geometry()

         self.move((screen.width() - size.width()) / 2, (screen.height() - size.height()) / 2)

         # fix up text

         if msgtype in Notification.TYPES:

             typetext = { 

                 'information': 'OpenSecurity information', 

                 'warning': '<b>OpenSecurity warning</b>', 

                 'critical': '<b>OpenSecurity critical error</b>'

             }

             self.lbMsgType.setText(typetext[msgtype])

             captiontext = { 

                 'information': 'OpenSecurity: Information', 

                 'warning': 'OpenSecurity: Warning', 

                 'critical': 'OpenSecurity: Critical'

             }

             self.setWindowTitle(captiontext[msgtype])

         else:

             raise ValueError('unknown msgtype')

         self.lbText.setText(text)

     def clicked_about(self):

         """clicked the about button"""

         dlgAbout = About()

         dlgAbout.exec_()

     def clicked_ok(self):

         """clicked the ok button"""

         self.accept()

     def setup_ui(self, msgtype):

         """Create the widgets."""

         lyMain = QtGui.QVBoxLayout(self)

         lyMain.setContentsMargins(8, 8, 8, 8)

         # content area: left pixmap, right text

         lyContent = QtGui.QHBoxLayout()

         lyMain.addLayout(lyContent)

         # pixmap

         lbPix = QtGui.QLabel()

         lbPix.setPixmap(QtGui.QPixmapCache.find('opensecurity_icon_64'))

         lyContent.addWidget(lbPix, 0, QtCore.Qt.Alignment(QtCore.Qt.AlignTop + QtCore.Qt.AlignHCenter))

         lyContent.addSpacing(16)

         # text ...

         lyText = QtGui.QVBoxLayout()

         lyContent.addLayout(lyText, 1)

         self.lbMsgType = QtGui.QLabel()

         lyText.addWidget(self.lbMsgType)

         self.lbText = QtGui.QLabel()

         lyText.addWidget(self.lbText)

         lyText.addStretch(1)

         lyMain.addStretch(1)

         # buttons

         lyButton = QtGui.QHBoxLayout()

         lyMain.addLayout(lyButton)

         lyButton.addStretch(1)

         btnOk = QtGui.QPushButton('&Ok', self)

         btnOk.setDefault(True)

         btnOk.setMinimumWidth(100)

         lyButton.addWidget(btnOk)

         btnAbout = QtGui.QPushButton('&About', self)

         btnAbout.setMinimumWidth(100)

         lyButton.addWidget(btnAbout)

         button_width = max(btnOk.width(), btnAbout.width())

         btnOk.setMinimumWidth(button_width)

         btnAbout.setMinimumWidth(button_width)

         # reduce to the max

         self.resize(lyMain.minimumSize())

         # connectors

         btnOk.clicked.connect(self.clicked_ok)

         btnAbout.clicked.connect(self.clicked_about)