1# -*- coding: utf-8 -*- 2 3#------------------------------------------------------------------------- 4# drawElements Quality Program utilities 5# -------------------------------------- 6# 7# Copyright 2015 The Android Open Source Project 8# 9# Licensed under the Apache License, Version 2.0 (the "License"); 10# you may not use this file except in compliance with the License. 11# You may obtain a copy of the License at 12# 13# http://www.apache.org/licenses/LICENSE-2.0 14# 15# Unless required by applicable law or agreed to in writing, software 16# distributed under the License is distributed on an "AS IS" BASIS, 17# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 18# See the License for the specific language governing permissions and 19# limitations under the License. 20# 21#------------------------------------------------------------------------- 22 23import os 24import re 25import sys 26import time 27import fnmatch 28 29SRC_FILE_PATTERNS = [ "*.c", "*.h", "*.cpp", "*.hpp", "*.inl", "*.java", "*.aidl", "*.py" ] 30COPYRIGHT_PATTERN = r'Copyright \(C\) ([0-9]{4})(-[0-9]{4})? drawElements Ltd.' 31COPYRIGHT_REPLACEMENT = r'Copyright (C) \1-' + time.strftime("%Y") + r' drawElements Ltd.' 32 33def isSrcFile (filename): 34 for pattern in SRC_FILE_PATTERNS: 35 if fnmatch.fnmatch(filename, pattern): 36 return True 37 return False 38 39def findSrcFiles (dir): 40 srcFiles = [] 41 for root, dirs, files in os.walk(dir): 42 for file in files: 43 if isSrcFile(file): 44 srcFiles.append(os.path.join(root, file)) 45 return srcFiles 46 47def processFile (filename): 48 print(filename) 49 file = open(filename, "rb") 50 data = file.read() 51 file.close() 52 data = re.sub(COPYRIGHT_PATTERN, COPYRIGHT_REPLACEMENT, data) 53 file = open(filename, "wb") 54 file.write(data) 55 file.close() 56 57def processDir (dir): 58 srcFiles = findSrcFiles(dir) 59 for file in srcFiles: 60 processFile(file) 61 62if __name__ == "__main__": 63 if len(sys.argv) < 2: 64 print(sys.argv[0] + ": [directory]") 65 else: 66 processDir(sys.argv[1]) 67