mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-03 05:29:12 +00:00
2646fa825b
- If a user manually edits the input file lines, clicking start should use that. Currently it just looks at the files selected last via file browser - We want to allow users to manually enter file paths in field. Which is why the field hasn't been set to read-only
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
# External Packages
|
|
from PyQt6 import QtWidgets
|
|
from PyQt6.QtCore import QDir
|
|
|
|
# Internal Packages
|
|
from src.utils.config import SearchType
|
|
|
|
|
|
class FileBrowser(QtWidgets.QWidget):
|
|
def __init__(self, title, search_type: SearchType=None, default_files=[]):
|
|
QtWidgets.QWidget.__init__(self)
|
|
layout = QtWidgets.QHBoxLayout()
|
|
self.setLayout(layout)
|
|
self.search_type = search_type
|
|
|
|
self.filter_name = self.getFileFilter(search_type)
|
|
self.dirpath = QDir.homePath()
|
|
|
|
self.label = QtWidgets.QLabel()
|
|
self.label.setText(title)
|
|
self.label.setFixedWidth(95)
|
|
layout.addWidget(self.label)
|
|
|
|
self.lineEdit = QtWidgets.QLineEdit(self)
|
|
self.lineEdit.setFixedWidth(180)
|
|
self.setFiles(default_files)
|
|
|
|
layout.addWidget(self.lineEdit)
|
|
|
|
self.button = QtWidgets.QPushButton('Add')
|
|
self.button.clicked.connect(self.storeFilesSelectedInFileDialog)
|
|
layout.addWidget(self.button)
|
|
layout.addStretch()
|
|
|
|
def getFileFilter(self, search_type):
|
|
if search_type == SearchType.Org:
|
|
return 'Org-Mode Files (*.org)'
|
|
elif search_type == SearchType.Ledger:
|
|
return 'Beancount Files (*.bean *.beancount)'
|
|
elif search_type == SearchType.Markdown:
|
|
return 'Markdown Files (*.md *.markdown)'
|
|
elif search_type == SearchType.Music:
|
|
return 'Org-Music Files (*.org)'
|
|
elif search_type == SearchType.Image:
|
|
return 'Images (*.jp[e]g)'
|
|
|
|
def storeFilesSelectedInFileDialog(self):
|
|
filepaths = []
|
|
if self.search_type == SearchType.Image:
|
|
filepaths.append(QtWidgets.QFileDialog.getExistingDirectory(self, caption='Choose Folder',
|
|
directory=self.dirpath))
|
|
else:
|
|
filepaths.extend(QtWidgets.QFileDialog.getOpenFileNames(self, caption='Choose Files',
|
|
directory=self.dirpath,
|
|
filter=self.filter_name)[0])
|
|
self.setFiles(filepaths)
|
|
|
|
def setFiles(self, paths):
|
|
self.filepaths = paths
|
|
if not self.filepaths or len(self.filepaths) == 0:
|
|
return
|
|
elif len(self.filepaths) == 1:
|
|
self.lineEdit.setText(self.filepaths[0])
|
|
else:
|
|
self.lineEdit.setText(",".join(self.filepaths))
|
|
|
|
def getPaths(self):
|
|
if self.lineEdit.text() == '':
|
|
return []
|
|
else:
|
|
return self.lineEdit.text().split(',')
|