thsrite/archived-MoviePilot-Plugins
1
0
Fork 0
mirror of https://github.com/thsrite/MoviePilot-Plugins.git synced 2026-03-27 10:05:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix 添加异常处理，解决cd2上本地目录获取存储空间失败的问题

Browse Source
This commit is contained in:
thsrite
2024-08-14 10:48:24 +08:00
parent 30179da830
commit e240dca065
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
plugins/cd2assistant/__init__.py
View File

@@ -2,12 +2,14 @@ import re
from datetime import datetime, timedelta
import pytz
try:
from clouddrive import CloudDriveClient, Client
from clouddrive.proto import CloudDrive_pb2
except ImportError:
from sys import executable
from subprocess import run
run([executable, "-m", "pip", "install", "-U", "clouddrive"], check=True)
from app import schemas
from app.core.config import settings
@@ -30,7 +32,7 @@ class Cd2Assistant(_PluginBase):
# 插件图标
plugin_icon = "https://raw.githubusercontent.com/thsrite/MoviePilot-Plugins/main/icons/clouddrive.png"
# 插件版本
plugin_version = "1.8.2"
plugin_version = "1.8.3"
# 插件作者
plugin_author = "thsrite"
# 作者主页
@@ -234,13 +236,16 @@ class Cd2Assistant(_PluginBase):
_space_info = "\n"
for f in fs.listdir():
if f and f not in self._black_dir.split(","):
space_info = self._cd2_client.GetSpaceInfo(CloudDrive_pb2.FileRequest(path=f))
space_info = self.__str_to_dict(space_info)
total = self.__convert_bytes(space_info.get("totalSpace"))
used = self.__convert_bytes(space_info.get("usedSpace"))
free = self.__convert_bytes(space_info.get("freeSpace"))
_space_info += f"{f}：{used}/{total}\n"
try:
if f and f not in self._black_dir.split(","):
space_info = self._cd2_client.GetSpaceInfo(CloudDrive_pb2.FileRequest(path=f))
space_info = self.__str_to_dict(space_info)
total = self.__convert_bytes(space_info.get("totalSpace"))
used = self.__convert_bytes(space_info.get("usedSpace"))
free = self.__convert_bytes(space_info.get("freeSpace"))
_space_info += f"{f}：{used}/{total}\n"
except Exception:
logger.error(f"获取云盘 {f} 空间信息失败")
return _space_info
@@ -675,7 +680,8 @@ class Cd2Assistant(_PluginBase):
'component': 'VTextField',
'props': {
'model': 'black_dir',
'label': 'cd2黑名单目录'
'label': 'cd2黑名单目录',
'placeholder': 'cd2上添加的本地目录'
}
}
]