บรรทัดของโค้ด 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

หากคุณต้องการจำนวนระดับการเยื้องตามบรรทัด previouse ด้วย:คุณสามารถทำได้โดยการเปลี่ยนแปลงเล็กน้อย:

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โครงสร้างบล็อกในรหัสสอดคล้องกับระดับการเยื้องของเอาต์พุต วิธีนี้เยื้องรหัสยังคงสะท้อนถึงการเยื้องเอาต์พุตโดยไม่ต้องเชื่อมต่อทั้งสองมากเกินไป ยังคงเป็นไปได้ที่จะ refactor รหัสลงในฟังก์ชั่นที่แตกต่างกันและมีการเยื้องอื่น ๆ ขึ้นอยู่กับโครงสร้างรหัสที่ไม่ยุ่งกับการเยื้องเอาท์พุท

#!/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.