Bir Python kodu satırı girintili iç içe geçme düzeyini bilebilir mi?

Böyle bir şeyden:

print(get_indentation_level())

    print(get_indentation_level())

        print(get_indentation_level())

Böyle bir şey almak istiyorum:

1
2
3

Kod bu şekilde kendini okuyabilir mi?

Tüm istediğim kodun daha iç içe bölümlerinden daha iç içe olması için çıktı. Bu, kodun okunmasını kolaylaştırdığı gibi, çıktıyı da okumayı kolaylaştırır.

Tabii ki bunu manuel olarak uygulayabilirim, örneğin .format(), ama aklımda olan şey, girinti seviyesinin print(i*' ' + string)nerede olacağı özel bir yazdırma iişleviydi. Bu, terminalimde okunabilir çıktı almanın hızlı bir yolu olacaktır.

El ile biçimlendirmeyi özenle önleyen bunu yapmanın daha iyi bir yolu var mı?

Boşluklar ve sekmeler yerine iç içe yerleştirme düzeyi açısından girinti istiyorsanız, işler zorlaşır. Örneğin, aşağıdaki kodda:

if True:
    print(
get_nesting_level())

çağrı get_nesting_levelhattında önde gelen boşluk olmamasına rağmen , çağrı aslında bir seviye derinliğe yerleştirilmiştir get_nesting_level. Bu arada, aşağıdaki kodda:

print(1,
      2,
      get_nesting_level())

get_nesting_levelhattında önde gelen boşluk olmasına rağmen , çağrı sıfır seviyelerin derinliklerine yerleştirilir.

Aşağıdaki kodda:

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

if True:
    print(get_nesting_level())

get_nesting_levelönde gelen boşluk aynı olmasına rağmen iki çağrı farklı yuvalama düzeyindedir.

Aşağıdaki kodda:

if True: print(get_nesting_level())

bu iç içe sıfır seviyeleri mi yoksa bir mi? Biçimsel dilbilgisi INDENTve DEDENTjetonları açısından sıfır derinliktedir, ancak aynı şekilde hissetmeyebilirsiniz.

Bunu yapmak istiyorsanız, tüm dosyayı arama ve sayma INDENTve DEDENTbelirteçler noktasına kadar jetonlamak zorunda kalacaksınız . tokenizeBu modül, bir işlev için çok yararlı olacaktır:

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

Evet, bu kesinlikle mümkün, işte çalışan bir örnek:

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_linenoSatır numarasını almak için kullanabilirsiniz . Ardından, girinti seviyesini bulmak için sıfır girintisi olan bir önceki satırı bulmanız ve ardından geçerli satır numarasını o satırın numarasından çıkarmanız gerekir, girintinin sayısını alırsınız:

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

Demo:

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

Ön çim çizgilerine dayanan girinti seviyesinin sayısını :istiyorsanız, sadece küçük bir değişiklikle yapabilirsiniz:

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(':'))

Demo:

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

Ve alternatif bir cevap olarak, girinti sayısını (beyaz boşluk) elde etmek için bir işlev var:

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]))

Sorunuza yol açan "gerçek" sorunu çözmek için, girinti düzeyini takip eden ve withkoddaki blok yapısının çıktının girinti düzeylerine karşılık gelmesini sağlayan bir bağlam yöneticisi uygulayabilirsiniz. Bu şekilde kod girintisi, her ikisini de çok fazla birleştirmeden çıkış girintisini yansıtır. Kodu farklı işlevlere yeniden aktive etmek ve çıkış yapısıyla karışmayan kod yapısına dayalı başka girintilere sahip olmak hala mümkündür.

#!/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()

Çıktı:

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.