provingground/platform_whitelist_tools/remove_from_whitelist.py

#!/usr/bin/python
# Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

__author__ = 'deepakg@google.com'

"""
Remove the filtered lines in temp file from the whitelist files.
"""

import os
import sys
import tempfile


def main():
    """
    Remove lines from the whitelist.
    """
    if len(sys.argv)!=3:
        print ('InvalidArguments: We need 2 arguments. \n'
               'example: remove_from_whitelist.py gold_whitelist gold_temp.')
    else:
        whitelist_name = sys.argv[1]
        tempFile = sys.argv[2]
        if 'whitelist' in str(sys.argv[2]):
            tempFile = sys.argv[1]
            whitelist_name = sys.argv[2]
        whitelist = open(whitelist_name, 'r')
        remove = [x.strip() for x in open(tempFile, 'r').readlines()]
        output = tempfile.NamedTemporaryFile(dir=os.path.dirname(whitelist_name),
                                             delete=False)
        for line in whitelist:
            if line.startswith('#'):
                continue
            if line.strip() in remove:
                continue
            output.write(line)
        output.close()
        whitelist.close()
        os.rename(output.name, whitelist_name)
        print 'We have re-written the file ' + str(whitelist_name)


if __name__ == '__main__':
    main()