small fix

This commit is contained in:
Greg DiCristofaro 2021-05-06 11:20:02 -04:00
parent 6cdb168a05
commit 46c4017fe7

View File

@ -744,8 +744,8 @@ def get_pg_schema(dbname: str, pg_username: str, pg_pword: str, pg_host: str, pg
""" """
os.environ['PGPASSWORD'] = pg_pword os.environ['PGPASSWORD'] = pg_pword
pg_dump = ["pg_dump", "-U", pg_username, "-h", pg_host, "-p", pg_port, "--schema-only", "-d", dbname, "-t", pg_dump = ["pg_dump", "-U", pg_username, "-h", pg_host, "-p", pg_port,
"public.*"] "--schema-only", "-d", dbname, "-t", "public.*"]
output = subprocess.check_output(pg_dump) output = subprocess.check_output(pg_dump)
output_str = output.decode('UTF-8') output_str = output.decode('UTF-8')
return sanitize_schema(output_str) return sanitize_schema(output_str)