codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
thsrite
2023-08-08 16:34:40 +08:00
parent 588089a079
commit 86e6122c86
1 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
app/plugins/nastoolsync/__init__.py
View File

@ -1,3 +1,4 @@
import json
import os import os
import sqlite3 import sqlite3
from datetime import datetime from datetime import datetime
@ -87,12 +88,13 @@ class NAStoolSync(_PluginBase):
self.update_config( self.update_config(
{ {
"clear": False, "clear": False,
"nt_db_path": "", "nt_db_path": self._nt_db_path,
"path": self._path, "path": self._path,
"downloader": self._downloader,
"site": self._site, "site": self._site,
"transfer": self._transfer, "transfer": False,
"plugin": self._plugin, "plugin": False,
"download": self._download, "download": False,
} }
) )
@ -130,19 +132,27 @@ class NAStoolSync(_PluginBase):
downloaders = self._downloader.split("\n") downloaders = self._downloader.split("\n")
for downloader in downloaders: for downloader in downloaders:
sub_downloaders = downloader.split(":") sub_downloaders = downloader.split(":")
# 转种记录 # 替换转种记录
if str(plugin_id) == "TorrentTransfer": if str(plugin_id) == "TorrentTransfer":
keys = str(plugin_key).split("-") keys = str(plugin_key).split("-")
if int(keys[0]) == int(sub_downloaders[0]): if keys[0].isdigit() and int(keys[0]) == int(sub_downloaders[0]):
# 替换key # 替换key
plugin_key = plugin_key.replace(keys[0], sub_downloaders[1]) plugin_key = plugin_key.replace(keys[0], sub_downloaders[1])
# 替换value # 替换value
if int(plugin_value.get("to_download")) == int(sub_downloaders[0]): if isinstance(plugin_value, str):
plugin_value = json.loads(plugin_value)
if str(plugin_value.get("to_download")).isdigit() and int(
plugin_value.get("to_download")) == int(sub_downloaders[0]):
plugin_value["to_download"] = sub_downloaders[1] plugin_value["to_download"] = sub_downloaders[1]
# 辅种记录
# 替换辅种记录
if str(plugin_id) == "IYUUAutoSeed": if str(plugin_id) == "IYUUAutoSeed":
if isinstance(plugin_value, str):
plugin_value = json.loads(plugin_value)
for value in plugin_value: for value in plugin_value:
if int(value.get("downloader")) == int(sub_downloaders[0]): if str(value.get("downloader")).isdigit() and int(value.get("downloader")) == int(
sub_downloaders[0]):
value["downloader"] = sub_downloaders[1] value["downloader"] = sub_downloaders[1]
self._plugindata.save(plugin_id=plugin_id, self._plugindata.save(plugin_id=plugin_id,
@ -385,7 +395,7 @@ class NAStoolSync(_PluginBase):
return nt_historys return nt_historys
def get_state(self) -> bool: def get_state(self) -> bool:
return True if self._nt_db_path else False return True if self._transfer or self._plugin or self._download else False
@staticmethod @staticmethod
def get_command() -> List[Dict[str, Any]]: def get_command() -> List[Dict[str, Any]]:
@ -588,6 +598,7 @@ class NAStoolSync(_PluginBase):
"download": False, "download": False,
"nt_db_path": "", "nt_db_path": "",
"path": "", "path": "",
"downloader": "",
"site": "", "site": "",
} }