diff --git a/app/db/downloadhistory_oper.py b/app/db/downloadhistory_oper.py index 7b19d997..e02299a5 100644 --- a/app/db/downloadhistory_oper.py +++ b/app/db/downloadhistory_oper.py @@ -36,3 +36,9 @@ class DownloadHistoryOper(DbOper): 分页查询下载历史 """ return DownloadHistory.list_by_page(self._db, page, count) + + def truncate(self): + """ + 清空转移记录 + """ + DownloadHistory.truncate(self._db) diff --git a/app/db/plugindata_oper.py b/app/db/plugindata_oper.py index 0a799413..b3d864e5 100644 --- a/app/db/plugindata_oper.py +++ b/app/db/plugindata_oper.py @@ -43,3 +43,9 @@ class PluginDataOper(DbOper): if ObjectUtils.is_obj(data.value): return json.loads(data.value) return data.value + + def truncate(self): + """ + 清空转移记录 + """ + PluginData.truncate(self._db) diff --git a/app/plugins/nastoolsync/__init__.py b/app/plugins/nastoolsync/__init__.py index ee3ab8f1..5727e92b 100644 --- a/app/plugins/nastoolsync/__init__.py +++ b/app/plugins/nastoolsync/__init__.py @@ -1,6 +1,9 @@ +import os import sqlite3 from datetime import datetime +from app.db.downloadhistory_oper import DownloadHistoryOper +from app.db.plugindata_oper import PluginDataOper from app.db.transferhistory_oper import TransferHistoryOper from app.plugins import _PluginBase from typing import Any, List, Dict, Tuple @@ -11,7 +14,7 @@ class NAStoolSync(_PluginBase): # 插件名称 plugin_name = "历史记录同步" # 插件描述 - plugin_desc = "同步NAStool历史记录到MoviePilot。" + plugin_desc = "同步NAStool历史记录、下载记录、插件记录到MoviePilot。" # 插件图标 plugin_icon = "sync.png" # 主题色 @@ -31,29 +34,153 @@ class NAStoolSync(_PluginBase): # 私有属性 _transferhistory = None + _plugindata = None + _downloadhistory = None _clear = None _nt_db_path = None _path = None + _site = None + _transfer = None + _plugin = None + _download = None def init_plugin(self, config: dict = None): self._transferhistory = TransferHistoryOper() + self._plugindata = PluginDataOper() + self._downloadhistory = DownloadHistoryOper() if config: self._clear = config.get("clear") self._nt_db_path = config.get("nt_db_path") self._path = config.get("path") + self._site = config.get("site") + self._transfer = config.get("transfer") + self._plugin = config.get("plugin") + self._download = config.get("download") - if self._nt_db_path: - # 导入转移历史 - self.sync_transfer_history() + if self._nt_db_path and (self._transfer or self._plugin or self._download): + # 读取sqlite数据 + gradedb = sqlite3.connect(self._nt_db_path) + # 创建游标cursor来执行executeＳＱＬ语句 + cursor = gradedb.cursor() - def sync_transfer_history(self): + # 转移历史记录 + if self._transfer: + transfer_history = self.get_nt_transfer_history(cursor) + # 导入历史记录 + self.sync_transfer_history(transfer_history) + # 插件历史记录 + if self._plugin: + plugin_history = self.get_nt_plugin_history(cursor) + # 导入插件记录 + self.sync_plugin_history(plugin_history) + # 下载历史记录 + if self._download: + download_history = self.get_nt_download_history(cursor) + # 导入下载记录 + self.sync_download_history(download_history) + + # 关闭游标 + cursor.close() + + self.update_config( + { + "clear": False, + "nt_db_path": "", + "path": self._path, + "site": self._site, + "transfer": self._transfer, + "plugin": self._plugin, + "download": self._download, + } + ) + + def sync_plugin_history(self, plugin_history): + """ + 导入插件记录 + """ + # 开始计时 + start_time = datetime.now() + logger.info("开始同步NAStool插件历史记录到MoviePilot") + # 清空MoviePilot插件记录 + if self._clear: + logger.info("MoviePilot插件记录已清空") + self._plugindata.truncate() + + for history in plugin_history: + plugin_id = history[1] + plugin_key = history[2] + plugin_value = history[3] + + self._plugindata.save(plugin_id=plugin_id, + key=plugin_key, + value=plugin_value) + + # 计算耗时 + end_time = datetime.now() + + logger.info(f"插件记录已同步完成。总耗时 {(end_time - start_time).seconds} 秒") + + def sync_download_history(self, download_history): + """ + 导入下载记录 + """ + # 开始计时 + start_time = datetime.now() + logger.info("开始同步NAStool下载历史记录到MoviePilot") + # 清空MoviePilot下载记录 + if self._clear: + logger.info("MoviePilot下载记录已清空") + self._downloadhistory.truncate() + + for history in download_history: + mpath = history[0] + mtype = history[1] + mtitle = history[2] + myear = history[3] + mtmdbid = history[4] + mseasons = history[5] + mepisodes = history[6] + mimages = history[7] + mdownload_hash = history[8] + mtorrent = history[9] + mdesc = history[10] + msite = history[11] + + # 处理站点映射 + if self._site: + sites = self._site.split("\n") + for site in sites: + sub_sites = site.split(":") + if str(msite) == str(sub_sites[0]): + msite = str(sub_sites[1]) + + self._downloadhistory.add( + path=os.path.basename(mpath), + type=mtype, + title=mtitle, + year=myear, + tmdbid=mtmdbid, + seasons=mseasons, + episodes=mepisodes, + image=mimages, + download_hash=mdownload_hash, + torrent_name=mtorrent, + torrent_description=mdesc, + torrent_site=msite + ) + + # 计算耗时 + end_time = datetime.now() + + logger.info(f"下载记录已同步完成。总耗时 {(end_time - start_time).seconds} 秒") + + def sync_transfer_history(self, transfer_history): """ 导入nt转移记录 """ # 开始计时 start_time = datetime.now() - - nt_historys = self.get_nt_transfer_history() + logger.info("开始同步NAStool转移历史记录到MoviePilot") # 清空MoviePilot转移记录 if self._clear: @@ -61,7 +188,7 @@ class NAStoolSync(_PluginBase): self._transferhistory.truncate() # 处理数据，存入mp数据库 - for history in nt_historys: + for history in transfer_history: msrc = history[0] mdest = history[1] mmode = history[2] @@ -105,68 +232,113 @@ class NAStoolSync(_PluginBase): ) logger.debug(f"{mtitle} {myear} {mtmdbid} {mseasons} {mepisodes} 已同步") - self.update_config( - { - "clear": False, - "nt_db_path": "", - "path": self._path - } - ) - # 计算耗时 end_time = datetime.now() logger.info(f"转移记录已同步完成。总耗时 {(end_time - start_time).seconds} 秒") - def get_nt_transfer_history(self): + def get_nt_plugin_history(self, cursor): + """ + 获取插件历史记录 + """ + sql = 'select * from PLUGIN_HISTORY;' + cursor.execute(sql) + plugin_history = cursor.fetchall() + + if not plugin_history: + logger.error("未获取到NAStool数据库文件中的插件历史，请检查数据库路径是正确") + return + + logger.info(f"获取到NAStool插件记录 {len(plugin_history)} 条") + return plugin_history + + def get_nt_download_history(self, cursor): + """ + 获取下载历史记录 + """ + sql = ''' + SELECT + SAVE_PATH, + TYPE, + TITLE, + YEAR, + TMDBID, + CASE + SE + WHEN NULL THEN + NULL ELSE substr( SE, 1, instr ( SE, ' ' ) - 1 ) + END AS seasons, + CASE + SE + WHEN NULL THEN + NULL ELSE substr( SE, instr ( SE, ' ' ) + 1 ) + END AS episodes, + POSTER, + DOWNLOAD_ID, + TORRENT, + DESC, + SITE + FROM + DOWNLOAD_HISTORY + WHERE + SAVE_PATH IS NOT NULL; + ''' + cursor.execute(sql) + download_history = cursor.fetchall() + + if not download_history: + logger.error("未获取到NAStool数据库文件中的下载历史，请检查数据库路径是正确") + return + + logger.info(f"获取到NAStool下载记录 {len(download_history)} 条") + return download_history + + def get_nt_transfer_history(self, cursor): """ 获取nt转移记录 """ - # 读取sqlite数据 - gradedb = sqlite3.connect(self._nt_db_path) - # 创建游标cursor来执行executeＳＱＬ语句 - cursor = gradedb.cursor() - sql = '''SELECT - t.SOURCE_PATH || '/' || t.SOURCE_FILENAME AS src, - t.DEST_PATH || '/' || t.DEST_FILENAME AS dest, - CASE - t.MODE - WHEN '硬链接' THEN - 'link' - WHEN '移动' THEN - 'move' - WHEN '复制' THEN - 'copy' - END AS mode, - CASE - t.TYPE - WHEN '动漫' THEN - '电视剧' ELSE t.TYPE - END AS type, - t.CATEGORY AS category, - t.TITLE AS title, - t.YEAR AS year, - t.TMDBID AS tmdbid, - CASE - t.SEASON_EPISODE - WHEN NULL THEN - NULL ELSE substr( t.SEASON_EPISODE, 1, instr ( t.SEASON_EPISODE, ' ' ) - 1 ) - END AS seasons, - CASE - t.SEASON_EPISODE - WHEN NULL THEN - NULL ELSE substr( t.SEASON_EPISODE, instr ( t.SEASON_EPISODE, ' ' ) + 1 ) - END AS episodes, - d.POSTER AS image, - d.DOWNLOAD_ID AS download_hash, - t.DATE AS date - FROM - TRANSFER_HISTORY t - LEFT JOIN ( SELECT * FROM DOWNLOAD_HISTORY GROUP BY TMDBID ) d ON t.TITLE = d.TITLE - AND t.TYPE = d.TYPE;''' + sql = ''' + SELECT + t.SOURCE_PATH || '/' || t.SOURCE_FILENAME AS src, + t.DEST_PATH || '/' || t.DEST_FILENAME AS dest, + CASE + t.MODE + WHEN '硬链接' THEN + 'link' + WHEN '移动' THEN + 'move' + WHEN '复制' THEN + 'copy' + END AS mode, + CASE + t.TYPE + WHEN '动漫' THEN + '电视剧' ELSE t.TYPE + END AS type, + t.CATEGORY AS category, + t.TITLE AS title, + t.YEAR AS year, + t.TMDBID AS tmdbid, + CASE + t.SEASON_EPISODE + WHEN NULL THEN + NULL ELSE substr( t.SEASON_EPISODE, 1, instr ( t.SEASON_EPISODE, ' ' ) - 1 ) + END AS seasons, + CASE + t.SEASON_EPISODE + WHEN NULL THEN + NULL ELSE substr( t.SEASON_EPISODE, instr ( t.SEASON_EPISODE, ' ' ) + 1 ) + END AS episodes, + d.POSTER AS image, + d.DOWNLOAD_ID AS download_hash, + t.DATE AS date + FROM + TRANSFER_HISTORY t + LEFT JOIN ( SELECT * FROM DOWNLOAD_HISTORY GROUP BY TMDBID ) d ON t.TITLE = d.TITLE + AND t.TYPE = d.TYPE; + ''' cursor.execute(sql) nt_historys = cursor.fetchall() - cursor.close() if not nt_historys: logger.error("未获取到NAStool数据库文件中的转移历史，请检查数据库路径是正确") @@ -200,19 +372,66 @@ class NAStoolSync(_PluginBase): 'component': 'VCol', 'props': { 'cols': 12, - 'md': 6 + 'md': 3 }, 'content': [ { 'component': 'VSwitch', 'props': { 'model': 'clear', - 'label': '清空记录', - 'placeholder': '开启会清空MoviePilot历史记录' + 'label': '清空记录' } } ] - } + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 3 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'transfer', + 'label': '转移记录' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 3 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'plugin', + 'label': '插件记录' + } + } + ] + }, + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 3 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'download', + 'label': '下载记录' + } + } + ] + }, ] }, { @@ -248,6 +467,7 @@ class NAStoolSync(_PluginBase): 'component': 'VTextarea', 'props': { 'model': 'path', + 'rows': '2', 'label': '路径映射', 'placeholder': 'NAStool路径:MoviePilot路径（一行一个）' } @@ -256,12 +476,60 @@ class NAStoolSync(_PluginBase): } ] }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VTextarea', + 'props': { + 'model': 'site', + 'label': '站点映射', + 'placeholder': 'NAStool站点名:MoviePilot站点名（一行一个）' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VAlert', + 'props': { + 'text': '只有开启转移记录、插件记录、下载记录其中之一插件才会启用。' + '开启清空记录时，会在导入历史数据之前删除MoviePilot之前的记录。' + '如果转移记录很多，同步时间可能会长，' + '所以点击确定后页面没反应是正常现象，后台正在处理。' + '路径映射在同步转移记录时有效、站点映射在同步下载记录时有效。' + } + } + ] + } + ] + } ] } ], { "clear": False, + "transfer": False, + "plugin": False, + "download": False, "nt_db_path": "", "path": "", + "site": "", } def get_page(self) -> List[dict]: