Merge pull request #43 from thsrite/main
This commit is contained in:
jxxghp
GitHub
commit
2d79044350
@ -36,3 +36,9 @@ class DownloadHistoryOper(DbOper):
|
||||
分页查询下载历史
|
||||
"""
|
||||
return DownloadHistory.list_by_page(self._db, page, count)
|
||||
|
||||
def truncate(self):
|
||||
"""
|
||||
清空转移记录
|
||||
"""
|
||||
DownloadHistory.truncate(self._db)
|
||||
|
||||
@ -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)
|
||||
|
||||
@ -187,7 +187,7 @@ class ChineseSubFinder(_PluginBase):
|
||||
"""
|
||||
调用ChineseSubFinder下载字幕
|
||||
"""
|
||||
if not self._host or not self._api_key:
|
||||
if not self._enabled or not self._host or not self._api_key:
|
||||
return
|
||||
item = event.event_data
|
||||
if not item:
|
||||
|
||||
@ -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,28 +232,73 @@ 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
|
||||
sql = '''
|
||||
SELECT
|
||||
t.SOURCE_PATH || '/' || t.SOURCE_FILENAME AS src,
|
||||
t.DEST_PATH || '/' || t.DEST_FILENAME AS dest,
|
||||
CASE
|
||||
@ -163,10 +335,10 @@ class NAStoolSync(_PluginBase):
|
||||
FROM
|
||||
TRANSFER_HISTORY t
|
||||
LEFT JOIN ( SELECT * FROM DOWNLOAD_HISTORY GROUP BY TMDBID ) d ON t.TITLE = d.TITLE
|
||||
AND t.TYPE = d.TYPE;'''
|
||||
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]:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user