diff --git a/pgantomizer/anonymize.py b/pgantomizer/anonymize.py index aeaec0a..3179edd 100644 --- a/pgantomizer/anonymize.py +++ b/pgantomizer/anonymize.py @@ -140,7 +140,7 @@ def anonymize_db(schema, db_args): def load_anonymize_remove(dump_file, schema, leave_dump=False, db_args=None): - schema = yaml.load(open(schema)) + schema = yaml.load(open(schema), Loader=yaml.FullLoader) db_args = db_args or get_db_args_from_env() try: load_db_to_new_instance(dump_file, db_args) diff --git a/pgantomizer/dump.py b/pgantomizer/dump.py index 1839301..41011ba 100644 --- a/pgantomizer/dump.py +++ b/pgantomizer/dump.py @@ -8,7 +8,7 @@ def dump_db(dump_path, schema_path, password='', *db_args): - schema = yaml.load(open(schema_path)) + schema = yaml.load(open(schema_path), Loader=yaml.FullLoader) password = password or os.environ.get('DB_DEFAULT_PASS', '') os.putenv('PGPASSWORD', password) cmd = 'PGPASSWORD={password} pg_dump -Fc -Z 9 {args} {tables} -f {filename}'.format(