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