Чи може рядок коду 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.