Add support for arbitrary port when downloading Postgres database

This commit is contained in:
Lance Edgar 2018-03-09 17:15:55 -06:00
parent 6f0b1318d5
commit 11db1579c4

View file

@ -143,7 +143,7 @@ def drop_db(name, checkfirst=True):
sudo('sudo -u postgres dropdb {0}'.format(name), shell=False)
def download_db(name, destination=None):
def download_db(name, destination=None, port=None):
"""
Download a database from the "current" server.
"""
@ -151,7 +151,10 @@ def download_db(name, destination=None):
destination = './{0}.sql.gz'.format(name)
run('touch {0}.sql'.format(name))
run('chmod 0666 {0}.sql'.format(name))
sudo('sudo -u postgres pg_dump --file={0}.sql {0}'.format(name), shell=False)
sudo('sudo -u postgres pg_dump {port} --file={name}.sql {name}'.format(
name=name,
port='--port={}'.format(port) if port else '',
), shell=False)
run('gzip --force {0}.sql'.format(name))
get('{0}.sql.gz'.format(name), destination)
run('rm {0}.sql.gz'.format(name))