def __init__(self):
super().__init__()
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.backend = Backend()
self.ui.scan_comments.setText(edocuments.config.get("scan_comments"))
default_index = 0
for s in edocuments.config.get("scans", []):
if s.get("default") is True:
default_index = self.ui.scan_type.count()
self.ui.scan_type.addItem(s.get("name"), s)
self.ui.scan_type.setCurrentIndex(default_index)
self.ui.scan_browse.clicked.connect(self.scan_browse)
self.ui.scan_to.returnPressed.connect(self.scan_start)
self.ui.scan_start.clicked.connect(self.scan_start)
self.ui.open.clicked.connect(self.open_selected)
self.ui.open_folder.clicked.connect(self.open_folder)
self.image_dialog = Dialog()
self.backend.scan_end.connect(self.end_scan)
self.backend.scan_error.connect(self.on_scan_error)
self.backend.update_library_progress.connect(
self.on_update_update_library_progress)
self.ui.search_text.textChanged.connect(self.search)
self.ui.search_result_list.itemSelectionChanged.connect(
self.selection_change)
self.ui.library_update.triggered.connect(self.update_library)
self.ui.library_optimize.triggered.connect(
self.backend.optimize_library)
self.ui.library_reset.triggered.connect(self.reset_library)
self.backend.process.progress.connect(self.on_progress)
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.backend = Backend()
self.ui.scan_comments.setText(edocuments.config.get("scan_comments"))
default_index = 0
for s in edocuments.config.get("scans", []):
if s.get("default") is True:
default_index = self.ui.scan_type.count()
self.ui.scan_type.addItem(s.get("name"), s)
self.ui.scan_type.setCurrentIndex(default_index)
self.ui.scan_browse.clicked.connect(self.scan_browse)
self.ui.scan_to.returnPressed.connect(self.scan_start)
self.ui.scan_start.clicked.connect(self.scan_start)
self.ui.open.clicked.connect(self.open_selected)
self.ui.open_folder.clicked.connect(self.open_folder)
self.image_dialog = Dialog()
self.backend.scan_end.connect(self.end_scan)
self.backend.scan_error.connect(self.on_scan_error)
self.backend.update_library_progress.connect(
self.on_update_update_library_progress)
self.ui.search_text.textChanged.connect(self.search)
self.ui.search_result_list.itemSelectionChanged.connect(
self.selection_change)
self.ui.library_update.triggered.connect(self.update_library)
self.ui.library_optimize.triggered.connect(
self.backend.optimize_library)
self.ui.library_reset.triggered.connect(self.reset_library)
self.backend.process.progress.connect(self.on_progress)
def open_selected(self):
item = self.ui.search_result_list.currentItem()
if item is not None:
cmd = edocuments.config.get('open_cmd').split(' ')
cmd.append(edocuments.long_path(item.result.get('path')))
call(cmd)
def open_folder(self):
item = self.ui.search_result_list.currentItem()
if item is not None:
cmd = edocuments.config.get('open_cmd').split(' ')
cmd.append(os.path.dirname(
edocuments.long_path(item.result.get('path'))))
call(cmd)
def selection_change(self):
item = self.ui.search_result_list.currentItem()
if item is not None:
self.ui.search_result_text.document().setHtml(
item.result.get('highlight'))
else:
self.ui.search_result_text.document().setHtml('')
def reset_library(self):
msg = QMessageBox(self)
msg.setWindowTitle("Reset the library...")
msg.setInformativeText("Are you sure to reset all you index?")
msg.setStandardButtons(QMessageBox.Ok | QMessageBox.Cancel)
if msg.exec() == QMessageBox.Ok:
index().clear()
def search(self, text):
model = self.ui.search_result_list.model()
model.removeRows(0, model.rowCount())
raw_results = index().search(self.ui.search_text.text())
dirs = dict([
(r.get('path'), -1)
for r in raw_results if r.get('directory')
])
results = {}
for index_, result in enumerate(raw_results):
path_ = result.get('path')
dir_ = os.path.dirname(path_)
if dir_ not in dirs:
results[path_] = [result, float(index_) / len(raw_results)]
else:
dirs[dir_] += 1
for dir_, count in dirs.items():
if dir_ in results:
results[dir_][1] += count
results = sorted(results.values(), key=lambda x: -x[1])
for result, count in results:
postfix = ' (%i)' % (count + 1) if result.get('directory') else ''
item = QListWidgetItem(
result['path'] + postfix,
self.ui.search_result_list
)
item.result = result
def scan_browse(self, event):
filename = QFileDialog.getSaveFileName(
self, "Scan to", directory=self.filename()
)[0]
filename = re.sub(r"\.[a-z0-9A-Z]{2,5}$", "", filename)
filename = edocuments.short_path(filename)
self.ui.scan_to.setText(filename)
def update_library(self):
self.update_library_progress = QProgressDialog(
"Update Library...", None, 0, 100, self)
self.update_library_progress.setWindowTitle('Updating the library...')
self.update_library_progress.setLabelText('Browsing the files...')
self.update_library_progress.setWindowModality(Qt.WindowModal)
self.update_library_progress.show()
t = Thread(target=self.backend.do_update_library)
t.start()
def on_update_update_library_progress(self, pos, text, status):
self.update_library_progress.setValue(pos)
self.update_library_progress.setLabelText(text)
self.statusBar().showMessage(status)
def filename(self):
return edocuments.long_path(self.ui.scan_to.text())
def scan_start(self, event=None):
if pathlib.Path(self.filename()).is_dir():
err = QErrorMessage(self)
err.setWindowTitle("eDocuments - Error")
err.showMessage("The destination is a directory!")
return
destination1, extension, _, _ = self.backend.process.destination_filename(
self.ui.scan_type.currentData().get("cmds"),
self.filename()
)
destination2, extension, _, _ = self.backend.process.destination_filename(
self.ui.scan_type.currentData().get("postprocess", {}),
destination1
)
if pathlib.Path(destination1).is_file() or pathlib.Path(destination2).is_file():
msg = QMessageBox(self)
msg.setWindowTitle("Scanning...")
msg.setText("The destination file already exists")
msg.setInformativeText("Do you want to overwrite it?")
msg.setStandardButtons(
QMessageBox.Ok | QMessageBox.Cancel | QMessageBox.Open)
ret = msg.exec()
if ret == QMessageBox.Ok:
if pathlib.Path(destination1).is_file():
os.unlink(destination1)
if pathlib.Path(destination2).is_file():
os.unlink(destination2)
self._scan()
elif ret == QMessageBox.Open:
if pathlib.Path(destination1).is_file():
cmd = edocuments.config.get('open_cmd').split(' ')
cmd.append(destination1)
call(cmd)
if pathlib.Path(destination2).is_file():
cmd = edocuments.config.get('open_cmd').split(' ')
cmd.append(destination2)
call(cmd)
else:
self._scan()
def _scan(self):
cmds = self.ui.scan_type.currentData().get("cmds")
self.progress = QProgressDialog(
"Scanning...", "Cancel", 0, len(cmds), self)
self.progress.setWindowTitle("Scanning...")
self.progress.setWindowModality(Qt.WindowModal)
self.progress.setLabelText('Scanning...')
self.progress.show()
t = Thread(
target=self.backend.do_scan,
args=[
self.filename(),
self.ui.scan_type.currentData().get("cmds"),
self.ui.scan_type.currentData().get("postprocess", []),
],
)
t.start()
def on_progress(self, no, name, cmd_cmd, cmd):
if self.progress is not None:
self.progress.setValue(no)
self.progress.setLabelText(cmd.get('display', ''))
if self.progress.wasCanceled() is True:
print("Cancel")
self.backend.process.cancel = True
self.statusBar().showMessage(cmd_cmd)
def end_scan(self, filename):
self.progress.hide()
self.image_dialog.set_image(filename)
self.image_dialog.exec()
self.ui.scan_to.setText(re.sub(
' ([0-9]{1,3})$',
lambda m: ' ' + str(int(m.group(1)) + 1),
self.ui.scan_to.text()
))
self.statusBar().showMessage('')
def on_scan_error(self, error):
print('Error: %s' % error)
err = QErrorMessage(self)
err.setWindowTitle("eDocuments - scan error")
err.showMessage(error)