pg_dump from now on works with postgres user.

1 files changed, 13 insertions, 4 deletions

diff --git a/scripts/rhodecode-setup b/scripts/rhodecode-setup
index 7769bcc..f348d6b 100755
--- a/scripts/rhodecode-setup
+++ b/scripts/rhodecode-setup
@@ -7,6 +7,7 @@ import datetime
 import os
 import subprocess
 import sys
+import tempfile
 
 
 DESCRIPTION = "Install and setup RhodeCode Linaro instance."
@@ -539,7 +540,7 @@ def backup_installation(home_dir):
     execute_command(cmd_args, work_dir=home_dir)
 
 
-def backup_postgres_db(home_dir, db_name):
+def backup_postgres_db(home_dir, db_name, postgres_usr="postgres"):
     """Backup the database as file in home directory.
 
     :param home_dir: Working directory.
@@ -548,9 +549,17 @@ def backup_postgres_db(home_dir, db_name):
     :type str
     """
     date = datetime.date.today()
-    db_filename = "%s.%s" % (DB_BACKUP_FILENAME, date.strftime("%Y.%m.%d"))
-    cmd_args = ["pg_dump", db_name, ">", DB_BACKUP_FILENAME]
-    execute_command(cmd_args, work_dir=home_dir)
+    try:
+        tempfile = tempfile.NamedTemporaryFile()
+        db_filename = "%s.%s" % (DB_BACKUP_FILENAME, date.strftime("%Y.%m.%d"))
+        cmd_args = ["-u", postgres_usr, "pg_dump", db_name, "-f",
+                    tempfile.name]
+        execute_command(cmd_args, work_dir=home_dir)
+
+        cmd_args = ["cp", tempfile.name, DB_BACKUP_FILENAME]
+        execute_command(cmd_args, work_dir=home_dir)
+    finally:
+        tempfile.close()
 
 
 def setup_rhodecode(rhodecode_dir, config_file, git_repos, admin_usr,