PyBoardTypeshedGenerator/sys_.py at main · hlovatt/PyBoardTypeshedGenerator · GitHub

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
"""
Generate `pyi` from corresponding `rst` docs.
"""

import rst
from rst2pyi import RST2PyI

__author__ = rst.__author__
__copyright__ = rst.__copyright__
__license__ = rst.__license__
__version__ = "7.5.3"  # Version set by https://github.com/hlovatt/tag2ver


def sys(shed: RST2PyI) -> None:
    shed.module(
        name="sys",
        old="system specific functions",
        post_doc="""
from typing import Callable, Final, Literal, NoReturn

from uio import IOBase

class Implementation(tuple[str, tuple[int, int, int], int]):
    name: str
    version: tuple[int, int, int]
    mpy: int

class ModuleType:
    __class__: str
    __name__: str
""",
        end="Functions",
    )

    shed.def_(
        old=r".. function:: exit(retval=0, /)",
        new="def exit(retval: object = 0, /) -> NoReturn",
        indent=0,
    )
    shed.def_(
        old=r".. function:: atexit(func)",
        new="def atexit(func: Callable[[], None] | None, /) -> Callable[[], None] | None",
        indent=0,
    )
    shed.def_(
        old=r".. function:: print_exception(exc, file=sys.stdout, /)",
        new="def print_exception(exc: BaseException, file: IOBase[str] = 'stdout', /) -> None",
        indent=0,
        end="Constants",
    )

    shed.consume_minuses_underline_line(and_preceding_lines=True)
    shed.vars(
        old=r".. data:: argv", class_var=None, type_="list[str]",
    )
    shed.vars(
        old=r".. data:: byteorder", class_var=None, type_='Literal["little", "big"]',
    )
    shed.vars(
        old=r".. data:: implementation", class_var=None, type_="Implementation",
    )
    shed.vars(
        old=r".. data:: maxsize", class_var=None,
    )
    shed.vars(
        old=r".. data:: modules", class_var=None, type_="dict[str, ModuleType]",
    )
    shed.vars(
        old=r".. data:: path", class_var=None, type_="list[str]",
    )
    shed.vars(
        old=r".. data:: platform", class_var=None, type_="str",
    )
    shed.vars(
        old=r".. data:: stderr", class_var=None, type_="IOBase[str]",
    )
    shed.vars(
        old=r".. data:: stdin", class_var=None, type_="IOBase[str]",
    )
    shed.vars(
        old=r".. data:: stdout", class_var=None, type_="IOBase[str]",
    )
    shed.vars(
        old=r".. data:: version", class_var=None, type_="str",
    )
    shed.vars(
        old=r".. data:: version_info", class_var=None, type_="tuple[int, int, int]",
    )

    shed.write(u_also=True)