def accept(self, success, message):
self.set_elements_enabled(True)
self.loading_dialog.hide()
if not success:
QMessageBox.warning(self, "Change Password Failed", message)
self.logger.info("Change password failed displayed")
return
self.logger.info("Changed password")
QMessageBox.information(self, "Password Changed",
"Password successfully changed.")
super().accept()
def accept(self, success):
self.setWindowTitle(self.title)
if self.reauth:
self.loading_dialog.hide()
self.set_elements_enabled(True)
if not success:
QMessageBox.warning(self, "Sign In Failed", "Invalid credentials.")
self.logger.info("Login failed displayed")
return
self.logger.info("Login succeded")
super().accept()
def position_slider_updated(self, pos):
if float(self.voltage_readout.text()[:-2]) < 5:
QMessageBox.warning(
None,
"Error",
"The voltage going through the servo is too low. Is your battery powered on?",
)
return
self.active_servo.move(pos)
self.position_slider_readout.setText(
f"{int(pos * 25 / 6) * 6 / 25:0.2f}°")
def createConnection(databaseName):
"""Create and open a database connection"""
connection = QSqlDatabase.addDatabase("QSQLITE")
connection.setDatabaseName(databaseName)
if not connection.open():
QMessageBox.warning(
None,
"RP Contact",
f"Database Error: {connection.lastError().text()}",
)
return False
_createContactsTable()
return True
def id_updated(self):
new_id = self.set_id_line_edit.text()
try:
servo = LX16A(int(new_id))
except ServoTimeoutError:
# Meaning this ID is not taken
self.active_servo.set_id(int(new_id))
self.id_selection_box.item(
self.id_selection_box.currentRow()).setText(new_id)
return
QMessageBox.warning(None, "Error", "ID already taken")
def importImages(self):
"""Import the images a user selects, remove duplicates, and add
items to the QListWidget."""
duplicate_images = [] # Store the names of duplicate images
image_paths, _ = QFileDialog.getOpenFileNames(self,
"Select Image Files", "", "Images (*.png *.xpm *.jpg *.jpeg)")
if image_paths:
if self.image_dir.exists():
for image_path in image_paths:
# Pass image path to QFileInfo object
image_info = QFileInfo(image_path)
file_name = image_info.fileName()
item_name = image_info.baseName()
# Copy the files into the Images directory, check for files
# with the same name
new_name = self.image_dir.absolutePath() + f"/{file_name}"
file_exists = QFile.copy(image_path, new_name)
if file_exists == False:
duplicate_images.append(image_path)
else:
self.createListItems(image_path, item_name, image_info, new_name)
if self.is_delete_checked == True: # Handle deleting images
QFile.moveToTrash(image_path)
else:
QMessageBox.warning(self, "Images Location Not Found",
"""<p>The Images Location cannot be found. Restart the application to
recreate the directory.</p>""")
# Display a custom dialog to inform the user of duplicate images
if len(duplicate_images) != 0:
duplicates_dialog = QMessageBox(self)
duplicates_dialog.setIcon(QMessageBox.Icon.Information)
duplicates_dialog.setWindowTitle("Duplicate Images")
duplicates_dialog.setText("""<p>Some images were not imported because
they already exist.</p>""")
details = '\n'.join([item for item in duplicate_images])
duplicates_dialog.setDetailedText(details)
duplicates_dialog.exec()
duplicate_images.clear() # Clear the list
# Check if window is still in focus. If not, give it focus
if self.isActiveWindow() == False:
self.activateWindow()
def callback_exception(self, exc=None):
if exc is None:
exc = common.get_exception()
Log.append(self.callback_exception, 'Error', exc)
if QMessageBox.warning(None, self.lang.title_crash,
self.lang.description_crash):
self.export_log()
def closeEvent(self, event):
if self.thread.is_alive() and QMessageBox.warning(
self, '关闭', '战斗正在进行,确认关闭?',
QMessageBox.Yes | QMessageBox.No) != QMessageBox.Yes:
event.ignore()
return
fgoFunc.control.terminate()
if not self.thread._started: self.thread.join()
event.accept()
def addFile(self, path):
lines = None
content = ""
filename = os.path.basename(path)
roomname = self.roomNameFromFileName(filename)
try:
with open(path, "r", encoding="utf-16-le") as f:
content = f.read()
logging.info("Add room " + roomname + " to list.")
except Exception as e:
self.ignoredPaths.append(path)
QMessageBox.warning(
None,
"Read a log file failed!",
"File: {0} - problem: {1}".format(path, str(e)),
QMessageBox.Ok,
)
return None
lines = content.split("\n")
if path not in self.fileData or (
roomname in LOCAL_NAMES and "charname" not in self.fileData.get(path, [])
):
self.fileData[path] = {}
if roomname in LOCAL_NAMES:
charname = None
sessionStart = None
# for local-chats we need more infos
for line in lines:
if "Listener:" in line:
charname = line[line.find(":") + 1 :].strip()
elif "Session started:" in line:
sessionStr = line[line.find(":") + 1 :].strip()
sessionStart = datetime.datetime.strptime(
sessionStr, "%Y.%m.%d %H:%M:%S"
)
if charname and sessionStart:
self.fileData[path]["charname"] = charname
self.fileData[path]["sessionstart"] = sessionStart
break
self.fileData[path]["lines"] = len(lines)
logging.debug([m.encode("ascii", "ignore") for m in lines])
return lines
def deleteContact(self):
row = self.table.currentIndex().row()
if row < 0:
return
messageBox = QMessageBox.warning(
self,
"Warning!",
"Do you want to remove the selected contact?",
QMessageBox.Ok | QMessageBox.Cancel,
)
if messageBox == QMessageBox.Ok:
self.contactsModel.deleteContact(row)
def askQuit(self):
if self.worker.is_alive():
if QMessageBox.warning(
self, 'FGO-py', '战斗正在进行,确认关闭?',
QMessageBox.StandardButton.Yes
| QMessageBox.StandardButton.No, QMessageBox.
StandardButton.No) != QMessageBox.StandardButton.Yes:
return False
fgoKernel.schedule.stop('Quit')
self.worker.join()
self.TRAY.hide()
self.config.save()
return True
def clearRowsFromTable(self):
"""Delete all rows and data."""
answer = QMessageBox.warning(
self, "Clear Table",
f"Do you really want to clear the {self._headers[0]} table?",
QMessageBox.StandardButton.No | QMessageBox.StandardButton.Yes,
QMessageBox.StandardButton.No)
if answer == QMessageBox.StandardButton.Yes:
# Clear the table and update the Money Left Over table
self.model.removeRows(self.model.rowCount(QModelIndex()), 0,
self.model.rowCount(QModelIndex()))
self.totals_updated.emit(self.running_totals)
self.addRowToTable()
def deleteImages(self):
"""Delete images from the QListWidget and from where images
are stored on disk."""
number_of_photos = len(self.image_view_lw.selectedItems())
answer = QMessageBox.warning(self, "Delete Image(s)",
f"Are you sure you want to delete {number_of_photos} image(s)?",
QMessageBox.StandardButtons.No | QMessageBox.StandardButtons.Yes,
QMessageBox.StandardButtons.No)
if answer == QMessageBox.StandardButtons.Yes:
for item in self.image_view_lw.selectedItems():
index = self.image_view_lw.indexFromItem(item).row()
# Get the image's information before deletion
image_info = self.image_view_lw.images_info_list[index]
# Remove items from the Images directory, from the list widget,
# and the images_info_list that stores QFileInfo objects
QFile.moveToTrash(image_info.absoluteFilePath())
self.image_view_lw.takeItem(index)
del self.image_view_lw.images_info_list[index]
del item
class YTdownloader(QWidget):
def __init__(self):
super().__init__()
# setup some flags
self.isFetching = False
self.isDownloading = False
# default output path
self.outputPath = f'{QDir.homePath()}/videos'
# setup some window specific things
self.setWindowTitle('YouTube Downloader')
self.setWindowIcon(QIcon('assets/yt-icon.ico'))
self.setFixedSize(705, 343)
# parent layout
layout = QVBoxLayout()
layout.setContentsMargins(15, 15, 15, 10)
self.setLayout(layout)
# top bar layout
topBar = QHBoxLayout()
# detail section
detailSec = QHBoxLayout()
metaSec = QVBoxLayout()
# download section
downloadSec = QHBoxLayout()
downloadBtn = QVBoxLayout()
# output path link button
self.outputBtn = QPushButton('📂 Output Path')
self.outputBtn.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.outputBtn.setToolTip(self.outputPath)
self.outputBtn.clicked.connect(self.setOutputPath)
# status bar
self.statusBar = QStatusBar()
# message box
self.message = QMessageBox()
# setting up widgets
self.urlBox = QLineEdit()
self.urlBox.setFocusPolicy(Qt.FocusPolicy.ClickFocus or Qt.FocusPolicy.NoFocus)
self.urlBox.setPlaceholderText('🔍 Enter or paste video URL...')
self.button = QPushButton('Get')
self.button.setDefault(True)
self.button.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.button.clicked.connect(self.getDetails)
# thumbnail
pixmap = QPixmap('assets\placeholder.jpg')
self.thumb = QLabel()
self.thumb.setFixedSize(250, 141)
self.thumb.setScaledContents(True)
self.thumb.setPixmap(pixmap)
# detail widgets
self.title = QLabel('Title: ')
self.author = QLabel('Author: ')
self.length = QLabel('Duration: ')
self.publish_date = QLabel('Published: ')
# progress bar
self.progress_bar = QProgressBar()
# download options
self.download = QComboBox()
self.download.setPlaceholderText('Download Video')
self.download.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.download.activated.connect(lambda: self.getContent(0))
self.download.setEnabled(False)
# download audio button
self.download_audio = QPushButton('Download Audio')
self.download_audio.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.download_audio.clicked.connect(lambda: self.getContent(1))
self.download_audio.setEnabled(False)
# add widgets and layouts
topBar.addWidget(self.urlBox)
topBar.addWidget(self.button)
# detail section
metaSec.addWidget(self.title)
metaSec.addWidget(self.author)
metaSec.addWidget(self.length)
metaSec.addWidget(self.publish_date)
detailSec.addWidget(self.thumb)
detailSec.addSpacing(20)
detailSec.addLayout(metaSec)
# download section
downloadBtn.addWidget(self.download)
downloadBtn.addWidget(self.download_audio)
downloadSec.addWidget(self.progress_bar)
downloadSec.addSpacing(10)
downloadSec.addLayout(downloadBtn)
# status bar
self.statusBar.setSizeGripEnabled(False)
self.statusBar.addPermanentWidget(self.outputBtn)
# add content to parent layout
layout.addLayout(topBar)
layout.addSpacing(20)
layout.addLayout(detailSec)
layout.addSpacing(5)
layout.addLayout(downloadSec)
layout.addWidget(self.statusBar)
# setup a connection thread to keep checking internet connectivity
self.connection = ConnectionThread()
self.connection.start()
# catch the connection response signal
self.connection.con_response.connect(self.connection_slot)
# connection slot
def connection_slot(self, status):
curMsg = self.statusBar.currentMessage()
# connection succeeded
if status:
if curMsg == '🔴 Disconnected':
self.statusBar.showMessage('🟢 Connection restored!', 3000)
elif curMsg != '🟢 Connected':
self.statusBar.showMessage('🟢 Connected')
# connection failed
elif curMsg == '🟢 Connected':
self.statusBar.showMessage('🔴 Connection interrupted!', 3000)
elif curMsg != '🔴 Disconnected':
self.statusBar.showMessage('🔴 Disconnected')
# set output path slot
def setOutputPath(self):
# update the output path
path = str(QFileDialog.getExistingDirectory(self, "Select Output Directory"))
if path:
self.outputPath = path
# update tooltip
self.outputBtn.setToolTip(path)
# get button slot
def getDetails(self):
curMsg = self.statusBar.currentMessage()
if curMsg == '🔴 Disconnected' or curMsg == '🔴 Connection interrupted!':
self.message.critical(
self,
'Error',
'Connection failed!\nAre you sure you\'re connected to the internet ? '
)
elif self.button.text() == 'Get':
self.button.setText('Stop')
# indicate progress bar as busy
self.progress_bar.setRange(0, 0)
# set fetching flag
self.isFetching = True
# setup a worker thread to keep UI responsive
self.worker = WorkerThread(self.urlBox.text())
self.worker.start()
# catch the finished signal
self.worker.finished.connect(self.finished_slot)
# catch the response signal
self.worker.worker_response.connect(self.response_slot)
# catch the error signal
self.worker.worker_err_response.connect(self.err_slot)
elif self.button.text() == 'Stop':
if self.isFetching:
# stop worker thread
self.worker.terminate()
# set back the button text
self.button.setText('Get')
elif self.isDownloading:
# stop download thread
self.download_thread.terminate()
# show the warning message
self.message.information(
self,
'Interrupted',
'Download interrupted!\nThe process was aborted while the file was being downloaded... '
)
# reset pogress bar
self.progress_bar.reset()
# download options slot
def getContent(self, id):
if self.isFetching:
# show the warning message
self.message.warning(
self,
'Warning',
'Please wait!\nWait while the details are being fetched... '
)
else:
# disable the download options
self.download.setDisabled(True)
self.download_audio.setDisabled(True)
# set downloading flag
self.isDownloading = True
# set button to stop
self.button.setText('Stop')
# setup download thread
if id == 0:
self.download_thread = DownloadThread(self.yt, self.download.currentText()[:4], self.outputPath)
else:
self.download_thread = DownloadThread(self.yt, 'audio', self.outputPath)
# start the thread
self.download_thread.start()
# catch the finished signal
self.download_thread.finished.connect(self.download_finished_slot)
# catch the response signal
self.download_thread.download_response.connect(self.download_response_slot)
# catch the complete signal
self.download_thread.download_complete.connect(self.download_complete_slot)
# catch the error signal
self.download_thread.download_err.connect(self.download_err_slot)
# finished slot
def finished_slot(self):
# remove progress bar busy indication
self.progress_bar.setRange(0, 100)
# unset fetching flag
self.isFetching = False
# response slot
def response_slot(self, res):
# set back the button text
self.button.setText('Get')
# save the yt object for speeding up download
self.yt = res[0]
# set the actual thumbnail of requested video
self.thumb.setPixmap(res[1])
# slice the title if it is more than the limit
if len(res[2]) > 50:
self.title.setText(f'Title: {res[2][:50]}...')
else:
self.title.setText(f'Title: {res[2]}')
# set leftover details
self.author.setText(f'Author: {res[3]}')
self.length.setText(f'Duration: {timedelta(seconds=res[4])}')
self.publish_date.setText(f'Published: {res[5].strftime("%d/%m/%Y")}')
# clear any previous items if any
self.download.clear()
# add resolutions as items to the download button and enable them
self.download.addItems([item for item in res[6]])
self.download.setDisabled(False)
self.download_audio.setDisabled(False)
# error slot
def err_slot(self):
# show the warning message
self.message.warning(
self,
'Warning',
'Something went wrong!\nProbably a broken link or some restricted content... '
)
# set back the button text
self.button.setText('Get')
# download finished slot
def download_finished_slot(self):
# set back the button text
self.button.setText('Get')
# now enable the download options
self.download.setDisabled(False)
self.download_audio.setDisabled(False)
# unset downloading flag
self.isDownloading = False
# reset pogress bar
self.progress_bar.reset()
# download response slot
def download_response_slot(self, per):
# update progress bar
self.progress_bar.setValue(per)
# adjust the font color to maintain the contrast
if per > 52:
self.progress_bar.setStyleSheet('QProgressBar { color: #fff }')
else:
self.progress_bar.setStyleSheet('QProgressBar { color: #000 }')
# download complete slot
def download_complete_slot(self, location):
# use native separators
location = QDir.toNativeSeparators(location)
# show the success message
if self.message.information(
self,
'Downloaded',
f'Download complete!\nFile was successfully downloaded to :\n{location}\n\nOpen the downloaded file now ?',
QMessageBox.StandardButtons.Open,
QMessageBox.StandardButtons.Cancel
) is QMessageBox.StandardButtons.Open: subprocess.Popen(f'explorer /select,{location}')
# download error slot
def download_err_slot(self):
# show the error message
self.message.critical(
self,
'Error',
'Error!\nSomething unusual happened and was unable to download...'
)
class B23Download(QWidget):
def __init__(self):
super(B23Download, self).__init__()
# setup some flags
self.is_fetching = False
self.is_downloading = False
# default output path
basepath = os.path.dirname(os.path.abspath(__file__))
path = os.path.join(basepath, "videos")
self.output_path = path
# setup some window specific things
self.setWindowTitle("Bilibili Favorite Downloader")
self.setWindowIcon(QIcon("images/icon_bilibili.ico"))
self.setFixedSize(705, 343)
# parent layout
main_layout = QVBoxLayout()
main_layout.setContentsMargins(15, 15, 15, 10)
self.setLayout(main_layout)
# top bar layout
top_layout = QHBoxLayout()
# detail section
mid_main_layout = QHBoxLayout()
mid_right_layout = QVBoxLayout()
# download section
bottom_main_layout = QHBoxLayout()
bottom_right_layout = QVBoxLayout()
# output path link button
self.output_btn = QPushButton("📂 Output Path")
self.output_btn.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.output_btn.setToolTip(self.output_path)
self.output_btn.clicked.connect(self.set_output_path)
# status bar
self.status_bar = QStatusBar()
# message box
self.message_box = QMessageBox()
# setting up widgets
self.url_edit = QLineEdit()
self.url_edit.setPlaceholderText("🔍 Enter or paste favorite URL...")
self.get_btn = QPushButton("Get")
self.get_btn.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.get_btn.clicked.connect(self.get_details)
# thumbnail
pixmap = QPixmap("images/placeholder.png")
self.thumb = QLabel()
self.thumb.setFixedSize(250, 141)
self.thumb.setScaledContents(True)
self.thumb.setPixmap(pixmap)
# detail widgets
self.title = QLabel("Title: ")
self.author = QLabel("Author: ")
self.length = QLabel("Videos: ")
self.publish_date = QLabel("Published: ")
# progress bar
self.progress_bar = QProgressBar()
# download options
self.download_btn = QPushButton(" Download Videos ")
self.download_btn.setCursor(QCursor(Qt.CursorShape.PointingHandCursor))
self.download_btn.clicked.connect(self.get_content)
self.download_btn.setEnabled(False)
self.download_btn.setShortcut("Ctrl+Return")
self.download_btn.setMinimumWidth(200)
# add widgets and layouts
top_layout.addWidget(self.url_edit)
top_layout.addWidget(self.get_btn)
# detail section
mid_right_layout.addWidget(self.title)
mid_right_layout.addWidget(self.author)
mid_right_layout.addWidget(self.length)
mid_right_layout.addWidget(self.publish_date)
mid_main_layout.addWidget(self.thumb)
mid_main_layout.addSpacing(20)
mid_main_layout.addLayout(mid_right_layout)
# download section
bottom_right_layout.addWidget(self.download_btn)
bottom_main_layout.addWidget(self.progress_bar)
bottom_main_layout.addSpacing(10)
bottom_main_layout.addLayout(bottom_right_layout)
# status bar
self.status_bar.setSizeGripEnabled(False)
self.status_bar.addPermanentWidget(self.output_btn)
# add content to parent layout
main_layout.addLayout(top_layout)
main_layout.addSpacing(20)
main_layout.addLayout(mid_main_layout)
main_layout.addSpacing(5)
main_layout.addLayout(bottom_main_layout)
main_layout.addWidget(self.status_bar)
# set output path slot
def set_output_path(self):
# update the output path
path = str(
QFileDialog.getExistingDirectory(self, "Select Output Directory"))
if path:
self.output_path = path
# update tooltip
self.output_btn.setToolTip(path)
# get button slot
def get_details(self):
text = self.url_edit.text().strip()
if not text:
return
if text.find("fid") < 0:
self.message_box.warning(
self,
"Error",
("Input a correct favorite URL!\n"
"For example: https://space.bilibili.com/xxx/favlist?fid=xxx..."
),
)
return
if self.get_btn.text() == "Get":
self.get_btn.setText("Stop")
# indicate progress bar as busy
self.progress_bar.setRange(0, 0)
# set fetching flag
self.is_fetching = True
# setup a worker thread to keep UI responsive
self.media_id = text.split("fid=")[-1].split("&")[0]
self.worker = WorkerThread(self.media_id)
self.worker.start()
# catch the finished signal
self.worker.finished.connect(self.finished_slot)
# catch the response signal
self.worker.worker_response.connect(self.response_slot)
# catch the error signal
self.worker.worker_err_response.connect(self.err_slot)
elif self.get_btn.text() == "Stop":
if self.is_fetching:
# stop worker thread
self.worker.terminate()
# set back the get_btn text
self.get_btn.setText("Get")
elif self.is_downloading:
# stop download thread
self.download_thread.terminate()
# show the warning message_box
self.message_box.information(
self,
"Interrupted",
"Download interrupted!\nThe process was aborted while the file was being downloaded... ",
)
# reset progress bar
self.progress_bar.reset()
# download options slot
def get_content(self):
if self.is_fetching:
# show the warning message
self.message_box.critical(
self,
"Error",
"Please wait!\nWait while the details are being fetched... ",
)
else:
# disable the download options
self.download_btn.setDisabled(True)
# set downloading flag
self.is_downloading = True
# set button to stop
self.get_btn.setText("Stop")
self.download_thread = DownloadThread(
self.media_id,
self.media_counts,
self.first_page_medias,
self.output_path,
)
# start the thread
self.download_thread.start()
# catch the finished signal
self.download_thread.finished.connect(self.download_finished_slot)
# catch the response signal
self.download_thread.download_response.connect(
self.download_response_slot)
# catch the complete signal
self.download_thread.download_complete.connect(
self.download_complete_slot)
# catch the error signal
self.download_thread.download_err.connect(self.download_err_slot)
# handling enter key for get/stop button
def keyPressEvent(self, event):
self.url_edit.setFocus()
if (event.key() == Qt.Key.Key_Enter.value
or event.key() == Qt.Key.Key_Return.value):
self.get_details()
# finished slot
def finished_slot(self):
# remove progress bar busy indication
self.progress_bar.setRange(0, 100)
# unset fetching flag
self.is_fetching = False
# response slot
def response_slot(self, res):
# set back the button text
self.get_btn.setText("Get")
# set the actual thumbnail of requested video
self.thumb.setPixmap(res.thumb_img)
# slice the title if it is more than the limit
if len(res.title) > 50:
self.title.setText(f"Title: {res.title[:50]}...")
else:
self.title.setText(f"Title: {res.title}")
# cache first page medias
self.first_page_medias = res.medias
self.media_counts = res.media_counts
# set leftover details
self.author.setText(f"Author: {res.author}")
self.length.setText(f"Videos: {res.media_counts}")
self.publish_date.setText(
f'Published: {time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(res.publish_date))}'
)
self.download_btn.setDisabled(False)
# error slot
def err_slot(self):
# show the warning message
self.message_box.warning(
self,
"Warning",
"Something went wrong!\nProbably a broken link or some restricted content... ",
)
# set back the button text
self.get_btn.setText("Get")
# download finished slot
def download_finished_slot(self):
# set back the button text
self.get_btn.setText("Get")
# now enable the download options
self.download_btn.setDisabled(False)
# unset downloading flag
self.is_downloading = False
# reset pogress bar
self.progress_bar.reset()
# download response slot
def download_response_slot(self, per):
# update progress bar
self.progress_bar.setValue(per)
# adjust the font color to maintain the contrast
if per > 52:
self.progress_bar.setStyleSheet("QProgressBar { color: #fff }")
else:
self.progress_bar.setStyleSheet("QProgressBar { color: #000 }")
# download complete slot
def download_complete_slot(self, location):
# use native separators
location = QDir.toNativeSeparators(location)
# show the success message
if (self.message_box.information(
self,
"Downloaded",
f"Download complete!\nFile was successfully downloaded to :\n{location}\n\nOpen the downloaded file now ?",
QMessageBox.StandardButtons.Open,
QMessageBox.StandardButtons.Cancel,
) is QMessageBox.StandardButtons.Open):
subprocess.Popen(f"explorer /select,{location}")
# download error slot
def download_err_slot(self):
# show the error message
self.message_box.critical(
self,
"Error",
"Error!\nSomething unusual happened and was unable to download...",
)
def snapshot_load_failed(self):
QMessageBox.warning(self, "Load failed", "Failed to load backups.")
self.reset_gui_state()
def get_current_configuration(self) -> Union[Dict[str, Any], None]:
if len(self.header_name_input.text()) == 0:
QMessageBox.critical(self, "Input Data Error",
"Header Name cannot be Empty!",
QMessageBox.StandardButton.Ok,
QMessageBox.StandardButton.Ok)
return None
if len(self.header_output_path_input.text()) == 0:
QMessageBox.critical(self, "Input Data Error",
"Header Output Path cannot be Empty!",
QMessageBox.StandardButton.Ok,
QMessageBox.StandardButton.Ok)
return None
if self.header_name_input.text()[self.header_name_input.text().find('.'):].lower() \
not in ['.h', '.hh', '.hpp']:
# noinspection PyTypeChecker
if QMessageBox.StandardButton.Yes != QMessageBox.warning(
self, "Header Extension Unknown", """Header extension is
not .h, .hpp or .hh, continue?""", QMessageBox.StandardButton.Yes
| QMessageBox.StandardButton.No,
QMessageBox.StandardButton.No):
return None
name_violations: [int] = []
for i in range(len(self.warnings)):
if not self.warnings[i]['Name']:
name_violations.append(i)
if name_violations:
as_string = ''
for i in name_violations:
as_string += ' ' + str(i + 1) + '\n'
if as_string:
as_string = as_string[:-1]
QMessageBox.critical(
self, "Unnamed Warning Suppression",
"The items at the following rows are not named:\n" + as_string,
QMessageBox.StandardButton.Ok, QMessageBox.StandardButton.Ok)
return None
type_violations: Dict[str, ([str], str)] = {}
final_values: List[Dict[str, Union[str, int]]] = []
for platform in self.platforms:
for i in range(len(self.warnings)):
while i >= len(final_values):
final_values.append({'Name': self.warnings[i]['Name']})
if platform.warning_type.value == Platform.WarningType.INTEGER:
try:
if len(self.warnings[i][platform.platform_name]) == 0:
continue
final_values[i][platform.platform_name] = int(
self.warnings[i][platform.platform_name])
except ValueError:
if platform.platform_name not in type_violations.keys(
):
type_violations[platform.platform_name] = (
[], 'Integer')
type_violations[platform.platform_name][0].append(
self.warnings[i]['Name'])
if platform.warning_type.value == Platform.WarningType.COMMENT:
try:
if len(self.warnings[i][platform.platform_name]) == 0:
continue
final_values[i][platform.platform_name] = str(
self.warnings[i][platform.platform_name])
except ValueError:
if platform.platform_name not in type_violations.keys(
):
type_violations[platform.platform_name] = (
[], 'Comment')
type_violations[platform.platform_name][0].append(
self.warnings[i]['Name'])
if type_violations.keys():
all_str: str = ''
for key in type_violations.keys():
current_str = f"{len(type_violations[key][0])} '{key}' platform violations, of type" \
f" '{type_violations[key][1]}':\n"
for name in type_violations[key][0]:
current_str += f' {name}\n'
all_str += current_str + '\n'
QMessageBox.critical(
self, "Type Validation Error",
"The following Warning Suppression Fields have type "
"validation errors:\n" + all_str,
QMessageBox.StandardButton.Ok, QMessageBox.StandardButton.Ok)
return None
return self.get_current_configuration_from_values(final_values)
