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

Update __init__.py

Browse Source
This commit is contained in:
thsrite 2023-08-31 19:01:14 +08:00 committed by GitHub
parent 93ec8df713
commit 7baa07474c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 246 additions and 242 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -54,7 +54,7 @@ class SpeedLimiter(_PluginBase):
_bandwidth: float = 0 _bandwidth: float = 0
_allocation_ratio: str = "" _allocation_ratio: str = ""
_auto_limit: bool = False _auto_limit: bool = False
_limit_enabled: bool = False _limit_enabled: bool = True
# 不限速地址 # 不限速地址
_unlimited_ips = {"ipv4": "0.0.0.0/0", "ipv6": "::/0"} _unlimited_ips = {"ipv4": "0.0.0.0/0", "ipv6": "::/0"}
# 当前限速状态 # 当前限速状态
@ -72,10 +72,10 @@ class SpeedLimiter(_PluginBase):
try: try:
# 总带宽 # 总带宽
self._bandwidth = int(float(config.get("bandwidth") or 0)) * 1000000 self._bandwidth = int(float(config.get("bandwidth") or 0)) * 1000000
# 自动限速开关
self._auto_limit = True
except Exception: except Exception:
self._bandwidth = 0 self._bandwidth = 0
# 自动限速开关
self._auto_limit = True if self._bandwidth else False
self._allocation_ratio = config.get("allocation_ratio") or "" self._allocation_ratio = config.get("allocation_ratio") or ""
# 不限速地址 # 不限速地址
self._unlimited_ips["ipv4"] = config.get("ipv4") or "0.0.0.0/0" self._unlimited_ips["ipv4"] = config.get("ipv4") or "0.0.0.0/0"
@ -115,240 +115,240 @@ class SpeedLimiter(_PluginBase):
def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: def get_form(self) -> Tuple[List[dict], Dict[str, Any]]:
return [ return [
{ {
'component': 'VForm', 'component': 'VForm',
'content': [ 'content': [
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'enabled', 'model': 'enabled',
'label': '启用插件', 'label': '启用插件',
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VSwitch', 'component': 'VSwitch',
'props': { 'props': {
'model': 'notify', 'model': 'notify',
'label': '发送通知', 'label': '发送通知',
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'content': [ 'content': [
{ {
'component': 'VSelect', 'component': 'VSelect',
'props': { 'props': {
'chips': True, 'chips': True,
'multiple': True, 'multiple': True,
'model': 'downloader', 'model': 'downloader',
'label': '下载器', 'label': '下载器',
'items': [ 'items': [
{'title': 'Qbittorrent', 'value': 'qbittorrent'}, {'title': 'Qbittorrent', 'value': 'qbittorrent'},
{'title': 'Transmission', 'value': 'transmission'}, {'title': 'Transmission', 'value': 'transmission'},
] ]
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'play_up_speed', 'model': 'play_up_speed',
'label': '播放限速（上传）', 'label': '播放限速（上传）',
'placeholder': 'KB/s' 'placeholder': 'KB/s'
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'play_down_speed', 'model': 'play_down_speed',
'label': '播放限速（下载）', 'label': '播放限速（下载）',
'placeholder': 'KB/s' 'placeholder': 'KB/s'
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'noplay_up_speed', 'model': 'noplay_up_speed',
'label': '未播放限速（上传）', 'label': '未播放限速（上传）',
'placeholder': 'KB/s' 'placeholder': 'KB/s'
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'noplay_down_speed', 'model': 'noplay_down_speed',
'label': '未播放限速（下载）', 'label': '未播放限速（下载）',
'placeholder': 'KB/s' 'placeholder': 'KB/s'
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'bandwidth', 'model': 'bandwidth',
'label': '智能限速上行带宽', 'label': '智能限速上行带宽',
'placeholder': '设置上行带宽后，媒体服务器有媒体播放时根据上行带宽和媒体播放占用带宽计算上传限速数值。' 'placeholder': '设置上行带宽后，媒体服务器有媒体播放时根据上行带宽和媒体播放占用带宽计算上传限速数值。'
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'allocation_ratio', 'model': 'allocation_ratio',
'label': '智能限速分配比例', 'label': '智能限速分配比例',
'placeholder': '多个下载器设置分配比例，如两个下载器设置1:2,留空均分' 'placeholder': '多个下载器设置分配比例，如两个下载器设置1:2,留空均分'
} }
} }
] ]
} }
] ]
}, },
{ {
'component': 'VRow', 'component': 'VRow',
'content': [ 'content': [
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'ipv4', 'model': 'ipv4',
'label': '不限速地址范围（ipv4）', 'label': '不限速地址范围（ipv4）',
'placeholder': '192.168.1.0/24' 'placeholder': '192.168.1.0/24'
} }
} }
] ]
}, },
{ {
'component': 'VCol', 'component': 'VCol',
'props': { 'props': {
'cols': 12, 'cols': 12,
'md': 6 'md': 6
}, },
'content': [ 'content': [
{ {
'component': 'VTextField', 'component': 'VTextField',
'props': { 'props': {
'model': 'ipv6', 'model': 'ipv6',
'label': '不限速地址范围（ipv6）', 'label': '不限速地址范围（ipv6）',
'placeholder': 'FE80::/10' 'placeholder': 'FE80::/10'
} }
} }
] ]
} }
] ]
}, },
] ]
} }
], { ], {
"enabled": False, "enabled": False,
"notify": True, "notify": True,
"downloader": [], "downloader": [],
"play_up_speed": 0, "play_up_speed": 0,
"play_down_speed": 0, "play_down_speed": 0,
"noplay_up_speed": 0, "noplay_up_speed": 0,
"noplay_down_speed": 0, "noplay_down_speed": 0,
"bandwidth": 0, "bandwidth": 0,
"allocation_ratio": "", "allocation_ratio": "",
"ipv4": "", "ipv4": "",
"ipv6": "", "ipv6": "",
} }
def get_page(self) -> List[dict]: def get_page(self) -> List[dict]:
pass pass
@ -425,13 +425,15 @@ class SpeedLimiter(_PluginBase):
if total_bit_rate: if total_bit_rate:
# 开启智能限速计算上传限速 # 开启智能限速计算上传限速
if self._auto_limit: if self._auto_limit:
self.__calc_limit(total_bit_rate) play_up_speed = self.__calc_limit(total_bit_rate)
else:
play_up_speed = self._play_up_speed
# 当前正在播放，开始限速 # 当前正在播放，开始限速
self.__set_limiter(limit_type="播放", upload_limit=self._play_up_speed, self.__set_limiter(limit_type="播放", upload_limit=play_up_speed,
download_limit=self._play_down_speed) download_limit=self._play_down_speed)
else: else:
# 当前没有播放，开始限速 # 当前没有播放，取消限速
self.__set_limiter(limit_type="未播放", upload_limit=self._noplay_up_speed, self.__set_limiter(limit_type="未播放", upload_limit=self._noplay_up_speed,
download_limit=self._noplay_down_speed) download_limit=self._noplay_down_speed)
@ -441,9 +443,11 @@ class SpeedLimiter(_PluginBase):
""" """
residual_bandwidth = (self._bandwidth - total_bit_rate) residual_bandwidth = (self._bandwidth - total_bit_rate)
if residual_bandwidth < 0: if residual_bandwidth < 0:
self._play_up_speed = 10 play_up_speed = 10
else: else:
self._play_up_speed = residual_bandwidth / 8 / 1024 play_up_speed = residual_bandwidth / 8 / 1024
return play_up_speed
def __set_limiter(self, limit_type: str, upload_limit: float, download_limit: float): def __set_limiter(self, limit_type: str, upload_limit: float, download_limit: float):
""" """