scripts/update-copyright-year.py

   1 # -*- coding: utf-8 -*-
   2
   3 import os
   4 import re
   5 import sys
   6 import time
   7 import fnmatch
   8
   9 SRC_FILE_PATTERNS               = [ "*.c", "*.h", "*.cpp", "*.hpp", "*.inl", "*.java", "*.aidl", "*.py" ]
  10 COPYRIGHT_PATTERN               = r'Copyright \(C\) ([0-9]{4})(-[0-9]{4})? drawElements Ltd.'
  11 COPYRIGHT_REPLACEMENT   = r'Copyright (C) \1-' + time.strftime("%Y") + r' drawElements Ltd.'
  12
  13 def isSrcFile (filename):
  14         for pattern in SRC_FILE_PATTERNS:
  15                 if fnmatch.fnmatch(filename, pattern):
  16                         return True
  17         return False
  18
  19 def findSrcFiles (dir):
  20         srcFiles = []
  21         for root, dirs, files in os.walk(dir):
  22                 for file in files:
  23                         if isSrcFile(file):
  24                                 srcFiles.append(os.path.join(root, file))
  25         return srcFiles
  26
  27 def processFile (filename):
  28         print filename
  29         file = open(filename, "rb")
  30         data = file.read()
  31         file.close()
  32         data = re.sub(COPYRIGHT_PATTERN, COPYRIGHT_REPLACEMENT, data)
  33         file = open(filename, "wb")
  34         file.write(data)
  35         file.close()
  36
  37 def processDir (dir):
  38         srcFiles = findSrcFiles(dir)
  39         for file in srcFiles:
  40                 processFile(file)
  41
  42 if __name__ == "__main__":
  43         if len(sys.argv) < 2:
  44                 print sys.argv[0] + ": [directory]"
  45         else:
  46                 processDir(sys.argv[1])