76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
from PyQt5.QtWidgets import QTableView, QAction, QMenu, QMessageBox, qApp
|
|
from PyQt5.Qt import Qt, QHeaderView, QCursor
|
|
from SamAuthenticator.AddKeyDialog import AddKeyDialog
|
|
import SamAuthenticator.KeysDataModel as model
|
|
|
|
|
|
class KeyDataView(QTableView):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
self.data_model = None
|
|
|
|
self.add_key_dialog = AddKeyDialog()
|
|
self.add_key_dialog.new_key_to_add_signal.connect(self.add_new_key)
|
|
|
|
self.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
|
|
self.setSelectionBehavior(QTableView.SelectRows)
|
|
self.setSelectionMode(QTableView.SingleSelection)
|
|
self.setContextMenuPolicy(Qt.CustomContextMenu)
|
|
self.customContextMenuRequested.connect(self.contextMenuEvent)
|
|
|
|
def contextMenuEvent(self, event):
|
|
table_context_menu = QMenu(self)
|
|
menu = QMenu(self)
|
|
copy_action = QAction('Copy', self)
|
|
copy_action.triggered.connect(self.copy_selected)
|
|
add_key_action = QAction('Add new key', self)
|
|
add_key_action.triggered.connect(self.add_new_key_from_dialog)
|
|
remove_key_action = QAction('Remove key', self)
|
|
remove_key_action.triggered.connect(self.remove_row)
|
|
|
|
menu.addAction(copy_action)
|
|
menu.addSeparator()
|
|
menu.addAction(add_key_action)
|
|
if len(self.selectedIndexes()) > 0:
|
|
menu.addAction(remove_key_action)
|
|
|
|
menu.popup(QCursor.pos())
|
|
|
|
def copy_selected(self):
|
|
cells = self.selectedIndexes()
|
|
if len(cells) == 0:
|
|
return
|
|
if len(cells) != self.model().columnCount():
|
|
return
|
|
for el in cells:
|
|
if el.column() == model.AuthenticatorKeysDataModel.TOKEN_COL:
|
|
qApp.clipboard().setText(str(el.data()))
|
|
|
|
def remove_row(self):
|
|
cells = self.selectedIndexes()
|
|
if len(cells) == 0:
|
|
return
|
|
if len(cells) != self.model().columnCount():
|
|
return
|
|
for el in cells:
|
|
row = el.row()
|
|
self.model().removeRow(row)
|
|
return
|
|
|
|
def add_new_key_from_dialog(self):
|
|
self.add_key_dialog.get_new_key()
|
|
|
|
def add_new_key(self, name, secret):
|
|
try:
|
|
self.data_model.getKeysObject().test_secret_validity(secret)
|
|
self.data_model.getKeysObject().set_secret(name, secret)
|
|
self.data_model.refreshAll()
|
|
except Exception as e:
|
|
self.add_key_dialog.close()
|
|
QMessageBox.warning(self, "Error", "Testing the secret you entered failed. " + str(e))
|
|
return
|
|
|
|
def set_real_data_model(self, the_model):
|
|
self.data_model = the_model
|