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

fix #726

Browse Source
This commit is contained in:
jxxghp 2023-10-03 16:14:04 -07:00
parent a0ced4e43c
commit 0e30ea92f1
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/plugins/brushflow/__init__.py
View File

@ -49,6 +49,7 @@ class BrushFlow(_PluginBase):
siteshelper = None siteshelper = None
siteoper = None siteoper = None
torrents = None torrents = None
sites = None
qb = None qb = None
tr = None tr = None
# 添加种子定时 # 添加种子定时
@ -88,6 +89,7 @@ class BrushFlow(_PluginBase):
self.siteshelper = SitesHelper() self.siteshelper = SitesHelper()
self.siteoper = SiteOper() self.siteoper = SiteOper()
self.torrents = TorrentsChain() self.torrents = TorrentsChain()
self.sites = SitesHelper()
if config: if config:
self._enabled = config.get("enabled") self._enabled = config.get("enabled")
self._notify = config.get("notify") self._notify = config.get("notify")
@ -115,6 +117,13 @@ class BrushFlow(_PluginBase):
self._save_path = config.get("save_path") self._save_path = config.get("save_path")
self._clear_task = config.get("clear_task") self._clear_task = config.get("clear_task")
# 过滤掉已删除的站点
self._brushsites = [site.get("id") for site in self.sites.get_indexers() if
not site.get("public") and site.get("id") in self._brushsites]
# 保存配置
self.__update_config()
if self._clear_task: if self._clear_task:
# 清除统计数据 # 清除统计数据
self.save_data("statistic", {}) self.save_data("statistic", {})
@ -1113,7 +1122,7 @@ class BrushFlow(_PluginBase):
{ {
'component': 'thead', 'component': 'thead',
'props': { 'props': {
'class': 'text-no-wrap' 'class': 'text-no-wrap'
}, },
'content': [ 'content': [
{ {
@ -1290,10 +1299,10 @@ class BrushFlow(_PluginBase):
else: else:
end_size = 0 end_size = 0
if begin_size and not end_size \ if begin_size and not end_size \
and torrent.size > float(begin_size) * 1024**3: and torrent.size > float(begin_size) * 1024 ** 3:
continue continue
elif begin_size and end_size \ elif begin_size and end_size \
and not float(begin_size) * 1024**3 <= torrent.size <= float(end_size) * 1024**3: and not float(begin_size) * 1024 ** 3 <= torrent.size <= float(end_size) * 1024 ** 3:
continue continue
# 做种人数 # 做种人数
if self._seeder: if self._seeder:
@ -1350,7 +1359,7 @@ class BrushFlow(_PluginBase):
break break
# 保种体积（GB） # 保种体积（GB）
if self._disksize \ if self._disksize \
and (torrents_size + torrent.size) > float(self._disksize) * 1024**3: and (torrents_size + torrent.size) > float(self._disksize) * 1024 ** 3:
logger.warn(f"当前做种体积 {StringUtils.str_filesize(torrents_size)} " logger.warn(f"当前做种体积 {StringUtils.str_filesize(torrents_size)} "
f"已超过保种体积 {self._disksize}，停止新增任务") f"已超过保种体积 {self._disksize}，停止新增任务")
break break
@ -1877,4 +1886,3 @@ class BrushFlow(_PluginBase):
except Exception as e: except Exception as e:
print(str(e)) print(str(e))
return 0 return 0