Fix plugins enablement code

[cascardo/ipsilon.git] / ipsilon / info / common.py

diff --git a/ipsilon/info/common.py b/ipsilon/info/common.py
index 92a3ba2..03de66a 100755 (executable)
--- a/ipsilon/info/common.py
+++ b/ipsilon/info/common.py
@@ -13,37 +13,35 @@ class InfoProviderBase(PluginObject, Log):
 
     def __init__(self):
         super(InfoProviderBase, self).__init__()
-        self.enabled = False
+        self._site = None
+        self.is_enabled = False
 
     def get_user_attrs(self, user):
         raise NotImplementedError
 
-    @property
-    def is_enabled(self):
-        return self.enabled
-
     def enable(self, site):
-        self.enabled = True
-
-        plugins = site[FACILITY]
-        if self in plugins['enabled']:
+        if self.is_enabled:
             return
 
+        if not self._site:
+            self._site = site
+        plugins = self._site[FACILITY]
+
         # configure self
         if self.name in plugins['config']:
             self.set_config(plugins['config'][self.name])
 
         plugins['enabled'].append(self)
+        self.is_enabled = True
         self.debug('Info plugin enabled: %s' % self.name)
 
     def disable(self, site):
-        self.enabled = False
-
-        plugins = site[FACILITY]
-        if self not in plugins['enabled']:
+        if not self.is_enabled:
             return
 
+        plugins = self._site[FACILITY]
         plugins['enabled'].remove(self)
+        self.is_enabled = False
         self.debug('Info plugin disabled: %s' % self.name)