main commit

venv/lib/python3.12/site-packages/alembic/testing/env.py

@@ -1,6 +1,5 @@
 import importlib.machinery
 import os
-from pathlib import Path
 import shutil
 import textwrap
 
@@ -17,7 +16,7 @@ from ..script import ScriptDirectory
 
 def _get_staging_directory():
     if provision.FOLLOWER_IDENT:
-        return f"scratch_{provision.FOLLOWER_IDENT}"
+        return "scratch_%s" % provision.FOLLOWER_IDENT
     else:
         return "scratch"
 
@@ -25,7 +24,7 @@ def _get_staging_directory():
 def staging_env(create=True, template="generic", sourceless=False):
     cfg = _testing_config()
     if create:
-        path = _join_path(_get_staging_directory(), "scripts")
+        path = os.path.join(_get_staging_directory(), "scripts")
         assert not os.path.exists(path), (
             "staging directory %s already exists; poor cleanup?" % path
         )
@@ -48,7 +47,7 @@ def staging_env(create=True, template="generic", sourceless=False):
                 "pep3147_everything",
             ), sourceless
             make_sourceless(
-                _join_path(path, "env.py"),
+                os.path.join(path, "env.py"),
                 "pep3147" if "pep3147" in sourceless else "simple",
             )
 
@@ -64,14 +63,14 @@ def clear_staging_env():
 
 
 def script_file_fixture(txt):
-    dir_ = _join_path(_get_staging_directory(), "scripts")
-    path = _join_path(dir_, "script.py.mako")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
+    path = os.path.join(dir_, "script.py.mako")
     with open(path, "w") as f:
         f.write(txt)
 
 
 def env_file_fixture(txt):
-    dir_ = _join_path(_get_staging_directory(), "scripts")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
     txt = (
         """
 from alembic import context
@@ -81,7 +80,7 @@ config = context.config
         + txt
     )
 
-    path = _join_path(dir_, "env.py")
+    path = os.path.join(dir_, "env.py")
     pyc_path = util.pyc_file_from_path(path)
     if pyc_path:
         os.unlink(pyc_path)
@@ -91,26 +90,26 @@ config = context.config
 
 
 def _sqlite_file_db(tempname="foo.db", future=False, scope=None, **options):
-    dir_ = _join_path(_get_staging_directory(), "scripts")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
     url = "sqlite:///%s/%s" % (dir_, tempname)
-    if scope:
+    if scope and util.sqla_14:
         options["scope"] = scope
     return testing_util.testing_engine(url=url, future=future, options=options)
 
 
 def _sqlite_testing_config(sourceless=False, future=False):
-    dir_ = _join_path(_get_staging_directory(), "scripts")
-    url = f"sqlite:///{dir_}/foo.db"
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
+    url = "sqlite:///%s/foo.db" % dir_
 
     sqlalchemy_future = future or ("future" in config.db.__class__.__module__)
 
     return _write_config_file(
-        f"""
+        """
 [alembic]
-script_location = {dir_}
-sqlalchemy.url = {url}
-sourceless = {"true" if sourceless else "false"}
-{"sqlalchemy.future = true" if sqlalchemy_future else ""}
+script_location = %s
+sqlalchemy.url = %s
+sourceless = %s
+%s
 
 [loggers]
 keys = root,sqlalchemy
@@ -119,7 +118,7 @@ keys = root,sqlalchemy
 keys = console
 
 [logger_root]
-level = WARNING
+level = WARN
 handlers = console
 qualname =
 
@@ -141,25 +140,29 @@ keys = generic
 format = %%(levelname)-5.5s [%%(name)s] %%(message)s
 datefmt = %%H:%%M:%%S
     """
+        % (
+            dir_,
+            url,
+            "true" if sourceless else "false",
+            "sqlalchemy.future = true" if sqlalchemy_future else "",
+        )
     )
 
 
 def _multi_dir_testing_config(sourceless=False, extra_version_location=""):
-    dir_ = _join_path(_get_staging_directory(), "scripts")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
     sqlalchemy_future = "future" in config.db.__class__.__module__
 
     url = "sqlite:///%s/foo.db" % dir_
 
     return _write_config_file(
-        f"""
+        """
 [alembic]
-script_location = {dir_}
-sqlalchemy.url = {url}
-sqlalchemy.future = {"true" if sqlalchemy_future else "false"}
-sourceless = {"true" if sourceless else "false"}
-path_separator = space
-version_locations = %(here)s/model1/ %(here)s/model2/ %(here)s/model3/ \
-{extra_version_location}
+script_location = %s
+sqlalchemy.url = %s
+sqlalchemy.future = %s
+sourceless = %s
+version_locations = %%(here)s/model1/ %%(here)s/model2/ %%(here)s/model3/ %s
 
 [loggers]
 keys = root
@@ -168,7 +171,7 @@ keys = root
 keys = console
 
 [logger_root]
-level = WARNING
+level = WARN
 handlers = console
 qualname =
 
@@ -185,63 +188,26 @@ keys = generic
 format = %%(levelname)-5.5s [%%(name)s] %%(message)s
 datefmt = %%H:%%M:%%S
     """
-    )
-
-
-def _no_sql_pyproject_config(dialect="postgresql", directives=""):
-    """use a postgresql url with no host so that
-    connections guaranteed to fail"""
-    dir_ = _join_path(_get_staging_directory(), "scripts")
-
-    return _write_toml_config(
-        f"""
-[tool.alembic]
-script_location ="{dir_}"
-{textwrap.dedent(directives)}
-
-        """,
-        f"""
-[alembic]
-sqlalchemy.url = {dialect}://
-
-[loggers]
-keys = root
-
-[handlers]
-keys = console
-
-[logger_root]
-level = WARNING
-handlers = console
-qualname =
-
-[handler_console]
-class = StreamHandler
-args = (sys.stderr,)
-level = NOTSET
-formatter = generic
-
-[formatters]
-keys = generic
-
-[formatter_generic]
-format = %%(levelname)-5.5s [%%(name)s] %%(message)s
-datefmt = %%H:%%M:%%S
-
-""",
+        % (
+            dir_,
+            url,
+            "true" if sqlalchemy_future else "false",
+            "true" if sourceless else "false",
+            extra_version_location,
+        )
     )
 
 
 def _no_sql_testing_config(dialect="postgresql", directives=""):
     """use a postgresql url with no host so that
     connections guaranteed to fail"""
-    dir_ = _join_path(_get_staging_directory(), "scripts")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
     return _write_config_file(
-        f"""
+        """
 [alembic]
-script_location ={dir_}
-sqlalchemy.url = {dialect}://
-{directives}
+script_location = %s
+sqlalchemy.url = %s://
+%s
 
 [loggers]
 keys = root
@@ -250,7 +216,7 @@ keys = root
 keys = console
 
 [logger_root]
-level = WARNING
+level = WARN
 handlers = console
 qualname =
 
@@ -268,16 +234,10 @@ format = %%(levelname)-5.5s [%%(name)s] %%(message)s
 datefmt = %%H:%%M:%%S
 
 """
+        % (dir_, dialect, directives)
     )
 
 
-def _write_toml_config(tomltext, initext):
-    cfg = _write_config_file(initext)
-    with open(cfg.toml_file_name, "w") as f:
-        f.write(tomltext)
-    return cfg
-
-
 def _write_config_file(text):
     cfg = _testing_config()
     with open(cfg.config_file_name, "w") as f:
@@ -290,10 +250,7 @@ def _testing_config():
 
     if not os.access(_get_staging_directory(), os.F_OK):
         os.mkdir(_get_staging_directory())
-    return Config(
-        _join_path(_get_staging_directory(), "test_alembic.ini"),
-        _join_path(_get_staging_directory(), "pyproject.toml"),
-    )
+    return Config(os.path.join(_get_staging_directory(), "test_alembic.ini"))
 
 
 def write_script(
@@ -313,7 +270,9 @@ def write_script(
     script = Script._from_path(scriptdir, path)
     old = scriptdir.revision_map.get_revision(script.revision)
     if old.down_revision != script.down_revision:
-        raise Exception("Can't change down_revision on a refresh operation.")
+        raise Exception(
+            "Can't change down_revision " "on a refresh operation."
+        )
     scriptdir.revision_map.add_revision(script, _replace=True)
 
     if sourceless:
@@ -353,9 +312,9 @@ def three_rev_fixture(cfg):
     write_script(
         script,
         a,
-        f"""\
+        """\
 "Rev A"
-revision = '{a}'
+revision = '%s'
 down_revision = None
 
 from alembic import op
@@ -368,7 +327,8 @@ def upgrade():
 def downgrade():
     op.execute("DROP STEP 1")
 
-""",
+"""
+        % a,
     )
 
     script.generate_revision(b, "revision b", refresh=True, head=a)
@@ -398,10 +358,10 @@ def downgrade():
     write_script(
         script,
         c,
-        f"""\
+        """\
 "Rev C"
-revision = '{c}'
-down_revision = '{b}'
+revision = '%s'
+down_revision = '%s'
 
 from alembic import op
 
@@ -413,7 +373,8 @@ def upgrade():
 def downgrade():
     op.execute("DROP STEP 3")
 
-""",
+"""
+        % (c, b),
     )
     return a, b, c
 
@@ -435,10 +396,10 @@ def multi_heads_fixture(cfg, a, b, c):
     write_script(
         script,
         d,
-        f"""\
+        """\
 "Rev D"
-revision = '{d}'
-down_revision = '{b}'
+revision = '%s'
+down_revision = '%s'
 
 from alembic import op
 
@@ -450,7 +411,8 @@ def upgrade():
 def downgrade():
     op.execute("DROP STEP 4")
 
-""",
+"""
+        % (d, b),
     )
 
     script.generate_revision(
@@ -459,10 +421,10 @@ def downgrade():
     write_script(
         script,
         e,
-        f"""\
+        """\
 "Rev E"
-revision = '{e}'
-down_revision = '{d}'
+revision = '%s'
+down_revision = '%s'
 
 from alembic import op
 
@@ -474,7 +436,8 @@ def upgrade():
 def downgrade():
     op.execute("DROP STEP 5")
 
-""",
+"""
+        % (e, d),
     )
 
     script.generate_revision(
@@ -483,10 +446,10 @@ def downgrade():
     write_script(
         script,
         f,
-        f"""\
+        """\
 "Rev F"
-revision = '{f}'
-down_revision = '{b}'
+revision = '%s'
+down_revision = '%s'
 
 from alembic import op
 
@@ -498,7 +461,8 @@ def upgrade():
 def downgrade():
     op.execute("DROP STEP 6")
 
-""",
+"""
+        % (f, b),
     )
 
     return d, e, f
@@ -507,25 +471,25 @@ def downgrade():
 def _multidb_testing_config(engines):
     """alembic.ini fixture to work exactly with the 'multidb' template"""
 
-    dir_ = _join_path(_get_staging_directory(), "scripts")
+    dir_ = os.path.join(_get_staging_directory(), "scripts")
 
     sqlalchemy_future = "future" in config.db.__class__.__module__
 
     databases = ", ".join(engines.keys())
     engines = "\n\n".join(
-        f"[{key}]\nsqlalchemy.url = {value.url}"
+        "[%s]\n" "sqlalchemy.url = %s" % (key, value.url)
         for key, value in engines.items()
     )
 
     return _write_config_file(
-        f"""
+        """
 [alembic]
-script_location = {dir_}
+script_location = %s
 sourceless = false
-sqlalchemy.future = {"true" if sqlalchemy_future else "false"}
-databases = {databases}
+sqlalchemy.future = %s
+databases = %s
 
-{engines}
+%s
 [loggers]
 keys = root
 
@@ -533,7 +497,7 @@ keys = root
 keys = console
 
 [logger_root]
-level = WARNING
+level = WARN
 handlers = console
 qualname =
 
@@ -550,8 +514,5 @@ keys = generic
 format = %%(levelname)-5.5s [%%(name)s] %%(message)s
 datefmt = %%H:%%M:%%S
     """
+        % (dir_, "true" if sqlalchemy_future else "false", databases, engines)
     )
-
-
-def _join_path(base: str, *more: str):
-    return str(Path(base).joinpath(*more).as_posix())