File tree Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Original file line number Diff line number Diff line change 6
6
7
7
import yaml
8
8
9
- from subsetter .config_model import SubsetterConfig
9
+ from subsetter .config_model import DatabaseOutputConfig , SubsetterConfig
10
10
from subsetter .plan_model import SubsetPlan
11
11
from subsetter .planner import Planner
12
12
from subsetter .sampler import Sampler
@@ -128,7 +128,7 @@ def _dict_merge(lhs, rhs):
128
128
exc_info = args .verbose > 1 ,
129
129
)
130
130
sys .exit (1 )
131
- return SubsetterConfig .model_validate (config_data )
131
+ config = SubsetterConfig .model_validate (config_data )
132
132
except ValueError as exc :
133
133
LOGGER .error (
134
134
"Unexpected subsetter config file format: %s" ,
@@ -137,6 +137,13 @@ def _dict_merge(lhs, rhs):
137
137
)
138
138
sys .exit (1 )
139
139
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
+
140
147
141
148
def _main_plan (args ):
142
149
config = _get_config (args )
Original file line number Diff line number Diff line change @@ -164,6 +164,7 @@ def get_connect_args(
164
164
sqlite_databases : Optional [Dict [str , str ]] = {}
165
165
isolation_level : Optional [IsolationLevel ] = None
166
166
ssl : SSLConfig = SSLConfig ()
167
+ echo : bool = False
167
168
168
169
def database_url (
169
170
self ,
@@ -198,6 +199,7 @@ def database_engine(
198
199
self .dialect or DEFAULT_DIALECT ,
199
200
env_prefix = env_prefix ,
200
201
),
202
+ echo = self .echo ,
201
203
)
202
204
203
205
@sa .event .listens_for (engine , "connect" )
You can’t perform that action at this time.
0 commit comments