class NotificationWidget(QWidget):
"""Widget for notification phone numbers"""
def __init__(self, name, default, parent=None):
"""
Initialise layout.
Arguments:
name - Name of the widget
default - Default value for the widget
parent - Parent widget
Returns:
None
"""
super(NotificationWidget, self).__init__(parent)
# Global content
self.name = name
self.default = default
if self.default == 'choose':
self.edit = QComboBox(self)
else:
self.edit = QLineEdit(self.name, self)
self.edit.setReadOnly(True)
self.check_box = QCheckBox(self.name, self)
# Event
self.check_box.stateChanged.connect(self._change_state)
# Layout
layout = QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(self.check_box)
layout.addWidget(self.edit)
self.edit.setEnabled(False)
# Default
if self.default == 'choose':
pass
else:
self.edit.setText(default)
# Object name
self.edit.setObjectName('noti_edit')
self.check_box.setObjectName('noti_check')
self.exceptions = []
@pyqtSlot()
def _change_state(self):
"""
Change enable state of the edit
Arguments:
None
Returns:
None
"""
self.edit.setEnabled(self.check_box.isChecked())
def add_exceptions(self, name):
"""
Add a person to the exception list and dont send notification anymore.
Arguments:
name - Name of the person
Returns:
None
"""
self.exceptions.append(name)
def update_combo_email(self, users):
"""
Update the combo boxes.
Arguments:
users - User dictionary
Returns:
None
"""
if self.default == 'choose':
items = [
'@ {0}'.format(key) for key in users
if '@ {0}'.format(key) not in self.exceptions
]
self.edit.addItems(sorted(items))
def update_combo_telegram(self, users):
"""
Update the combo boxes.
Arguments:
users - User dictionary
Returns:
None
"""
if self.default == 'choose':
items = [
'T {0}'.format(key) for key in users
if 'T {0}'.format(key) not in self.exceptions
]
self.edit.addItems(sorted(items))
def clear_combo(self):
"""
Remove all users from the combo box.
Arguments:
None
Returns:
None
"""
if self.default == 'choose':
self.edit.clear()
def get_settings(self):
"""
Get text of the currently selected combo item.
Arguments:
None
Returns:
Settings dictionary
"""
settings = {}
if self.default == 'choose':
settings[self.name] = '{0}\t{1}'.format(self.edit.currentText(),
self.check_box.isChecked())
else:
settings[self.name] = '{0}\t{1}'.format(self.edit.text(),
self.check_box.isChecked())
return settings
def set_settings(self, name, state):
"""
Set currently selected combo item in text.
Arguments:
name - Name of the person that should be currently selected.
state - State of the person (True/False;Enables/Disables)
Returns:
None
"""
self.check_box.setChecked(bool(state == 'True'))
if self.default == 'choose':
index = self.edit.findText(name)
self.edit.setCurrentIndex(index)
else:
self.edit.setText(name)
