mirror of
https://github.com/StuffAnThings/qbit_manage.git
synced 2025-10-19 10:16:56 +08:00
102 lines
4.3 KiB
Python
102 lines
4.3 KiB
Python
import os
|
|
from fnmatch import fnmatch
|
|
|
|
from modules import util
|
|
|
|
logger = util.logger
|
|
|
|
|
|
class RemoveOrphaned:
|
|
def __init__(self, qbit_manager):
|
|
self.qbt = qbit_manager
|
|
self.config = qbit_manager.config
|
|
self.client = qbit_manager.client
|
|
self.stats = 0
|
|
|
|
self.remote_dir = qbit_manager.config.remote_dir
|
|
self.root_dir = qbit_manager.config.root_dir
|
|
self.orphaned_dir = qbit_manager.config.orphaned_dir
|
|
|
|
self.rem_orphaned()
|
|
|
|
def rem_orphaned(self):
|
|
"""Remove orphaned files from remote directory"""
|
|
self.stats = 0
|
|
logger.separator("Checking for Orphaned Files", space=False, border=False)
|
|
torrent_files = []
|
|
root_files = []
|
|
orphaned_files = []
|
|
excluded_orphan_files = []
|
|
orphaned_parent_path = set()
|
|
|
|
if self.remote_dir != self.root_dir:
|
|
root_files = [
|
|
os.path.join(path.replace(self.remote_dir, self.root_dir), name)
|
|
for path, subdirs, files in os.walk(self.remote_dir)
|
|
for name in files
|
|
if self.orphaned_dir.replace(self.remote_dir, self.root_dir) not in path
|
|
]
|
|
else:
|
|
root_files = [
|
|
os.path.join(path, name)
|
|
for path, subdirs, files in os.walk(self.root_dir)
|
|
for name in files
|
|
if self.orphaned_dir.replace(self.root_dir, self.remote_dir) not in path
|
|
]
|
|
|
|
# Get an updated list of torrents
|
|
torrent_list = self.qbt.get_torrents({"sort": "added_on"})
|
|
for torrent in torrent_list:
|
|
for file in torrent.files:
|
|
fullpath = os.path.join(torrent.save_path, file.name)
|
|
# Replace fullpath with \\ if qbm is running in docker (linux) but qbt is on windows
|
|
fullpath = fullpath.replace(r"/", "\\") if ":\\" in fullpath else fullpath
|
|
torrent_files.append(fullpath)
|
|
|
|
orphaned_files = set(root_files) - set(torrent_files)
|
|
orphaned_files = sorted(orphaned_files)
|
|
|
|
if self.config.orphaned["exclude_patterns"]:
|
|
exclude_patterns = self.config.orphaned["exclude_patterns"]
|
|
excluded_orphan_files = [
|
|
file
|
|
for file in orphaned_files
|
|
for exclude_pattern in exclude_patterns
|
|
if fnmatch(file, exclude_pattern.replace(self.remote_dir, self.root_dir))
|
|
]
|
|
|
|
orphaned_files = set(orphaned_files) - set(excluded_orphan_files)
|
|
|
|
if orphaned_files:
|
|
os.makedirs(self.orphaned_dir, exist_ok=True)
|
|
body = []
|
|
num_orphaned = len(orphaned_files)
|
|
logger.print_line(f"{num_orphaned} Orphaned files found", self.config.loglevel)
|
|
body += logger.print_line("\n".join(orphaned_files), self.config.loglevel)
|
|
body += logger.print_line(
|
|
f"{'Did not move' if self.config.dry_run else 'Moved'} {num_orphaned} Orphaned files "
|
|
f"to {self.orphaned_dir.replace(self.remote_dir,self.root_dir)}",
|
|
self.config.loglevel,
|
|
)
|
|
|
|
attr = {
|
|
"function": "rem_orphaned",
|
|
"title": f"Removing {num_orphaned} Orphaned Files",
|
|
"body": "\n".join(body),
|
|
"orphaned_files": list(orphaned_files),
|
|
"orphaned_directory": self.orphaned_dir.replace(self.remote_dir, self.root_dir),
|
|
"total_orphaned_files": num_orphaned,
|
|
}
|
|
self.config.send_notifications(attr)
|
|
# Delete empty directories after moving orphan files
|
|
logger.info("Cleaning up any empty directories...")
|
|
if not self.config.dry_run:
|
|
for file in orphaned_files:
|
|
src = file.replace(self.root_dir, self.remote_dir)
|
|
dest = os.path.join(self.orphaned_dir, file.replace(self.root_dir, ""))
|
|
util.move_files(src, dest, True)
|
|
orphaned_parent_path.add(os.path.dirname(file).replace(self.root_dir, self.remote_dir))
|
|
for parent_path in orphaned_parent_path:
|
|
util.remove_empty_directories(parent_path, "**/*")
|
|
else:
|
|
logger.print_line("No Orphaned Files found.", self.config.loglevel)
|