Printing the code line number

[arbiel]

Hi

I am looking for the code to return the line number of the call, so I could code

print(line_number())

Arbiel

[Gribouillis]

Use my printat() function

def printat(*args, **kwargs):
    """Print function with additional line number and filename information.

    Adds a string such as "at line 31 in foo.py" to the printed output,
    to indicate the position where the printat() function was called.

    All the calls to print() in a program can be changed
    to provide additional information by adding

        print = printat

    at the top of the program.
    """
    import os, sys
    level = kwargs.pop('level', 0)
    frame = sys._getframe(level+1)
    try:
        lineno, code = frame.f_lineno, frame.f_code
        args += (f'at line {lineno} in {os.path.basename(code.co_filename)}',)
    finally:
        del frame
    print(*args, **kwargs)

[arbiel]

Hi Gribouillis
Thank you
It works fine.
Arbiel

[arbiel]

Hi Gribouillis
Maybe, you should add a piece of advice, to not include the printat fonction in the program, but to record it apart and import it ; else the "print(*args, **kwargs)" at line 24 generates an endless recursive call to printat.

[deanhystad]

There is no line 24, nor is there any kind of recursion. Can you post code that demonstrates the problem you see?

[Gribouillis]

"print(*args, **kwargs)" at line 24 generates an endless recursive call to printat.

Yes this happens if you write print = printat in the file where the printat function is defined. You can avoid recursion by writing

from builtins import print as _print
def printat(...)
    ....
    _print(*args, **kwargs) # note the _print instead of print

Now if you add print = printat somewhere in the file, it won't generate a recursion.

[arbiel]

Hi

Regarding the line number, I made the error to get it from the file I recorded and in which I inserted two lines in front of gribouillis' function.

Having recorded the file with the name 'ap_gbr_printat.py', I did the following :

import ap_gbr_printat as prt
print = prt.printat