main commit

venv/lib/python3.12/site-packages/alembic/util/pyfiles.py

@@ -6,13 +6,9 @@ import importlib
 import importlib.machinery
 import importlib.util
 import os
-import pathlib
 import re
 import tempfile
-from types import ModuleType
-from typing import Any
 from typing import Optional
-from typing import Union
 
 from mako import exceptions
 from mako.template import Template
@@ -22,14 +18,9 @@ from .exc import CommandError
 
 
 def template_to_file(
-    template_file: Union[str, os.PathLike[str]],
-    dest: Union[str, os.PathLike[str]],
-    output_encoding: str,
-    *,
-    append_with_newlines: bool = False,
-    **kw: Any,
+    template_file: str, dest: str, output_encoding: str, **kw
 ) -> None:
-    template = Template(filename=_preserving_path_as_str(template_file))
+    template = Template(filename=template_file)
     try:
         output = template.render_unicode(**kw).encode(output_encoding)
     except:
@@ -45,13 +36,11 @@ def template_to_file(
             "template-oriented traceback." % fname
         )
     else:
-        with open(dest, "ab" if append_with_newlines else "wb") as f:
-            if append_with_newlines:
-                f.write("\n\n".encode(output_encoding))
+        with open(dest, "wb") as f:
             f.write(output)
 
 
-def coerce_resource_to_filename(fname_or_resource: str) -> pathlib.Path:
+def coerce_resource_to_filename(fname: str) -> str:
     """Interpret a filename as either a filesystem location or as a package
     resource.
 
@@ -59,9 +48,8 @@ def coerce_resource_to_filename(fname_or_resource: str) -> pathlib.Path:
     are interpreted as resources and coerced to a file location.
 
     """
-    # TODO: there seem to be zero tests for the package resource codepath
-    if not os.path.isabs(fname_or_resource) and ":" in fname_or_resource:
-        tokens = fname_or_resource.split(":")
+    if not os.path.isabs(fname) and ":" in fname:
+        tokens = fname.split(":")
 
         # from https://importlib-resources.readthedocs.io/en/latest/migration.html#pkg-resources-resource-filename  # noqa E501
 
@@ -71,48 +59,37 @@ def coerce_resource_to_filename(fname_or_resource: str) -> pathlib.Path:
         ref = compat.importlib_resources.files(tokens[0])
         for tok in tokens[1:]:
             ref = ref / tok
-        fname_or_resource = file_manager.enter_context(  # type: ignore[assignment]  # noqa: E501
+        fname = file_manager.enter_context(  # type: ignore[assignment]
             compat.importlib_resources.as_file(ref)
         )
-    return pathlib.Path(fname_or_resource)
+    return fname
 
 
-def pyc_file_from_path(
-    path: Union[str, os.PathLike[str]],
-) -> Optional[pathlib.Path]:
+def pyc_file_from_path(path: str) -> Optional[str]:
     """Given a python source path, locate the .pyc."""
 
-    pathpath = pathlib.Path(path)
-    candidate = pathlib.Path(
-        importlib.util.cache_from_source(pathpath.as_posix())
-    )
-    if candidate.exists():
+    candidate = importlib.util.cache_from_source(path)
+    if os.path.exists(candidate):
         return candidate
 
     # even for pep3147, fall back to the old way of finding .pyc files,
     # to support sourceless operation
-    ext = pathpath.suffix
+    filepath, ext = os.path.splitext(path)
     for ext in importlib.machinery.BYTECODE_SUFFIXES:
-        if pathpath.with_suffix(ext).exists():
-            return pathpath.with_suffix(ext)
+        if os.path.exists(filepath + ext):
+            return filepath + ext
     else:
         return None
 
 
-def load_python_file(
-    dir_: Union[str, os.PathLike[str]], filename: Union[str, os.PathLike[str]]
-) -> ModuleType:
+def load_python_file(dir_: str, filename: str):
     """Load a file from the given path as a Python module."""
 
-    dir_ = pathlib.Path(dir_)
-    filename_as_path = pathlib.Path(filename)
-    filename = filename_as_path.name
-
     module_id = re.sub(r"\W", "_", filename)
-    path = dir_ / filename
-    ext = path.suffix
+    path = os.path.join(dir_, filename)
+    _, ext = os.path.splitext(filename)
     if ext == ".py":
-        if path.exists():
+        if os.path.exists(path):
             module = load_module_py(module_id, path)
         else:
             pyc_path = pyc_file_from_path(path)
@@ -122,32 +99,12 @@ def load_python_file(
                 module = load_module_py(module_id, pyc_path)
     elif ext in (".pyc", ".pyo"):
         module = load_module_py(module_id, path)
-    else:
-        assert False
     return module
 
 
-def load_module_py(
-    module_id: str, path: Union[str, os.PathLike[str]]
-) -> ModuleType:
+def load_module_py(module_id: str, path: str):
     spec = importlib.util.spec_from_file_location(module_id, path)
     assert spec
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)  # type: ignore
     return module
-
-
-def _preserving_path_as_str(path: Union[str, os.PathLike[str]]) -> str:
-    """receive str/pathlike and return a string.
-
-    Does not convert an incoming string path to a Path first, to help with
-    unit tests that are doing string path round trips without OS-specific
-    processing if not necessary.
-
-    """
-    if isinstance(path, str):
-        return path
-    elif isinstance(path, pathlib.PurePath):
-        return str(path)
-    else:
-        return str(pathlib.Path(path))