aboutsummaryrefslogtreecommitdiffstats
path: root/library/python/runtime_py3/entry_points.py
blob: 37d9af864ab4c36fd5067185a60cd52773fffac6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import sys

import __res


def repl():
    user_ns = {}
    py_main = __res.find("PY_MAIN")

    if py_main:
        mod_name, func_name = (py_main.split(b":", 1) + [None])[:2]
        try:
            import importlib

            mod = importlib.import_module(mod_name.decode("UTF-8"))
            user_ns = mod.__dict__
        except ModuleNotFoundError:
            import traceback

            traceback.print_exc()

        if func_name and "__main__" not in user_ns:

            def run(args):
                if isinstance(args, str):
                    import shlex

                    args = shlex.split(args)

                import sys

                sys.argv = [sys.argv[0]] + args
                getattr(mod, func_name)()

            user_ns["__main__"] = run

    try:
        import IPython
    except ModuleNotFoundError:
        pass
    else:
        return IPython.start_ipython(user_ns=user_ns)

    import code

    code.interact(local=user_ns)


def resource_files():
    sys.stdout.buffer.write(b"\n".join(sorted(__res.resfs_files()) + [b""]))


def run_constructors():
    for key, module_name in __res.iter_keys(b"py/constructors/"):
        import importlib

        module = importlib.import_module(module_name.decode())
        init_func = getattr(module, __res.find(key).decode())
        init_func()