feat: Add search engine selection functionality

This commit is contained in:
2024-07-12 18:06:13 +02:00
parent d4ffd8864c
commit 777ca49c18

49
app.py
View File

@ -88,6 +88,17 @@ class URLManager(QWidget):
# Load settings
self.load_settings()
# Search engines
self.search_engines = {
"Bing": "https://www.bing.com/search?q=",
"Brave": "https://search.brave.com/search?q=",
"DuckDuckGo": "https://duckduckgo.com/?q=",
"Ecosia": "https://www.ecosia.org/search?q=",
"Google": "https://www.google.com/search?q=",
"Startpage": "https://www.startpage.com/do/search?q=",
"Swisscows": "https://swisscows.com/web?query="
}
def update_icon_color(self):
# Heuristic to check if the application is in dark mode
palette = self.palette()
@ -141,8 +152,8 @@ class URLManager(QWidget):
def search_url(self):
search_term = self.search_input.text()
if search_term:
search_engine = "https://www.google.com/search?q="
search_url = search_engine + search_term
search_engine_url = self.search_engines.get(self.search_engine, "https://www.google.com/search?q=")
search_url = search_engine_url + search_term
QMessageBox.information(self, "Search", f"Searching for: {search_term}\nURL: {search_url}")
self.search_input.clear()
@ -180,13 +191,15 @@ class URLManager(QWidget):
if os.path.getsize(settings_path) > 0:
with open(settings_path, 'r') as file:
settings = json.load(file)
# Use settings (for now, we only have the data_directory)
self.data_directory = settings.get('data_directory', 'data/')
self.search_engine = settings.get('search_engine', 'Google')
else:
self.data_directory = 'data/'
self.search_engine = 'Google'
self.save_settings()
else:
self.data_directory = 'data/'
self.search_engine = 'Google'
self.save_settings()
def save_settings(self):
@ -194,7 +207,8 @@ class URLManager(QWidget):
os.makedirs(os.path.dirname(settings_path), exist_ok=True)
with open(settings_path, 'w') as file:
settings = {
'data_directory': self.data_directory
'data_directory': self.data_directory,
'search_engine': self.search_engine
}
json.dump(settings, file)
@ -203,18 +217,35 @@ class SettingsDialog(QDialog):
def __init__(self, parent=None):
super().__init__(parent)
self.setWindowTitle("Settings")
self.setFixedWidth(360) # Set the width of the window to 360 pixels
self.parent = parent
layout = QVBoxLayout()
# File/Directory selection
form_layout = QFormLayout()
# Data Directory label
directory_label = QLabel("Data Directory:")
layout.addWidget(directory_label)
# File/Directory selection layout
directory_layout = QHBoxLayout()
self.directory_input = QLineEdit(self.parent.data_directory)
self.browse_button = QPushButton("Browse...")
self.browse_button.clicked.connect(self.browse_directory)
form_layout.addRow("Data Directory:", self.directory_input)
form_layout.addWidget(self.browse_button)
directory_layout.addWidget(self.directory_input)
directory_layout.addWidget(self.browse_button)
layout.addLayout(directory_layout)
# Add 4 pixels of empty space
layout.addSpacerItem(QSpacerItem(0, 4, QSizePolicy.Minimum, QSizePolicy.Fixed))
# Search engine selection
form_layout = QFormLayout()
self.search_engine_combobox = QComboBox()
search_engines = sorted(self.parent.search_engines.keys())
self.search_engine_combobox.addItems(search_engines)
self.search_engine_combobox.setCurrentText(self.parent.search_engine)
form_layout.addRow("Search Engine:", self.search_engine_combobox)
layout.addLayout(form_layout)
@ -232,10 +263,10 @@ class SettingsDialog(QDialog):
def save_settings(self):
self.parent.data_directory = self.directory_input.text()
self.parent.search_engine = self.search_engine_combobox.currentText()
self.parent.save_settings()
self.accept()
if __name__ == '__main__':
app = QApplication(sys.argv)
app.setWindowIcon(QIcon('assets/logo.png')) # Set the application icon using PNG for compatibility