Pastable version of filter by indentation level

<pre class='prettyprint' language='python'>#!/usr/bin/env python
import sys, os

try:
    treshold = int(sys.argv[1])
except IndexError:
    print &quot;usage: %s TRESHOLD [INPUT_FILE]&quot; % sys.argv[0]
    sys.exit(1)

try:
    f = open(sys.argv[2])
except IndexError:
    f = sys.stdin

indent_level = 0
lines_indented = 0
trunc_line_count = 0

whole = f.readlines()

def count_leading_spaces(input_str):
    space_count = 0
    for char in input_str:
        if char.isspace():
            space_count += 1
        else:
            return space_count

def print_truncated_notice():
    if trunc_line_count &gt; 0:
        # suffer extra pain to pretty print a &quot;truncated&quot; line
        print &quot; &quot;*(indent_level) + &quot;...&quot;, trunc_line_count, &quot;line&quot; + ('','s')[trunc_line_count != 1], &quot;truncated ...&quot;

for line in whole:
    leading_spaces = count_leading_spaces(line)
    if leading_spaces == indent_level:
        lines_indented += 1
        if lines_indented &gt; treshold:
            trunc_line_count += 1
        else:
            print line,
    else:
        print_truncated_notice()
        indent_level = leading_spaces
        trunc_line_count = 0
        lines_indented = 1
        print line,
# did we truncate at the last indent level ?
print_truncated_notice()</pre>
<a href="http://www.refactormycode.com/codes/1041-filter-by-indentation-level" style="color:#fff" title="As seen on RefactorMyCode.com"><img alt="Small_logo" src="http://www.refactormycode.com/images/small_logo.gif" style="border:0" /></a>

Refactor :my => 'code'

Recent

Popular

Pastable version of filter by indentation level