build_modules.py

#!/usr/bin/python
#
# Copyright (c) 2015, Arista Networks, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#   Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
#
#   Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
#
#   Neither the name of Arista Networks nor the names of its
#   contributors may be used to endorse or promote products derived from
#   this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ARISTA NETWORKS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
# IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
import glob
import re
import argparse

common = open('common/eos.py').read()

start_re = re.compile(r'#<<EOS_COMMON_MODULE_START>>', re.M)
stop_re = re.compile(r'#<<EOS_COMMON_MODULE_END>>', re.M)

parser = argparse.ArgumentParser()

def build_parser():
    parser.add_argument('--module', '-m',
                        help='Only update the specified module')

def should_process(name, args=None):
    if name.endswith('__init__.py'):
        return False
    if not args.module and name.endswith('.py'):
        return True
    return name.endswith('%s.py' % args.module)

def main():
    build_parser()
    args = parser.parse_args()
    for name in glob.glob('library/*.py'):
        if should_process(name, args):
            print 'processing', name
            mod = open(name).read()

            start = start_re.search(mod, re.M)
            stop = stop_re.search(mod, re.M)

            with open(name, 'w') as f:
                f.write(mod[:start.start()].strip())
                f.write('\n%s\n\n' % common.strip())
                f.write(mod[stop.end():].strip())

if __name__ == '__main__':
    main()