fix: fixed extension schema search

Author: cuprumtan

mamonsu/plugins/pgsql/driver/pool.py

@@ -172,11 +172,15 @@ def extension_schema(self, extension, db=None):
         db = self._normalize_db(db)
         if db in self._cache['extension_schema'][extension]:
             return self._cache['extension_schema'][extension][db]
-        self._cache['extension_schema'][extension][db] = self.query(
-            "select n.nspname from pg_extension e "
-            "join pg_namespace n "
-            "on e.extnamespace = n.oid where e.extname = '{0}'".format(extension), db)[0][0]
-        return self._cache['extension_schema'][extension][db]
+        try:
+            self._cache['extension_schema'][extension][db] = self.query(
+                "select n.nspname from pg_extension e "
+                "join pg_namespace n "
+                "on e.extnamespace = n.oid where e.extname = '{0}'".format(extension), db)[0][0]
+            return self._cache['extension_schema'][extension][db]
+        except:
+            self._connections[db].log.info('{0} is not installed'.format(extension))
+            self._cache['pgproee'][db] = False
 
     def databases(self):
         result, databases = self.query(

mamonsu/plugins/pgsql/pg_buffercache.py

@@ -62,8 +62,11 @@ def graphs(self, template, dashboard=False):
                                    'position': 2}}]
 
     def keys_and_queries(self, template_zabbix):
-        result = []
-        for i, item in enumerate(self.Items):
-            result.append('{0}[*],$2 $1 -c "{1}"'.format(self.key.format('.' + item[0]),
-                                                         self.query[i].format(self.extension_schema(extension='pg_buffercache'))))
-        return template_zabbix.key_and_query(result)
+        if self.extension_installed('pg_buffercache'):
+            result = []
+            for i, item in enumerate(self.Items):
+                result.append('{0}[*],$2 $1 -c "{1}"'.format(self.key.format('.' + item[0]),
+                                                             self.query[i].format(self.extension_schema(extension='pg_buffercache'))))
+            return template_zabbix.key_and_query(result)
+        else:
+            return

mamonsu/plugins/pgsql/pg_stat_statement.py

@@ -150,24 +150,26 @@ def graphs(self, template, dashboard=False):
             return []
 
     def keys_and_queries(self, template_zabbix):
-        result = []
-        if LooseVersion(self.VersionPG) < LooseVersion('13'):
-            self.Items[5][1] = self.Items[5][1].format("total_time")
-            all_items = self.Items
-        else:
-            self.Items[5][1] = self.Items[5][1].format("total_exec_time+total_plan_time")
-            all_items = self.Items + self.Items_pg_13
+        if self.extension_installed('pg_stat_statements'):
+            result = []
+            if LooseVersion(self.VersionPG) < LooseVersion('13'):
+                self.Items[5][1] = self.Items[5][1].format("total_time")
+                all_items = self.Items
+            else:
+                self.Items[5][1] = self.Items[5][1].format("total_exec_time+total_plan_time")
+                all_items = self.Items + self.Items_pg_13
 
-        for i, item in enumerate(all_items):
-            keys = item[0].split('[')
-            result.append('{0}[*],$2 $1 -c "{1}"'.format('{0}{1}.{2}'.format(self.key, keys[0], keys[1][:-1]),
-                                                         self.query.format(item[1])))
-
-        if LooseVersion(self.VersionPG) >= LooseVersion('14'):
-            all_items = self.Items_pg_14
             for i, item in enumerate(all_items):
                 keys = item[0].split('[')
                 result.append('{0}[*],$2 $1 -c "{1}"'.format('{0}{1}.{2}'.format(self.key, keys[0], keys[1][:-1]),
-                                                             self.query_info.format(item[1])))
+                                                             self.query.format(item[1])))
 
-        return template_zabbix.key_and_query(result)
+            if LooseVersion(self.VersionPG) >= LooseVersion('14'):
+                all_items = self.Items_pg_14
+                for i, item in enumerate(all_items):
+                    keys = item[0].split('[')
+                    result.append('{0}[*],$2 $1 -c "{1}"'.format('{0}{1}.{2}'.format(self.key, keys[0], keys[1][:-1]),
+                                                                 self.query_info.format(item[1])))
+            return template_zabbix.key_and_query(result)
+        else:
+            return