psw/borgmatic
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor password assignment logic

Browse source
This commit is contained in:
Divyansh Singh 2023-06-15 22:54:06 +05:30
parent a9386b7a87
commit b7423c488e
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
borgmatic/actions/restore.py
View file

@ -72,7 +72,7 @@ def restore_single_database(
): # pragma: no cover
'''
Given (among other things) an archive name, a database hook name, the hostname,
port, username and password as connection params, and a configured database
port, username and password as connection params, and a configured database
configuration dict, restore that database from the archive.
'''
logger.info(

5
borgmatic/hooks/postgresql.py
View file

@ -30,7 +30,10 @@ def make_extra_environment(database, restore_connection_params=None):
extra = dict()
try:
extra['PGPASSWORD'] = restore_connection_params.get('password') or database['restore_password'] or database['password']
if restore_connection_params:
extra['PGPASSWORD'] = restore_connection_params.get('password') or database.get('restore_password', database['password'])
else:
extra['PGPASSWORD'] = database['password']
except (AttributeError, KeyError):
pass