mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-06 17:01:07 -04:00
Mike Bayer
be8ffcfa4d
for py314b2 all issues should be resolved py314: yes Change-Id: I498a1f623aeb5eb664289236e01e35d8a3dec99f
79 lines
2.0 KiB
Python
79 lines
2.0 KiB
Python
"""Synchronizes test files that require the future annotation import.
|
|
|
|
.. versionadded:: 2.0
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from tempfile import NamedTemporaryFile
|
|
from typing import Any
|
|
from typing import Iterable
|
|
|
|
from sqlalchemy.util.tool_support import code_writer_cmd
|
|
|
|
header = '''\
|
|
"""This file is automatically generated from the file
|
|
{source!r}
|
|
by the {this_file!r} script.
|
|
|
|
Do not edit manually, any change will be lost.
|
|
""" # noqa: E501
|
|
|
|
from __future__ import annotations
|
|
|
|
'''
|
|
|
|
home = Path(__file__).parent.parent
|
|
this_file = Path(__file__).relative_to(home).as_posix()
|
|
remove_str = "# anno only: "
|
|
|
|
|
|
def run_operation(
|
|
name: str, source: str, dest: str, cmd: code_writer_cmd
|
|
) -> None:
|
|
source_data = Path(source).read_text().replace(remove_str, "")
|
|
dest_data = header.format(source=source, this_file=this_file) + source_data
|
|
|
|
with NamedTemporaryFile(
|
|
mode="w",
|
|
delete=False,
|
|
suffix=".py",
|
|
) as buf:
|
|
buf.write(dest_data)
|
|
|
|
cmd.run_black(buf.name)
|
|
cmd.write_output_file_from_tempfile(buf.name, dest)
|
|
|
|
|
|
def main(file: str, cmd: code_writer_cmd) -> None:
|
|
if file == "all":
|
|
operations: Iterable[Any] = files.items()
|
|
else:
|
|
operations = [(file, files[file])]
|
|
|
|
for name, info in operations:
|
|
run_operation(name, info["source"], info["dest"], cmd)
|
|
|
|
|
|
files = {
|
|
"typed_annotation": {
|
|
"source": "test/orm/declarative/test_typed_mapping.py",
|
|
"dest": "test/orm/declarative/test_tm_future_annotations_sync.py",
|
|
},
|
|
"dc_typed_annotation": {
|
|
"source": "test/orm/declarative/test_dc_transforms.py",
|
|
"dest": "test/orm/declarative/test_dc_transforms_future_anno_sync.py",
|
|
},
|
|
}
|
|
|
|
if __name__ == "__main__":
|
|
cmd = code_writer_cmd(__file__)
|
|
with cmd.add_arguments() as parser:
|
|
parser.add_argument(
|
|
"--file", choices=list(files) + ["all"], default="all"
|
|
)
|
|
|
|
with cmd.run_program():
|
|
main(cmd.args.file, cmd)
|