mirror of
https://github.com/SELinuxProject/setools
synced 2025-02-21 22:46:50 +00:00
8ecedcdf9c
The Ctrl-C and Ctrl-X shortcuts are handled by the copy() and cut() functions in the ApolMainWindow, which just get the currently focused widget and call its function of the same name. However, the custom SEToolsTableView and SEToolsTreeView widgets do not use these functions to implement Ctrl-C/X, but instead override the event() function and check if each received event is a copy/cut key sequence. Functionally this is the same as copy() and cut(), but this leads to an abort in newer versions for Fedora and/or PyQT5 (the reason is not obvious). To avoid the abort, and arguably make things a little more clear, this overrides the copy() and cut() functions in these widgets, moves the specialize copy logic into them, and removes the event() function. Closes #77 Signed-off-by: Steve Lawrence <slawrence@owlcyberdefense.com>
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
# Copyright 2016, Chris PeBenito <pebenito@ieee.org>
|
|
#
|
|
# SPDX-License-Identifier: LGPL-2.1-only
|
|
#
|
|
#
|
|
import csv
|
|
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtGui import QKeySequence, QCursor
|
|
from PyQt5.QtWidgets import QAction, QApplication, QFileDialog, QMenu, QTableView
|
|
|
|
|
|
class SEToolsTableView(QTableView):
|
|
|
|
"""QTableView class extended for SETools use."""
|
|
|
|
def __init__(self, parent):
|
|
super(SEToolsTableView, self).__init__(parent)
|
|
|
|
# set up right-click context menu
|
|
self.save_csv_action = QAction("Save table to CSV...", self)
|
|
self.menu = QMenu(self)
|
|
self.menu.addAction(self.save_csv_action)
|
|
|
|
# connect signals
|
|
self.save_csv_action.triggered.connect(self.choose_csv_save_location)
|
|
|
|
def contextMenuEvent(self, event):
|
|
self.menu.popup(QCursor.pos())
|
|
|
|
def copy(self):
|
|
datamodel = self.model()
|
|
|
|
selected_text = []
|
|
current_row = None
|
|
current_col = None
|
|
prev_row = None
|
|
prev_col = None
|
|
for index in sorted(self.selectionModel().selectedIndexes()):
|
|
current_row = index.row()
|
|
current_col = index.column()
|
|
|
|
if prev_row is not None and current_row != prev_row:
|
|
selected_text.append('\n')
|
|
elif prev_col is not None and current_col != prev_col:
|
|
selected_text.append('\t')
|
|
|
|
selected_text.append(datamodel.data(index, Qt.DisplayRole))
|
|
|
|
prev_row = current_row
|
|
prev_col = current_col
|
|
|
|
QApplication.clipboard().setText("".join(selected_text))
|
|
|
|
def cut(self):
|
|
self.copy()
|
|
|
|
def choose_csv_save_location(self):
|
|
filename = QFileDialog.getSaveFileName(self, "Save to CSV", "table.csv",
|
|
"Comma Separated Values Spreadsheet (*.csv);;"
|
|
"All Files (*)")[0]
|
|
|
|
if filename:
|
|
self.save_csv(filename)
|
|
|
|
def save_csv(self, filename):
|
|
"""Save the current table data to the specified CSV file."""
|
|
|
|
datamodel = self.model()
|
|
row_count = datamodel.rowCount()
|
|
col_count = datamodel.columnCount()
|
|
|
|
with open(filename, 'w') as fd:
|
|
writer = csv.writer(fd, quoting=csv.QUOTE_MINIMAL)
|
|
|
|
# write headers
|
|
csv_row = []
|
|
for col in range(col_count):
|
|
csv_row.append(datamodel.headerData(col, Qt.Horizontal, Qt.DisplayRole))
|
|
|
|
writer.writerow(csv_row)
|
|
|
|
# write data
|
|
for row in range(row_count):
|
|
csv_row = []
|
|
|
|
for col in range(col_count):
|
|
index = datamodel.index(row, col)
|
|
csv_row.append(datamodel.data(index, Qt.DisplayRole))
|
|
|
|
writer.writerow(csv_row)
|