from sqlalchemy import testing
from sqlalchemy.testing import config
from sqlalchemy.testing import provision
+ from sqlalchemy.engine import url as sa_url
if options.dburi:
db_urls = list(options.dburi)
config._current = None
- expanded_urls = list(provision.generate_db_urls(db_urls, extra_drivers))
-
- for db_url in expanded_urls:
- log.info("Adding database URL: %s", db_url)
-
- if options.write_idents and provision.FOLLOWER_IDENT:
+ if options.write_idents and provision.FOLLOWER_IDENT:
+ for db_url in [sa_url.make_url(db_url) for db_url in db_urls]:
with open(options.write_idents, "a") as file_:
file_.write(
f"{provision.FOLLOWER_IDENT} "
f"{db_url.render_as_string(hide_password=False)}\n"
)
+ expanded_urls = list(provision.generate_db_urls(db_urls, extra_drivers))
+
+ for db_url in expanded_urls:
+ log.info("Adding database URL: %s", db_url)
+
cfg = provision.setup_config(
db_url, options, file_config, provision.FOLLOWER_IDENT
)