Может ли строка кода Python знать свой уровень вложенности отступа?

Из чего-то вроде этого:

print(get_indentation_level())

    print(get_indentation_level())

        print(get_indentation_level())

Я хотел бы получить что-то вроде этого:

1
2
3

Может ли код читать себя таким образом?

Все, что я хочу, - это вывод из более вложенных частей кода, чтобы быть более вложенными. Точно так же, как это облегчает чтение кода, облегчает чтение вывода.

Конечно, я мог бы реализовать это вручную, используя, например .format(), но я имел в виду пользовательскую функцию печати, print(i*' ' + string)где iбудет уровень отступа. Это был бы быстрый способ сделать читаемый вывод на моем терминале.

Есть ли лучший способ сделать это, чтобы избежать кропотливого ручного форматирования?

Если вы хотите сделать отступ с точки зрения уровня вложенности, а не пробелов и табуляций, все будет сложно. Например, в следующем коде:

if True:
    print(
get_nesting_level())

призыв к get_nesting_level фактически вложен на один уровень глубиной, несмотря на то, что в строке get_nesting_levelвызова нет начальных пробелов . Между тем, в следующем коде:

print(1,
      2,
      get_nesting_level())

призыв к get_nesting_levelвложенным нулевым уровням глубины, несмотря на наличие ведущих пробелов на его линии.

В следующем коде:

if True:
  if True:
    print(get_nesting_level())

if True:
    print(get_nesting_level())

два вызова get_nesting_level находятся на разных уровнях вложенности, несмотря на то, что начальные пробелы идентичны.

В следующем коде:

if True: print(get_nesting_level())

это вложенные нулевые уровни или один? С точки зрения INDENTиDEDENT токенов в формальной грамматике, это нулевой уровень, но вы можете не чувствовать себя так же.

Если вы хотите сделать это, вам нужно будет токенизировать весь файл до точки вызова, подсчета INDENTи DEDENTтокенов. tokenizeМодуль был бы очень полезен для такой функции:

import inspect
import tokenize

def get_nesting_level():
    caller_frame = inspect.currentframe().f_back
    filename, caller_lineno, _, _, _ = inspect.getframeinfo(caller_frame)
    with open(filename) as f:
        indentation_level = 0
        for token_record in tokenize.generate_tokens(f.readline):
            token_type, _, (token_lineno, _), _, _ = token_record
            if token_lineno > caller_lineno:
                break
            elif token_type == tokenize.INDENT:
                indentation_level += 1
            elif token_type == tokenize.DEDENT:
                indentation_level -= 1
        return indentation_level

Да, это определенно возможно, вот рабочий пример:

import inspect

def get_indentation_level():
    callerframerecord = inspect.stack()[1]
    frame = callerframerecord[0]
    info = inspect.getframeinfo(frame)
    cc = info.code_context[0]
    return len(cc) - len(cc.lstrip())

if 1:
    print get_indentation_level()
    if 1:
        print get_indentation_level()
        if 1:
            print get_indentation_level()

Вы можете использовать sys.current_frame.f_linenoдля того, чтобы получить номер строки. Затем, чтобы найти номер уровня отступа, вам нужно найти предыдущую строку с нулевым отступом, а затем вычесть текущий номер строки из номера этой строки, и вы получите номер отступа:

import sys
current_frame = sys._getframe(0)

def get_ind_num():
    with open(__file__) as f:
        lines = f.readlines()
    current_line_no = current_frame.f_lineno
    to_current = lines[:current_line_no]
    previous_zoro_ind = len(to_current) - next(i for i, line in enumerate(to_current[::-1]) if not line[0].isspace())
    return current_line_no - previous_zoro_ind

Демо-версия:

if True:
    print get_ind_num()
    if True:
        print(get_ind_num())
        if True:
            print(get_ind_num())
            if True: print(get_ind_num())
# Output
1
3
5
6

Если вам нужен номер уровня отступа, основанный на предыдущих строках, :вы можете просто сделать это с небольшим изменением:

def get_ind_num():
    with open(__file__) as f:
        lines = f.readlines()

    current_line_no = current_frame.f_lineno
    to_current = lines[:current_line_no]
    previous_zoro_ind = len(to_current) - next(i for i, line in enumerate(to_current[::-1]) if not line[0].isspace())
    return sum(1 for line in lines[previous_zoro_ind-1:current_line_no] if line.strip().endswith(':'))

Демо-версия:

if True:
    print get_ind_num()
    if True:
        print(get_ind_num())
        if True:
            print(get_ind_num())
            if True: print(get_ind_num())
# Output
1
2
3
3

И в качестве альтернативного ответа здесь есть функция для получения номера отступа (пробела):

import sys
from itertools import takewhile
current_frame = sys._getframe(0)

def get_ind_num():
    with open(__file__) as f:
        lines = f.readlines()
    return sum(1 for _ in takewhile(str.isspace, lines[current_frame.f_lineno - 1]))

Чтобы решить «реальную» проблему, которая приводит к вашему вопросу, вы можете реализовать контекстный менеджер, который отслеживает уровень отступа и делает withструктуру блока в коде соответствующей уровням отступа выходных данных. Таким образом, отступ кода по-прежнему отражает отступ вывода без слишком сильного связывания обоих. Тем не менее все еще возможно преобразовать код в другие функции и иметь другие отступы, основанные на структуре кода, не мешая выходному отступу.

#!/usr/bin/env python
# coding: utf8
from __future__ import absolute_import, division, print_function


class IndentedPrinter(object):

    def __init__(self, level=0, indent_with='  '):
        self.level = level
        self.indent_with = indent_with

    def __enter__(self):
        self.level += 1
        return self

    def __exit__(self, *_args):
        self.level -= 1

    def print(self, arg='', *args, **kwargs):
        print(self.indent_with * self.level + str(arg), *args, **kwargs)


def main():
    indented = IndentedPrinter()
    indented.print(indented.level)
    with indented:
        indented.print(indented.level)
        with indented:
            indented.print('Hallo', indented.level)
            with indented:
                indented.print(indented.level)
            indented.print('and back one level', indented.level)


if __name__ == '__main__':
    main()

Вывод:

0
  1
    Hallo 2
      3
    and back one level 2

>>> import inspect
>>> help(inspect.indentsize)
Help on function indentsize in module inspect:

indentsize(line)
    Return the indent size, in spaces, at the start of a line of text.