125 lines
4.5 KiB
Python
125 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# This file is a part of Lector, a Qt based ebook reader
|
|
# Copyright (C) 2017 BasioMeusPuga
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import pathlib
|
|
from multiprocessing.dummy import Pool
|
|
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, file_list, database_path, prune_required, parent=None):
|
|
super(BackGroundBookAddition, self).__init__(parent)
|
|
self.file_list = file_list
|
|
self.parent = parent
|
|
self.database_path = database_path
|
|
self.prune_required = prune_required
|
|
|
|
def run(self):
|
|
books = sorter.BookSorter(
|
|
self.file_list,
|
|
'addition',
|
|
self.database_path,
|
|
self.parent.settings['auto_tags'])
|
|
parsed_books = books.initiate_threads()
|
|
self.parent.lib_ref.generate_model('addition', parsed_books)
|
|
if self.prune_required:
|
|
self.parent.lib_ref.prune_models(self.file_list)
|
|
database.DatabaseFunctions(self.database_path).add_to_database(parsed_books)
|
|
|
|
|
|
class BackGroundBookDeletion(QtCore.QThread):
|
|
def __init__(self, hash_list, database_path, parent=None):
|
|
super(BackGroundBookDeletion, self).__init__(parent)
|
|
self.parent = parent
|
|
self.hash_list = hash_list
|
|
self.database_path = database_path
|
|
|
|
def run(self):
|
|
database.DatabaseFunctions(
|
|
self.database_path).delete_from_database('Hash', self.hash_list)
|
|
|
|
|
|
class BackGroundBookSearch(QtCore.QThread):
|
|
# TODO
|
|
# Change existing sorter module functionality to handle preset tags
|
|
# Change database to accomodate User Tags, Folder Name, Folder Tags
|
|
|
|
def __init__(self, data_list, parent=None):
|
|
super(BackGroundBookSearch, self).__init__(parent)
|
|
self.parent = parent
|
|
self.valid_files = []
|
|
|
|
# Filter for checked directories
|
|
self.valid_directories = [
|
|
[i[0], i[1], i[2]] for i in data_list if i[3] == QtCore.Qt.Checked]
|
|
self.unwanted_directories = [
|
|
pathlib.Path(i[0]) for i in data_list if i[3] == QtCore.Qt.Unchecked]
|
|
|
|
def run(self):
|
|
|
|
def is_wanted(directory):
|
|
directory_parents = pathlib.Path(directory).parents
|
|
for i in self.unwanted_directories:
|
|
if i in directory_parents:
|
|
return False
|
|
return True
|
|
|
|
def traverse_directory(incoming_data):
|
|
root_directory = incoming_data[0]
|
|
folder_name = incoming_data[1]
|
|
folder_tags = incoming_data[2]
|
|
|
|
for directory, subdirs, files in os.walk(root_directory, topdown=True):
|
|
# Black magic fuckery
|
|
# Skip subdir tree in case it's not wanted
|
|
subdirs[:] = [d for d in subdirs if is_wanted(os.path.join(directory, d))]
|
|
for filename in files:
|
|
if os.path.splitext(filename)[1][1:] in sorter.available_parsers:
|
|
self.valid_files.append(
|
|
(os.path.join(directory, filename), folder_name, folder_tags))
|
|
|
|
def initiate_threads():
|
|
_pool = Pool(5)
|
|
_pool.map(traverse_directory, self.valid_directories)
|
|
_pool.close()
|
|
_pool.join()
|
|
|
|
initiate_threads()
|
|
print(len(self.valid_files), 'books found')
|