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

fix #448 TR限速不生效的问题

Browse Source
This commit is contained in:
jxxghp 2023-09-05 16:35:15 +08:00
parent 6a08b4ba7f
commit c69762d4c9
3 changed files with 23 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
app/modules/qbittorrent/qbittorrent.py
View File

@ -339,6 +339,7 @@ class Qbittorrent(metaclass=Singleton):
try: try:
self.qbc.transfer.upload_limit = int(upload_limit) self.qbc.transfer.upload_limit = int(upload_limit)
self.qbc.transfer.download_limit = int(download_limit) self.qbc.transfer.download_limit = int(download_limit)
return True
except Exception as err: except Exception as err:
logger.error(f"设置速度限制出错：{err}") logger.error(f"设置速度限制出错：{err}")
return False return False

5
app/modules/transmission/transmission.py
View File

@ -243,14 +243,14 @@ class Transmission(metaclass=Singleton):
logger.error(f"获取传输信息出错：{err}") logger.error(f"获取传输信息出错：{err}")
return None return None
def set_speed_limit(self, download_limit: float = None, upload_limit: float = None): def set_speed_limit(self, download_limit: float = None, upload_limit: float = None) -> bool:
""" """
设置速度限制 设置速度限制
:param download_limit: 下载速度限制，单位KB/s :param download_limit: 下载速度限制，单位KB/s
:param upload_limit: 上传速度限制，单位kB/s :param upload_limit: 上传速度限制，单位kB/s
""" """
if not self.trc: if not self.trc:
return return False
try: try:
download_limit_enabled = True if download_limit else False download_limit_enabled = True if download_limit else False
upload_limit_enabled = True if upload_limit else False upload_limit_enabled = True if upload_limit else False
@ -260,6 +260,7 @@ class Transmission(metaclass=Singleton):
speed_limit_down_enabled=download_limit_enabled, speed_limit_down_enabled=download_limit_enabled,
speed_limit_up_enabled=upload_limit_enabled speed_limit_up_enabled=upload_limit_enabled
) )
return True
except Exception as err: except Exception as err:
logger.error(f"设置速度限制出错：{err}") logger.error(f"设置速度限制出错：{err}")
return False return False

38
app/plugins/speedlimiter/__init__.py
View File

@ -539,25 +539,25 @@ class SpeedLimiter(_PluginBase):
title=title, title=title,
text=f"Qbittorrent 已取消限速" text=f"Qbittorrent 已取消限速"
) )
else: else:
if self._tr: if self._tr:
self._tr.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit) self._tr.set_speed_limit(download_limit=download_limit, upload_limit=upload_limit)
# 发送通知 # 发送通知
if self._notify: if self._notify:
title = "【播放限速】" title = "【播放限速】"
if upload_limit or download_limit: if upload_limit or download_limit:
subtitle = f"Transmission 开始{limit_type}限速" subtitle = f"Transmission 开始{limit_type}限速"
self.post_message( self.post_message(
mtype=NotificationType.MediaServer, mtype=NotificationType.MediaServer,
title=title, title=title,
text=f"{subtitle}\n{text}" text=f"{subtitle}\n{text}"
) )
else: else:
self.post_message( self.post_message(
mtype=NotificationType.MediaServer, mtype=NotificationType.MediaServer,
title=title, title=title,
text=f"Transmission 已取消限速" text=f"Transmission 已取消限速"
) )
except Exception as e: except Exception as e:
logger.error(f"设置限速失败：{str(e)}") logger.error(f"设置限速失败：{str(e)}")