in scripts/generate_imports.py [0:0]
def generate_imports_plain(filename, classes):
with io.StringIO() as file:
# Sort classes by qualified name
classes = sorted(classes, key=lambda tup: classname(tup[1]))
print(copyright_header, file=file)
print(autogen_msg, file=file)
for s, cls in classes:
print(f"from {cls.__module__} import {s}", file=file)
print("", file=file)
print("__all__ = [", file=file)
all_list = ",\n\t".join([f'"{s}"' for s, _ in classes])
print("\t", end="", file=file)
print(all_list, file=file)
print("]", end="", file=file)
contents = file.getvalue()
with open(filename, "w") as real_file:
print(
black.format_file_contents(contents, fast=fast, mode=mode),
file=real_file,
end="",
)