Skip to content

Commit dcc5b68

Browse files
authored
Make 3+ verbose level turn on engine echo (#37)
1 parent 4166c03 commit dcc5b68

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

subsetter/__main__.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import yaml
88

9-
from subsetter.config_model import SubsetterConfig
9+
from subsetter.config_model import DatabaseOutputConfig, SubsetterConfig
1010
from subsetter.plan_model import SubsetPlan
1111
from subsetter.planner import Planner
1212
from subsetter.sampler import Sampler
@@ -128,7 +128,7 @@ def _dict_merge(lhs, rhs):
128128
exc_info=args.verbose > 1,
129129
)
130130
sys.exit(1)
131-
return SubsetterConfig.model_validate(config_data)
131+
config = SubsetterConfig.model_validate(config_data)
132132
except ValueError as exc:
133133
LOGGER.error(
134134
"Unexpected subsetter config file format: %s",
@@ -137,6 +137,13 @@ def _dict_merge(lhs, rhs):
137137
)
138138
sys.exit(1)
139139

140+
if args.verbose > 2:
141+
config.source.echo = True
142+
if config.sampler and isinstance(config.sampler.output, DatabaseOutputConfig):
143+
config.sampler.output.echo = True
144+
145+
return config
146+
140147

141148
def _main_plan(args):
142149
config = _get_config(args)

subsetter/common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,7 @@ def get_connect_args(
164164
sqlite_databases: Optional[Dict[str, str]] = {}
165165
isolation_level: Optional[IsolationLevel] = None
166166
ssl: SSLConfig = SSLConfig()
167+
echo: bool = False
167168

168169
def database_url(
169170
self,
@@ -198,6 +199,7 @@ def database_engine(
198199
self.dialect or DEFAULT_DIALECT,
199200
env_prefix=env_prefix,
200201
),
202+
echo=self.echo,
201203
)
202204

203205
@sa.event.listens_for(engine, "connect")

0 commit comments

Comments
 (0)