1#!/usr/bin/python 2# 3# Copyright 2010 Google Inc. All Rights Reserved. 4 5"""Script to clear lock directories for latest builds; rescheduling tests. 6 7Running this script will remove all test locks for the latest builds of every 8board. Which will cause all tests whose locks were removed to be rescheduled the 9next time Test Scheduler runs. The latest build is determined by the contents of 10the LATEST file in each board directory. 11 12By default the script will process all boards under DEFAULT_IMAGES_PATH. To 13process only specific boards and/or a different images directory, use the -b 14and/or -i options. 15 16Expected images directory structure looks like: 17 18 <board_0>\LATEST 19 <board_0>\<build_0>\netbook_<test_0> 20 <board_0>\<build_0>\netbook_<test_1> 21 . 22 . 23 . 24 <board_0>\<build_0>\netbook_<test_n> 25 . 26 . 27 . 28 <board_n>\LATEST 29 <board_n>\<build_0>\netbook_<test_0> 30 <board_n>\<build_0>\netbook_<test_n> 31""" 32 33__author__ = 'dalecurtis@google.com (Dale Curtis)' 34 35import optparse 36import os 37 38 39# Path to Dev Server's images directory. 40DEFAULT_IMAGES_PATH = '/usr/local/google/images' 41 42 43def ParseOptions(): 44 """Parse command line options. Returns options structure.""" 45 46 parser = optparse.OptionParser('usage: %prog [options]') 47 48 parser.add_option('-b', '--boards', dest='boards', 49 help='Comma separated list of boards to process. By default' 50 ' all boards in the images directory will be processed.') 51 parser.add_option('-i', '--images', dest='images', 52 help='Path to Dev Server images directory. Defaults to %s' % 53 DEFAULT_IMAGES_PATH, default=DEFAULT_IMAGES_PATH) 54 55 options = parser.parse_args()[0] 56 57 if not os.path.exists(options.images): 58 parser.error('The specified images directory (%s) does not exist. Please ' 59 'specify another.' % options.images) 60 61 if options.boards: 62 options.boards = options.boards.split(',') 63 64 return options 65 66 67def main(): 68 options = ParseOptions() 69 70 os.chdir(options.images) 71 72 # Build board listing. 73 if options.boards: 74 boards = options.boards 75 else: 76 boards = [board for board in os.listdir('.') if os.path.isdir(board)] 77 78 for board in boards: 79 latest_path = os.path.join(board, 'LATEST') 80 81 # Make sure directory contains a LATEST file. 82 if not os.path.exists(latest_path): 83 continue 84 85 build_path = os.path.join(board, open(latest_path, 'r').read().strip()) 86 87 # Make sure LATEST file points to a valid build. 88 if not os.path.exists(build_path): 89 continue 90 91 # Remove test locks in latest build directory. 92 for test in os.listdir(build_path): 93 test_path = os.path.join(build_path, test) 94 95 # Only remove directories we know (well, pretty sure) are test locks. 96 if not os.path.isdir(test_path) or not test.startswith('netbook_'): 97 continue 98 99 print 'Removing lock %s' % test_path 100 os.rmdir(test_path) 101 102 103if __name__ == '__main__': 104 main() 105