pythonstartup

def autoimport(*modules):
    autoimported = set()
    for module in modules:
        if module.endswith(".*"):
            base = module[:-2]
            try:
                exec(
                    f"from {base} import *",
                    globals(),
                    globals(),
                )
            except ImportError:
                print(f"Failed to import {module}")
            autoimported.add(module)
        elif "." in module:
            base, name = module.rsplit(".", 1)
            try:
                exec(
                    f"from {base} import {name}",
                    globals(),
                    globals(),
                )
            except ImportError:
                print(f"Failed to import {module}")
            autoimported.add(name)
        else:
            try:
                exec(
                    f"import {module}",
                    globals(),
                    globals(),
                )
            except ImportError:
                print(f"Failed to import {module}")
            autoimported.add(module)

    print("Auto-imported for your convenience:")
    print(", ".join(sorted(autoimported)))


import os

if not os.environ.get("IN_REGRTEST"):
    autoimport(
        "sys",
        "os",
        "re",
        "datetime",
        "time",
        "asyncio",
        "typing.*",
        "pathlib.Path",
        "dataclasses.dataclass",
    )