Bisakah satu baris kode Python mengetahui tingkat lekukan lekukannya?

Dari sesuatu seperti ini:

print(get_indentation_level())

    print(get_indentation_level())

        print(get_indentation_level())

Saya ingin mendapatkan sesuatu seperti ini:

1
2
3

Bisakah kode membaca sendiri dengan cara ini?

Yang saya inginkan adalah output dari bagian kode yang lebih bersarang menjadi lebih bersarang. Dengan cara yang sama ini membuat kode lebih mudah dibaca, itu akan membuat output lebih mudah dibaca.

Tentu saja saya bisa mengimplementasikan ini secara manual, menggunakan eg .format(), tetapi yang ada dalam pikiran saya adalah fungsi cetak kustom yang akan di print(i*' ' + string)mana itingkat indentasi. Ini akan menjadi cara cepat untuk membuat output yang dapat dibaca di terminal saya.

Apakah ada cara yang lebih baik untuk melakukan ini yang menghindari pemformatan manual yang telaten?

Jika Anda ingin lekukan dalam hal tingkat bersarang daripada spasi dan tab, segalanya menjadi rumit. Misalnya, dalam kode berikut:

if True:
    print(
get_nesting_level())

panggilan untuk get_nesting_levelsebenarnya bersarang sedalam satu tingkat, meskipun faktanya tidak ada spasi putih pada saluran get_nesting_levelpanggilan tersebut. Sementara itu, dalam kode berikut:

print(1,
      2,
      get_nesting_level())

panggilan untuk get_nesting_levelbersarang level nol dalam, meskipun ada spasi putih terkemuka di jalurnya.

Dalam kode berikut:

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

if True:
    print(get_nesting_level())

dua panggilan untuk get_nesting_levelberada pada tingkat bersarang yang berbeda, meskipun fakta bahwa spasi putih terkemuka identik.

Dalam kode berikut:

if True: print(get_nesting_level())

apakah itu level nol bersarang, atau satu? Dalam hal INDENTdan DEDENTtoken dalam tata bahasa formal, levelnya nol dalam, tetapi Anda mungkin tidak merasakan hal yang sama.

Jika Anda ingin melakukan ini, Anda harus memberi token seluruh file hingga titik panggilan dan menghitung INDENTserta DEDENTtoken. The tokenizeModul akan sangat berguna untuk fungsi seperti:

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

Ya, itu pasti mungkin, berikut ini contoh kerjanya:

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

Anda dapat menggunakan sys.current_frame.f_linenountuk mendapatkan nomor baris. Kemudian untuk menemukan jumlah level indentasi, Anda perlu menemukan baris sebelumnya dengan indentasi nol lalu mengurangkan nomor baris saat ini dari nomor baris itu Anda akan mendapatkan jumlah indentasi:

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

Jika Anda ingin jumlah level indentasi berdasarkan garis previouse dengan :Anda bisa melakukannya dengan sedikit perubahan:

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

Dan sebagai jawaban alternatif di sini adalah fungsi untuk mendapatkan jumlah indentasi (spasi putih):

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

Untuk memecahkan masalah "nyata" yang mengarah ke pertanyaan Anda, Anda bisa menerapkan manajer konteks yang melacak tingkat indensi dan membuat withstruktur blok dalam kode sesuai dengan tingkat indentasi output. Dengan cara ini lekukan kode masih mencerminkan lekukan output tanpa terlalu banyak menyambung. Dimungkinkan untuk memperbaiki kode menjadi fungsi yang berbeda dan memiliki lekukan lain berdasarkan pada struktur kode yang tidak mengacaukan lekukan keluaran.

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

Keluaran:

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.