mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-15 09:25:04 +00:00
227 lines
8.5 KiB
Python
227 lines
8.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# Copyright (C) 2022 Chris Caron <lead2gold@gmail.com>
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# This code is licensed under the MIT License.
|
||
|
|
#
|
||
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
# of this software and associated documentation files(the "Software"), to deal
|
||
|
|
# in the Software without restriction, including without limitation the rights
|
||
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and / or sell
|
||
|
|
# copies of the Software, and to permit persons to whom the Software is
|
||
|
|
# furnished to do so, subject to the following conditions :
|
||
|
|
#
|
||
|
|
# The above copyright notice and this permission notice shall be included in
|
||
|
|
# all copies or substantial portions of the Software.
|
||
|
|
#
|
||
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE
|
||
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
|
|
# THE SOFTWARE.
|
||
|
|
from ..plugins.NotifyBase import NotifyBase
|
||
|
|
from ..utils import URL_DETAILS_RE
|
||
|
|
from ..utils import parse_url
|
||
|
|
from ..utils import url_assembly
|
||
|
|
from ..utils import dict_full_update
|
||
|
|
from .. import common
|
||
|
|
from ..logger import logger
|
||
|
|
import inspect
|
||
|
|
|
||
|
|
|
||
|
|
class CustomNotifyPlugin(NotifyBase):
|
||
|
|
"""
|
||
|
|
Apprise Custom Plugin Hook
|
||
|
|
|
||
|
|
This gets initialized based on @notify decorator definitions
|
||
|
|
|
||
|
|
"""
|
||
|
|
# Our Custom notification
|
||
|
|
service_url = 'https://github.com/caronc/apprise/wiki/Custom_Notification'
|
||
|
|
|
||
|
|
# Over-ride our category since this inheritance of the NotifyBase class
|
||
|
|
# should be treated differently.
|
||
|
|
category = 'custom'
|
||
|
|
|
||
|
|
# Define object templates
|
||
|
|
templates = (
|
||
|
|
'{schema}://',
|
||
|
|
)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def parse_url(url):
|
||
|
|
"""
|
||
|
|
Parses the URL and returns arguments retrieved
|
||
|
|
|
||
|
|
"""
|
||
|
|
return parse_url(url, verify_host=False, simple=True)
|
||
|
|
|
||
|
|
def url(self, privacy=False, *args, **kwargs):
|
||
|
|
"""
|
||
|
|
General URL assembly
|
||
|
|
"""
|
||
|
|
return '{schema}://'.format(schema=self.secure_protocol)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def instantiate_plugin(url, send_func, name=None):
|
||
|
|
"""
|
||
|
|
The function used to add a new notification plugin based on the schema
|
||
|
|
parsed from the provided URL into our supported matrix structure.
|
||
|
|
"""
|
||
|
|
|
||
|
|
if not isinstance(url, str):
|
||
|
|
msg = 'An invalid custom notify url/schema ({}) provided in ' \
|
||
|
|
'function {}.'.format(url, send_func.__name__)
|
||
|
|
logger.warning(msg)
|
||
|
|
return None
|
||
|
|
|
||
|
|
# Validate that our schema is okay
|
||
|
|
re_match = URL_DETAILS_RE.match(url)
|
||
|
|
if not re_match:
|
||
|
|
msg = 'An invalid custom notify url/schema ({}) provided in ' \
|
||
|
|
'function {}.'.format(url, send_func.__name__)
|
||
|
|
logger.warning(msg)
|
||
|
|
return None
|
||
|
|
|
||
|
|
# Acquire our plugin name
|
||
|
|
plugin_name = re_match.group('schema').lower()
|
||
|
|
|
||
|
|
if not re_match.group('base'):
|
||
|
|
url = '{}://'.format(plugin_name)
|
||
|
|
|
||
|
|
# Keep a default set of arguments to apply to all called references
|
||
|
|
base_args = parse_url(
|
||
|
|
url, default_schema=plugin_name, verify_host=False, simple=True)
|
||
|
|
|
||
|
|
if plugin_name in common.NOTIFY_SCHEMA_MAP:
|
||
|
|
# we're already handling this object
|
||
|
|
msg = 'The schema ({}) is already defined and could not be ' \
|
||
|
|
'loaded from custom notify function {}.' \
|
||
|
|
.format(url, send_func.__name__)
|
||
|
|
logger.warning(msg)
|
||
|
|
return None
|
||
|
|
|
||
|
|
# We define our own custom wrapper class so that we can initialize
|
||
|
|
# some key default configuration values allowing calls to our
|
||
|
|
# `Apprise.details()` to correctly differentiate one custom plugin
|
||
|
|
# that was loaded from another
|
||
|
|
class CustomNotifyPluginWrapper(CustomNotifyPlugin):
|
||
|
|
|
||
|
|
# Our Service Name
|
||
|
|
service_name = name if isinstance(name, str) \
|
||
|
|
and name else 'Custom - {}'.format(plugin_name)
|
||
|
|
|
||
|
|
# Store our matched schema
|
||
|
|
secure_protocol = plugin_name
|
||
|
|
|
||
|
|
requirements = {
|
||
|
|
# Define our required packaging in order to work
|
||
|
|
'details': "Source: {}".format(inspect.getfile(send_func))
|
||
|
|
}
|
||
|
|
|
||
|
|
# Assign our send() function
|
||
|
|
__send = staticmethod(send_func)
|
||
|
|
|
||
|
|
# Update our default arguments
|
||
|
|
_base_args = base_args
|
||
|
|
|
||
|
|
def __init__(self, **kwargs):
|
||
|
|
"""
|
||
|
|
Our initialization
|
||
|
|
|
||
|
|
"""
|
||
|
|
# init parent
|
||
|
|
super().__init__(**kwargs)
|
||
|
|
|
||
|
|
self._default_args = {}
|
||
|
|
|
||
|
|
# Apply our updates based on what was parsed
|
||
|
|
dict_full_update(self._default_args, self._base_args)
|
||
|
|
dict_full_update(self._default_args, kwargs)
|
||
|
|
|
||
|
|
# Update our arguments (applying them to what we originally)
|
||
|
|
# initialized as
|
||
|
|
self._default_args['url'] = url_assembly(**self._default_args)
|
||
|
|
|
||
|
|
def send(self, body, title='', notify_type=common.NotifyType.INFO,
|
||
|
|
*args, **kwargs):
|
||
|
|
"""
|
||
|
|
Our send() call which triggers our hook
|
||
|
|
"""
|
||
|
|
|
||
|
|
response = False
|
||
|
|
try:
|
||
|
|
# Enforce a boolean response
|
||
|
|
result = self.__send(
|
||
|
|
body, title, notify_type, *args,
|
||
|
|
meta=self._default_args, **kwargs)
|
||
|
|
|
||
|
|
if result is None:
|
||
|
|
# The wrapper did not define a return (or returned
|
||
|
|
# None)
|
||
|
|
# this is treated as a successful return as it is
|
||
|
|
# assumed the developer did not care about the result
|
||
|
|
# of the call.
|
||
|
|
response = True
|
||
|
|
|
||
|
|
else:
|
||
|
|
# Perform boolean check (allowing obects to also be
|
||
|
|
# returned and check against the __bool__ call
|
||
|
|
response = True if result else False
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
# Unhandled Exception
|
||
|
|
self.logger.warning(
|
||
|
|
'An exception occured sending a %s notification.',
|
||
|
|
common.
|
||
|
|
NOTIFY_SCHEMA_MAP[self.secure_protocol].service_name)
|
||
|
|
self.logger.debug(
|
||
|
|
'%s Exception: %s',
|
||
|
|
common.NOTIFY_SCHEMA_MAP[self.secure_protocol], str(e))
|
||
|
|
return False
|
||
|
|
|
||
|
|
if response:
|
||
|
|
self.logger.info(
|
||
|
|
'Sent %s notification.',
|
||
|
|
common.
|
||
|
|
NOTIFY_SCHEMA_MAP[self.secure_protocol].service_name)
|
||
|
|
else:
|
||
|
|
self.logger.warning(
|
||
|
|
'Failed to send %s notification.',
|
||
|
|
common.
|
||
|
|
NOTIFY_SCHEMA_MAP[self.secure_protocol].service_name)
|
||
|
|
return response
|
||
|
|
|
||
|
|
# Store our plugin into our core map file
|
||
|
|
common.NOTIFY_SCHEMA_MAP[plugin_name] = CustomNotifyPluginWrapper
|
||
|
|
|
||
|
|
# Update our custom plugin map
|
||
|
|
module_pyname = str(send_func.__module__)
|
||
|
|
if module_pyname not in common.NOTIFY_CUSTOM_MODULE_MAP:
|
||
|
|
# Support non-dynamic includes as well...
|
||
|
|
common.NOTIFY_CUSTOM_MODULE_MAP[module_pyname] = {
|
||
|
|
'path': inspect.getfile(send_func),
|
||
|
|
|
||
|
|
# Initialize our template
|
||
|
|
'notify': {},
|
||
|
|
}
|
||
|
|
|
||
|
|
common.\
|
||
|
|
NOTIFY_CUSTOM_MODULE_MAP[module_pyname]['notify'][plugin_name] = {
|
||
|
|
# Our Serivice Description (for API and CLI --details view)
|
||
|
|
'name': CustomNotifyPluginWrapper.service_name,
|
||
|
|
# The name of the send function the @notify decorator wrapped
|
||
|
|
'fn_name': send_func.__name__,
|
||
|
|
# The URL that was provided in the @notify decorator call
|
||
|
|
# associated with the 'on='
|
||
|
|
'url': url,
|
||
|
|
# The Initialized Plugin that was generated based on the above
|
||
|
|
# parameters
|
||
|
|
'plugin': CustomNotifyPluginWrapper}
|
||
|
|
|
||
|
|
# return our plugin
|
||
|
|
return common.NOTIFY_SCHEMA_MAP[plugin_name]
|