import os from PyQt5.QtWidgets import (QMessageBox, QWidget, QApplication, QFileDialog, QLabel, QPushButton, QGridLayout, QGroupBox, QHBoxLayout, QVBoxLayout, QLineEdit) from PyQt5.QtCore import Qt from logline_leviathan.gui.checkbox_panel import * from logline_leviathan.gui.ui_helper import UIHelper from logline_leviathan.database.database_manager import session_scope from logline_leviathan.exporter.wordlist_export import generate_wordlist from logline_leviathan.gui.checkbox_panel import * import shutil import glob class DBBrowserWindow(QWidget): def __init__(self, app): super().__init__() self.app = app self.initialize_dbbrowser_window(app) def initialize_dbbdrowser_window(dbbrowser_window, app): dbbrowser_window.setWindowTitle('Logline Leviathan - Database-Browser') dbbrowser_window.mainLayout = QVBoxLayout(dbbrowser_window) #dbbrowser_window.extendedLayout = QHBoxLayout(dbbrowser_window) dbbrowser_window.db_session = None stylesheet = """ /* Style for the main window */ QWidget { background-color: #282C34; /* Dark grey background */ color: white; /* White text */ } /* Style for buttons */ QPushButton { background-color: #4B5563; /* Dark grey background */ color: white; /* White text */ border-style: outset; border-width: 2px; border-radius: 1px; /* Rounded corners */ border-color: #4A4A4A; padding: 6px; min-width: 50px; min-height: 15px; } QPushButton:hover { background-color: #6E6E6E; /* Slightly lighter grey on hover */ } QPushButton:pressed { background-color: #484848; /* Even darker grey when pressed */ } """ highlited_button_style = """ QPushButton { background-color: #3C8CCE; /* Lighter blue background */ color: white; /* White text */ border-style: outset; border-width: 2px; border-radius: 1px; /* Rounded corners */ border-color: #4A4A4A; padding: 6px; min-width: 50px; min-height: 15px; } QPushButton:hover { background-color: #7EC0EE; /* Even lighter blue on hover */ } QPushButton:pressed { background-color: #4A86E8; /* Slightly darker blue when pressed */ } """ dbbrowser_window.setStyleSheet(stylesheet) dbbrowser_window.statusLabel = QLabel(' Erwarte Selektion der Entitätentypen', dbbrowser_window) dbbrowser_window.statusLabel.setWordWrap(True) dbbrowser_window.statusLabel.setMinimumHeight(40) dbbrowser_window.statusLabel.setStyleSheet("QLabel { background-color: #3C4043; color: white; }") dbbrowser_window.mainLayout.addWidget(dbbrowser_window.statusLabel) # Create a GroupBox for the CheckboxPanel exportOptionsGroupBox = QGroupBox("SELEKTION", dbbrowser_window) exportOptionsLayout = QVBoxLayout(exportOptionsGroupBox) dbbrowser_window.checkboxPanel = CheckboxPanel() # Create a horizontal layout filterLayout = QHBoxLayout() # Create the "Check All" button checkAllButton = QPushButton("Alle markieren", dbbrowser_window) checkAllButton.clicked.connect(lambda: dbbrowser_window.checkboxPanel.checkAllVisible()) # Create the "Uncheck All" button uncheckAllButton = QPushButton("Keine markieren", dbbrowser_window) uncheckAllButton.clicked.connect(lambda: dbbrowser_window.checkboxPanel.uncheckAllVisible()) expandAllButton = QPushButton("Expandieren", dbbrowser_window) expandAllButton.clicked.connect(lambda: dbbrowser_window.checkboxPanel.expandAllTreeItems()) collapseAllButton = QPushButton("Komprimieren", dbbrowser_window) collapseAllButton.clicked.connect(lambda: dbbrowser_window.checkboxPanel.collapseAllTreeItems()) # Add buttons to the filter layout, to the left of the filter label filterLayout.addWidget(checkAllButton) filterLayout.addWidget(uncheckAllButton) filterLayout.addWidget(expandAllButton) filterLayout.addWidget(collapseAllButton) # Create the label for the filter filterLabel = QLabel("Filtern:") filterLayout.addWidget(filterLabel) # Add label to the horizontal layout # Add Text Input for Filtering filterLineEdit = QLineEdit(dbbrowser_window) filterLineEdit.setPlaceholderText(" nach Typ, Tooltip oder Kurzbezeichnung filtern...") filterLineEdit.setStyleSheet(""" QLineEdit { background-color: #3C4043; /* Background color */ color: white; /* Text color */ min-height: 20px; } """) filterLayout.addWidget(filterLineEdit) # Add line edit to the horizontal layout exportOptionsLayout.addLayout(filterLayout) # Add the horizontal layout to the export options layout # Add CheckboxPanel to the GroupBox's Layout exportOptionsLayout.addWidget(dbbrowser_window.checkboxPanel) # Connect the textChanged signal of QLineEdit to a new method filterLineEdit.textChanged.connect(dbbrowser_window.checkboxPanel.filterCheckboxes) dbbrowser_window.mainLayout.addWidget(exportOptionsGroupBox) copyWordlistToParserDirButton = QPushButton('Soeben generierte Wordlist zur Analyse hinzufügen (kopiert erzeugte Datei)', dbbrowser_window) copyWordlistToParserDirButton.clicked.connect(dbbrowser_window.copyWordlistToParserDir) dbbrowser_window.mainLayout.addWidget(copyWordlistToParserDirButton) # Exit Button Layout bottomLayout = QGridLayout() dbbrowser_window.openWordlistPathButton = QPushButton('Ziel-Dateipfad...', dbbrowser_window) dbbrowser_window.openWordlistPathButton.clicked.connect(dbbrowser_window.openWordlistPath) bottomLayout.addWidget(dbbrowser_window.openWordlistPathButton, 1, 1) # Start Export Button dbbrowser_window.startExportButton = QPushButton('Wordlist erstellen', dbbrowser_window) dbbrowser_window.startExportButton.clicked.connect(dbbrowser_window.start_export_process) dbbrowser_window.startExportButton.setStyleSheet(highlited_button_style) bottomLayout.addWidget(dbbrowser_window.startExportButton, 1, 2) # Output File Directory dbbrowser_window.selectOutputFileButton = QPushButton('Zieldateipfad setzen...', dbbrowser_window) dbbrowser_window.selectOutputFileButton.clicked.connect(dbbrowser_window.selectOutputFile) bottomLayout.addWidget(dbbrowser_window.selectOutputFileButton, 2, 1) # Exit Button dbbrowser_window.exitButton = QPushButton('Schließen', dbbrowser_window) dbbrowser_window.exitButton.clicked.connect(dbbrowser_window.close) bottomLayout.addWidget(dbbrowser_window.exitButton, 2, 2) dbbrowser_window.crossmatchesCheckbox = QCheckBox('Nur Kreuztreffer (Entitäten, die in mehreren Dateien vorkommen)', dbbrowser_window) bottomLayout.addWidget(dbbrowser_window.crossmatchesCheckbox, 0, 1) # Output File Path Label dbbrowser_window.WordlistPathLabel = QLabel('', dbbrowser_window) dbbrowser_window.updateWordlistPathLabel() # Call this method to set the initial text bottomLayout.addWidget(dbbrowser_window.WordlistPathLabel, 0, 2) dbbrowser_window.mainLayout.addLayout(bottomLayout) dbbrowser_window.setLayout(dbbrowser_window.mainLayout) def updateCheckboxes(self): with session_scope() as session: self.checkboxPanel.updateCheckboxes(session) def getSelectedCheckboxes(self): selected_checkboxes = [] def traverseTreeItems(treeItem): if treeItem.checkState(0) == Qt.Checked: selected_checkboxes.append(treeItem) for i in range(treeItem.childCount()): traverseTreeItems(treeItem.child(i)) for i in range(self.checkboxPanel.treeWidget.topLevelItemCount()): traverseTreeItems(self.checkboxPanel.treeWidget.topLevelItem(i)) return selected_checkboxes def updateWordlistPathLabel(self): outputDirPath = os.path.dirname(self.WordlistPath) display_text = f'{outputDirPath}/' self.WordlistPathLabel.setText(display_text) def openWordlistPath(self): outputDirPath = os.path.dirname(self.WordlistPath) wordlistPath = os.path.join(outputDirPath, 'wordlist') self.ui_helper.openFile(wordlistPath) def selectOutputFile(self): options = QFileDialog.Options() output_format = self.outputFormatList.currentItem().text().lower() extension_map = {'html': '.html', 'xlsx': '.xlsx'} default_extension = extension_map.get(output_format, '') selected_file, _ = QFileDialog.getSaveFileName( self, "Selektieren des Ziel-Dateipfads", self.WordlistPath, f"{output_format.upper()} Files (*{default_extension});;All Files (*)", options=options ) if selected_file: if not selected_file.endswith(default_extension): selected_file += default_extension self.WordlistPath = selected_file self.outputDir = os.path.dirname(selected_file) self.updateWordlistPathLabel() def get_unique_filename(self, base_path): directory, filename = os.path.split(base_path) name, extension = os.path.splitext(filename) counter = 1 new_path = base_path while os.path.exists(new_path): new_filename = f"{name}_{counter}{extension}" new_path = os.path.join(directory, new_filename) counter += 1 return new_path def copyWordlistToParserDir(self): try: # Path to the parser directory parser_dir = os.path.join(os.getcwd(), 'data', 'parser') # Ensure the parser directory exists os.makedirs(parser_dir, exist_ok=True) # Find the newest .txt file in the WordlistPath directory list_of_files = glob.glob(os.path.join(self.WordlistPath, '*.txt')) if not list_of_files: raise FileNotFoundError("No .txt files found in the WordlistPath directory.") newest_file = max(list_of_files, key=os.path.getctime) # Destination file path destination_file = os.path.join(parser_dir, 'generated_wordlist.txt') # Copy and overwrite the newest file to the destination shutil.copy2(newest_file, destination_file) self.statusLabel.setText(f" Wordlist erfolgreich kopiert nach {destination_file}") except Exception as e: self.message("Fehler beim kopieren", f"Fehler beim kopieren: {str(e)}") def start_export_process(self): # Base filename for the wordlist file base_filename = "wordlist.txt" # Construct the full path with the base filename full_output_path = os.path.join(self.WordlistPath, base_filename) # Generate a unique filename to avoid overwriting existing files unique_output_path = self.get_unique_filename(full_output_path) try: with session_scope() as session: selected_checkboxes = self.getSelectedCheckboxes() if not selected_checkboxes: self.message("Generieren nicht möglich", "Keine Typen selektiert. Auswahl vornehmen.") return only_crossmatches = self.crossmatchesCheckbox.isChecked() generate_wordlist(unique_output_path, session, selected_checkboxes, only_crossmatches) self.statusLabel.setText(f" Generierte Liste gespeichert unter {unique_output_path}") except Exception as e: self.statusLabel.setText(f" Fehler beim speichern: {str(e)}") logging.error(f"Export Error: {str(e)}") def message(self, title, text, extra_widget=None): msgBox = QMessageBox() msgBox.setStyleSheet(""" QMessageBox { background-color: #282C34; /* Dark grey background */ } QLabel { color: white; /* White text */ } QPushButton { color: white; /* White text for buttons */ background-color: #4B5563; /* Dark grey background for buttons */ border-style: solid; border-width: 2px; border-radius: 5px; border-color: #4A4A4A; padding: 6px; min-width: 80px; min-height: 30px; } """) msgBox.setIcon(QMessageBox.Warning) msgBox.setWindowTitle(title) msgBox.setText(text) if extra_widget: msgBox.setInformativeText('') msgBox.layout().addWidget(extra_widget, 1, 1) msgBox.exec_()