Skip to content

Commit

Permalink
Move dry_run override
Browse files Browse the repository at this point in the history
  • Loading branch information
jlubken committed Jan 11, 2022
1 parent 47cf570 commit 47218b9
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions src/dsdk/persistor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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)
Expand Down

0 comments on commit 47218b9

Please sign in to comment.