From a8d97f9fc033bacc03522c2540aa5f726a0eb71a Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Fri, 18 Mar 2022 08:26:12 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9develop=E5=91=BD=E4=BB=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 hb_server/apps/develop/tasks.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/hb_server/apps/develop/tasks.py b/hb_server/apps/develop/tasks.py
index b9c1991..580f525 100644
--- a/hb_server/apps/develop/tasks.py
+++ b/hb_server/apps/develop/tasks.py
@@ -4,7 +4,7 @@ from rest_framework.response import Response
 from celery import shared_task
 import os
 import subprocess
-from server.settings_pro import DATABASES, BACKUP_PATH, SH_PATH
+from server.settings_pro import DATABASES, BACKUP_PATH, SH_PATH, SD_PWD
 
 @shared_task
 def backup_database():
@@ -13,7 +13,8 @@ def backup_database():
     """
     import datetime
     name = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
-    command = 'pg_dump "user={} password={} dbname={}" > {}/hberp_{}.sql'.format(
+    command = 'echo "{}" | sudo -S pg_dump "user={} password={} dbname={}" > {}/hberp_{}.sql'.format(
+        SD_PWD,
         DATABASES['default']['USER'],
         DATABASES['default']['PASSWORD'],
         DATABASES['default']['NAME'],
@@ -30,7 +31,7 @@ def reload_server():
 
 @shared_task
 def reload_server_only():
-    command = 'sudo service supervisor reload'
+    command = 'echo "{}" | sudo -S service supervisor reload'.format(SD_PWD)
     completed = subprocess.run(command, shell=True, capture_output=True, text=True)
     return completed