Add support for disabling plugins.

Author: jburhenn

myDevices/plugins/disable.py

@@ -0,0 +1,23 @@
+import sys
+from myDevices.utils.config import Config
+from myDevices.utils.logger import setInfo, info, error
+from myDevices.plugins.manager import PLUGIN_FOLDER
+
+if __name__ == '__main__':
+    # Run the code to disable a plugin in a script so it can be called via sudo
+    setInfo()
+    info(sys.argv)
+    if len(sys.argv) != 3:
+        error('Plugin not disabled, invalid arguments')
+        sys.exit(1)
+    filename = sys.argv[1]
+    if filename.startswith(PLUGIN_FOLDER) and filename.endswith('.plugin'):
+        section = sys.argv[2]
+        config = Config(filename)
+        if section in config.sections():
+            config.set(section, 'enabled', 'false')
+        else:
+            error('Section \'{}\' not found in {}'.format(section, filename))
+            sys.exit(1)
+    else:
+        sys.exit(1)

myDevices/plugins/manager.py

@@ -10,14 +10,17 @@
 import myDevices.cloud.cayennemqtt as cayennemqtt
 from myDevices.utils.config import Config
 from myDevices.utils.logger import debug, error, exception, info
+from myDevices.utils.subprocess import executeCommand
+
+PLUGIN_FOLDER = '/etc/myDevices/plugins'
 
 
 class PluginManager():
     """Loads plugins and reads/writes plugin data"""
 
     def __init__(self):
         """Initializes the plugin manager and loads the plugin list"""
-        self.plugin_folder = '/etc/myDevices/plugins'
+        self.plugin_folder = PLUGIN_FOLDER
         self.plugins = {}
         self.load_plugins()
 
@@ -33,6 +36,8 @@ def load_plugin_from_file(self, filename):
                 enabled = config.get(section, 'enabled', 'true').lower() == 'true'
                 if enabled:
                     plugin = {
+                        'filename': filename,
+                        'section': section,
                         'channel': config.get(section, 'channel'),
                         'name': config.get(section, 'name', section),
                         'module': config.get(section, 'module'),
@@ -96,10 +101,13 @@ def convert_to_dict(self, value):
                     value_dict['unit'] = 'null'
         return value_dict
 
-    def is_plugin(self, plugin, channel):
-        """Returns True if the specified plugin:channel is a valid plugin"""
+    def is_plugin(self, plugin, channel=None):
+        """Returns True if the specified plugin or plugin:channel are valid plugins"""
         try:
-            return plugin + ':' + channel in self.plugins.keys()
+            key = plugin
+            if channel is not None:
+                key = plugin + ':' + channel
+            return key in self.plugins.keys()
         except:
             return False
 
@@ -116,4 +124,20 @@ def write_value(self, plugin, channel, value):
                 return False
         else:
             return False
-        return True
+        return True
+
+    def disable(self, plugin):
+        """Disable the specified plugin"""
+        disabled = False
+        try:
+            output, result = executeCommand('sudo python3 -m myDevices.plugins.disable "{}" "{}"'.format(self.plugins[plugin]['filename'], self.plugins[plugin]['section']))
+            if result == 0:
+                disabled = True
+                info('Plugin \'{}\' disabled'.format(plugin))
+            else:
+                info('Plugin \'{}\' not disabled'.format(plugin))
+            del self.plugins[plugin]
+        except Exception as e:
+            info(e)
+            pass
+        return disabled

myDevices/sensors/sensors.py

@@ -280,6 +280,8 @@ def RemoveSensor(self, name):
         """
         bVal = False
         try:
+            if self.pluginManager.is_plugin(name):
+                return self.pluginManager.disable(name)
             sensorRemove = name
             with self.sensorMutex:
                 retValue = manager.removeDevice(sensorRemove)