Extract C Code from source code automatically. Good for unittesting. -- requires ctags

extract_c_code.py

import sys

cmd = sys.argv[1]
f = open(sys.argv[2])
lines = [ l for l in f ]
f2 = sys.argv[3]

def find_struct_or_var_range(lines, i):
    pass

def find_function_range(lines, i):
    bracket=0
    end = i
    start = i
    while True:
        l = lines[end-1]
        go_out = 0
        for c in l:
            if c == '{':
                bracket += 1
            if c == '}':
                bracket -= 1
                if bracket == 0:
                    go_out = 1;
                    break
        if go_out == 1:
            break
        end += 1
    while True:
        l = lines[start-2]
        l = l.strip()
        if l[-1:] == '}' or l[-1:] == ';':
            break
        start -= 1
    return (start,end)

if cmd == '--func':
    n = int(sys.argv[3])
    (i,j) = find_function_range(lines,n)
    print "".join(lines[i:j])
elif cmd == '--decl':
    n = int(sys.argv[3])
    (i,j) = find_struct_or_var_range(lines,n)
    print "".join(lines[i:j])
elif cmd == '--update':
    (i,j) = (0,0)
    for index, l in enumerate(lines):
        if l == '/* PASTE THE CODE START */\n':
            i = index;
        if l == '/* PASTE THE CODE END */\n':
            j = index;
    if (i,j) == (0,0):
        print sys.argv[2] + " is not valid unittest file format\n"
        exit(1)
    of = open(sys.argv[3])
    output = lines[:i+1]
    for l in of:
       output.append(l)
    output.extend(lines[j:])
    print "".join(output)

python extract_c_code.py --cmds source.c <line_number | unittest.c>

#extract definition of functions
python extract_c_code.py --func source.c line_number