From a650840429fadd19c14b7246c68d245262afb403 Mon Sep 17 00:00:00 2001 From: Fabio Manganiello Date: Wed, 9 Sep 2020 02:14:59 +0200 Subject: [PATCH] Refactored variable plugin --- platypush/plugins/variable.py | 20 ++++++-------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/platypush/plugins/variable.py b/platypush/plugins/variable.py index 92ee103a..7810507d 100644 --- a/platypush/plugins/variable.py +++ b/platypush/plugins/variable.py @@ -18,14 +18,14 @@ class VariablePlugin(Plugin): _variable_table_name = 'variable' - def __init__(self, *args, **kwargs): + def __init__(self, **kwargs): """ The plugin will create a table named ``variable`` on the database configured in the :mod:`platypush.plugins.db` plugin. You'll have to specify a default ``engine`` in your ``db`` plugin configuration. """ - super().__init__(*args, **kwargs) + super().__init__(**kwargs) self.db_plugin = get_plugin('db') self.redis_plugin = get_plugin('redis') @@ -64,7 +64,7 @@ class VariablePlugin(Plugin): *self.db_config['args'], **self.db_config['kwargs']).output - return {name: rows[0]['value'] if rows else None} + return {name: rows[0]['value'] if rows else default_value} @action def set(self, **kwargs): @@ -74,8 +74,8 @@ class VariablePlugin(Plugin): :param kwargs: Key-value list of variables to set (e.g. ``foo='bar', answer=42``) """ - records = [ { 'name': k, 'value': v } - for (k,v) in kwargs.items() ] + records = [{'name': k, 'value': v} + for (k, v) in kwargs.items()] self.db_plugin.insert(table=self._variable_table_name, records=records, key_columns=['name'], @@ -86,7 +86,6 @@ class VariablePlugin(Plugin): return kwargs - @action def unset(self, name): """ @@ -96,7 +95,7 @@ class VariablePlugin(Plugin): :type name: str """ - records = [ { 'name': name } ] + records = [{'name': name}] self.db_plugin.delete(table=self._variable_table_name, records=records, engine=self.db_config['engine'], @@ -105,7 +104,6 @@ class VariablePlugin(Plugin): return True - @action def mget(self, name): """ @@ -114,8 +112,6 @@ class VariablePlugin(Plugin): :param name: Variable name :type name: str - :param default_value: What will be returned if the variable is not defined (default: None) - :returns: A map in the format ``{"":""}`` """ @@ -134,7 +130,6 @@ class VariablePlugin(Plugin): self.redis_plugin.mset(**kwargs) return kwargs - @action def munset(self, name): """ @@ -146,7 +141,6 @@ class VariablePlugin(Plugin): return self.redis_plugin.delete(name) - @action def expire(self, name, expire): """ @@ -161,6 +155,4 @@ class VariablePlugin(Plugin): return self.redis_plugin.expire(name, expire) - # vim:sw=4:ts=4:et: -