diff --git a/agents/plugins/mk_postgres.py b/agents/plugins/mk_postgres.py index 9d9d002dba5..05cfbaeabf9 100755 --- a/agents/plugins/mk_postgres.py +++ b/agents/plugins/mk_postgres.py @@ -141,7 +141,10 @@ def __init__(self, db_user, pg_binary_path, instance, process_match_patterns): self.pg_passfile = instance.get("pg_passfile", "") self.pg_version = instance.get("pg_version") self.my_env = os.environ.copy() - self.my_env["PGPASSFILE"] = instance.get("pg_passfile", "") + if instance.get("pg_passfile", ""): + self.my_env["PGPASSFILE"] = instance.get("pg_passfile", "") + self.sep = os.sep + self.psql_binary_name = "psql" if pg_binary_path is None: self.psql_binary_path = self.get_psql_binary_path() @@ -737,12 +740,15 @@ def get_bloat(self, databases, numeric_version): query = "\\pset footer off \\\\" cur_rows_only = False + output = "" for idx, database in enumerate(databases): query = "%s \\c %s \\\\ %s" % (query, database, bloat_query) if idx == 0: query = "%s \\pset tuples_only on" % query - - return self.run_sql_as_db_user(query, mixed_cmd=True, rows_only=cur_rows_only) + output += self.run_sql_as_db_user(query, mixed_cmd=True, rows_only=cur_rows_only) + cur_rows_only = True + query = "\\pset footer off \\\\" + return output class PostgresLinux(PostgresBase):