summaryrefslogtreecommitdiffstatshomepage
path: root/tests/misc/sys_settrace_loop.py
blob: 06d0dc17bf77b3e9e8428bc1e3b467cf35732c32 (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
60
# test sys.settrace with while and for loops

import sys

try:
    sys.settrace
except AttributeError:
    print("SKIP")
    raise SystemExit


def print_stacktrace(frame, level=0):
    print(
        "%2d: %s@%s:%s => %s:%d"
        % (
            level,
            "  ",
            frame.f_globals["__name__"],
            frame.f_code.co_name,
            # reduce full path to some pseudo-relative
            "misc" + "".join(frame.f_code.co_filename.split("tests/misc")[-1:]),
            frame.f_lineno,
        )
    )

    if frame.f_back:
        print_stacktrace(frame.f_back, level + 1)


trace_count = 0


def trace_tick_handler(frame, event, arg):
    global trace_count
    print("### trace_handler::main event:", event)
    trace_count += 1
    print_stacktrace(frame)
    return trace_tick_handler


def test_loop():
    # for loop
    r = 0
    for i in range(3):
        r += i
    print("test_for_loop", r)

    # while loop
    r = 0
    i = 0
    while i < 3:
        r += i
        i += 1
    print("test_while_loop", i)


sys.settrace(trace_tick_handler)
test_loop()
sys.settrace(None)
print("Total traces executed: ", trace_count)