#!/usr/bin/env python

'''
pycparser examples all contain the following boiler plate code
for running in tree. This script removes them:

# This is not required if you've installed pycparser into
# your site-packages/ with setup.py
#
sys.path.extend(['.', '..'])
'''

import sys
import os

boiler_plate = "sys.path.extend(['.', '..'])\n"
d = sys.argv[1]
for (root, dirs, files) in os.walk(d):
    for i in files:
        if not i.endswith('.py'):
            continue
        fname = os.path.join(root, i)
        lines = open(fname).readlines()
        try:
            start = lines.index(boiler_plate)
            end = start
        except ValueError:
            start = None
            end = start
        if start is not None:
            while lines[start-1].startswith('#'):
                start -= 1

        if start is not None and end is not None:
            f = open(fname, 'w')
            f.writelines(lines[:start])
            f.writelines(lines[end+1:])
            f.close()