compile_all.py

import sys, os
from subprocess import call, check_output, Popen, PIPE
from lazyme.string import color_print

path = '.'
action = 'compile'

def file_exists(file_path):
    if not file_path:
        return False
    else:
        return os.path.isfile(file_path)

def main():
  for root, dirs, files in os.walk(path):
    print('Checking ' + root)
    makefile = os.path.join(root, "Makefile")
    if file_exists(makefile):
      cmd = 'cd ' + root + '; make ' + action
      #cmd = 'ls -la'
      pipes = Popen(cmd, shell=True, stdout=PIPE, stderr=PIPE)
      std_out, std_err = pipes.communicate()
      
      if (action == 'compile') | (action == 'run'):
        if pipes.returncode != 0:
          # an error happened!
          err_msg = "%s. Code: %s" % (std_err.strip(), pipes.returncode)
          color_print('[E] Error on ' + root + ': ', color='red', bold=True)
          print(err_msg)
        elif len(std_err):
          # return code is 0 (no error), but we may want to
          # do something with the info on std_err
          # i.e. logger.warning(std_err)
          color_print('[OK]', color='green')
        else:
          color_print('[OK]', color='green')
      if action == 'measure':
        call(['sleep', '5'])

if __name__ == '__main__':
  if len(sys.argv) == 2:
    act = sys.argv[1]
    if (act == 'compile') | (act == 'run') | (act == 'clean') | (act == 'measure'):
      color_print('Performing \"' + act + '\" action...', color='yellow', bold=True)
      action = act
    else:
      color_print('Error: Unrecognized action \"' + act + '\"', color='red')
      sys.exit(1)
  else:
    color_print('Performing \"compile\" action...', color='yellow', bold=True)
    action = 'compile'
  
  main()