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 23 import os 24 import re 25 import sys 26 import time 27 import fnmatch 28 29 SRC_FILE_PATTERNS = [ "*.c", "*.h", "*.cpp", "*.hpp", "*.inl", "*.java", "*.aidl", "*.py" ] 30 COPYRIGHT_PATTERN = r'Copyright \(C\) ([0-9]{4})(-[0-9]{4})? drawElements Ltd.' 31 COPYRIGHT_REPLACEMENT = r'Copyright (C) \1-' + time.strftime("%Y") + r' drawElements Ltd.' 32 33 def isSrcFile (filename): 34 for pattern in SRC_FILE_PATTERNS: 35 if fnmatch.fnmatch(filename, pattern): 36 return True 37 return False 38 39 def 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 47 def 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 57 def processDir (dir): 58 srcFiles = findSrcFiles(dir) 59 for file in srcFiles: 60 processFile(file) 61 62 if __name__ == "__main__": 63 if len(sys.argv) < 2: 64 print sys.argv[0] + ": [directory]" 65 else: 66 processDir(sys.argv[1]) 67
