mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-01 05:27:02 +08:00
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from app.helper.service import ServiceBaseHelper
|
|
from app.schemas import NotificationConf, ServiceInfo
|
|
from app.schemas.types import SystemConfigKey, ModuleType
|
|
|
|
|
|
class NotificationHelper(ServiceBaseHelper[NotificationConf]):
|
|
"""
|
|
消息通知帮助类
|
|
"""
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
config_key=SystemConfigKey.Notifications,
|
|
conf_type=NotificationConf,
|
|
module_type=ModuleType.Notification
|
|
)
|
|
|
|
def is_notification(
|
|
self,
|
|
service_type: Optional[str] = None,
|
|
service: Optional[ServiceInfo] = None,
|
|
name: Optional[str] = None,
|
|
) -> bool:
|
|
"""
|
|
通用的消息通知服务类型判断方法
|
|
|
|
:param service_type: 消息通知服务的类型名称（如 'wechat', 'voicechat', 'telegram', 等）
|
|
:param service: 要判断的服务信息
|
|
:param name: 服务的名称
|
|
:return: 如果服务类型或实例为指定类型，返回 True；否则返回 False
|
|
"""
|
|
# 如果未提供 service 则通过 name 获取服务
|
|
service = service or self.get_service(name=name)
|
|
|
|
# 判断服务类型是否为指定类型
|
|
return bool(service and service.type == service_type)
|