58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
from PyQt5 import QtCore
|
|
|
|
import sorter
|
|
import database
|
|
|
|
|
|
class BackGroundTabUpdate(QtCore.QThread):
|
|
def __init__(self, database_path, all_metadata, parent=None):
|
|
super(BackGroundTabUpdate, self).__init__(parent)
|
|
self.database_path = database_path
|
|
self.all_metadata = all_metadata
|
|
|
|
def run(self):
|
|
hash_position_pairs = []
|
|
for i in self.all_metadata:
|
|
file_hash = i['hash']
|
|
position = i['position']
|
|
hash_position_pairs.append([file_hash, position])
|
|
|
|
database.DatabaseFunctions(
|
|
self.database_path).modify_position(hash_position_pairs)
|
|
|
|
|
|
class BackGroundBookAddition(QtCore.QThread):
|
|
def __init__(self, parent_window, file_list, database_path, parent=None):
|
|
super(BackGroundBookAddition, self).__init__(parent)
|
|
self.parent_window = parent_window
|
|
self.file_list = file_list
|
|
self.database_path = database_path
|
|
|
|
def run(self):
|
|
books = sorter.BookSorter(
|
|
self.file_list,
|
|
'addition',
|
|
self.database_path)
|
|
parsed_books = books.initiate_threads()
|
|
database.DatabaseFunctions(self.database_path).add_to_database(parsed_books)
|
|
self.parent_window.lib_ref.generate_model('addition', parsed_books)
|
|
|
|
|
|
class BackGroundBookSearch(QtCore.QThread):
|
|
def __init__(self, parent_window, root_directory, parent=None):
|
|
super(BackGroundBookSearch, self).__init__(parent)
|
|
self.parent_window = parent_window
|
|
self.root_directory = root_directory
|
|
self.valid_files = []
|
|
|
|
def run(self):
|
|
for directory, subdir, files in os.walk(self.root_directory):
|
|
for filename in files:
|
|
if os.path.splitext(filename)[1][1:] in sorter.available_parsers:
|
|
self.valid_files.append(os.path.join(directory, filename))
|
|
|
|
print(self.valid_files)
|