github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-30 11:38:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add ability to evaluate math in notification text

Browse source
This commit is contained in:
JonnyWong16 2021-05-24 12:53:41 -07:00
parent be86ddbec0
commit d9f702e7f2
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
2 changed files with 32 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
plexpy/config.py
View file

@ -175,6 +175,7 @@ _CONFIG_DEFINITIONS = {
'NOTIFY_SERVER_CONNECTION_THRESHOLD': (int, 'Monitoring', 60),
'NOTIFY_SERVER_UPDATE_REPEAT': (int, 'Monitoring', 0),
'NOTIFY_PLEXPY_UPDATE_REPEAT': (int, 'Monitoring', 0),
'NOTIFY_TEXT_EVAL': (int, 'Advanced', 0),
'PLEXPY_AUTO_UPDATE': (int, 'General', 0),
'REFRESH_LIBRARIES_INTERVAL': (int, 'Monitoring', 12),
'REFRESH_LIBRARIES_ON_STARTUP': (int, 'Monitoring', 1),

35
plexpy/notification_handler.py
View file

@ -1814,6 +1814,24 @@ def str_format(s, parameters):
return s
def str_eval(field_name, kwargs):
field_name = field_name.strip('`')
allowed_names = {
'bool': bool,
'divmod': divmod,
'float': float,
'int': int,
'round': round,
'str': str
}
allowed_names.update(kwargs)
code = compile(field_name, '<string>', 'eval')
for name in code.co_names:
if name not in allowed_names:
raise NameError('Use of {name} not allowed'.format(name=name))
return eval(code, {'__builtins__': {}}, allowed_names)
class CustomFormatter(Formatter):
def __init__(self, default='{{{0}}}'):
self.default = default
@ -1926,10 +1944,19 @@ class CustomFormatter(Formatter):
# used later on, then an exception will be raised
auto_arg_index = False
# given the field_name, find the object it references
# and the argument it came from
obj, arg_used = self.get_field(field_name, args, kwargs)
used_args.add(arg_used)
if plexpy.CONFIG.NOTIFY_TEXT_EVAL and field_name.startswith('`') and field_name.endswith('`'):
try:
obj = str_eval(field_name, kwargs)
used_args.add(field_name)
except (SyntaxError, NameError, ValueError, TypeError) as e:
logger.error("Tautulli NotificationHandler :: Failed to evaluate notification text %s: %s.",
field_name, e)
obj = field_name
else:
# given the field_name, find the object it references
# and the argument it came from
obj, arg_used = self.get_field(field_name, args, kwargs)
used_args.add(arg_used)
# do any conversion on the resulting object
obj = self.convert_field(obj, conversion)