From 47218b9c881b1139267f8acf7e4d5236406b4f3a Mon Sep 17 00:00:00 2001 From: Jason Lubken Date: Tue, 11 Jan 2022 06:14:05 -0500 Subject: [PATCH] Move dry_run override --- src/dsdk/persistor.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/src/dsdk/persistor.py b/src/dsdk/persistor.py index a892d27..f4f385e 100644 --- a/src/dsdk/persistor.py +++ b/src/dsdk/persistor.py @@ -171,10 +171,8 @@ def dry_run( parameters: Dict[str, Any], exceptions: Tuple = (), ): - """Execute sql found in asse with dry_run parameter set to 1.""" + """Execute sql found in asset with dry_run.""" logger.info(self.ON) - parameters = parameters.copy() - parameters["dry_run"] = 1 errors = [] for path, query in self.sql(): logger.info(self.DRY_RUN, path) @@ -192,12 +190,12 @@ def dry_run_query( parameters, exceptions: Tuple = (), ) -> Optional[Exception]: - """Dry run query.""" + """Dry run query with dry_run parameter set to 1.""" with self.rollback() as cur: rendered = self.render_without_keys( cur, query, - parameters, + {**parameters, "dry_run": 1}, ) with NamedTemporaryFile("w", delete=False, suffix=".sql") as fout: fout.write(rendered)