Add post_processing code to update copyright year automatically.

post_process/mangle.py

+# Copyright (c) 2011 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from __future__ import with_statement
+import os
+import re
+
+
+class MangleBase(object):
+  """For each of the modified file, process the file."""
+  def __init__(self, max_lines_search=None, once=False):
+    self.max_lines_search = max_lines_search
+    self.once = once
+
+  def process(self, checkout, patches):
+    for patch in patches.patches:
+      if patch.is_delete or patch.is_binary:
+        pass
+      filepath = os.path.join(checkout.project_path, patch.filename)
+      with open(filepath, 'rb') as f:
+        content = f.read().splitlines(True)
+      if not content:
+        continue
+      if self.process_lines(content):
+        with open(filepath, 'wb') as f:
+          f.write(''.join(content))
+
+  def process_lines(self, lines):
+    """Modifies lines in-place."""
+    modified = False
+    for i in xrange(min(self.max_lines_search, len(lines))):
+      old_line = lines[i]
+      lines[i] = self.process_line(lines[i])
+      modified |= (old_line != lines[i])
+      if self.once and modified:
+        break
+    return modified
+
+  def process_line(self, line):
+    raise NotImplementedError('Modifies a single line')
+
+
+class CopyrightMangle(MangleBase):
+  """Process the copyright in the first 5 lines."""
+  def __init__(self, pattern, replacement, max_lines_search=5):
+    """Replaces |pattern| with |replacement| once if found within first
+    |max_lines_search| lines of the file.
+    """
+    super(CopyrightMangle, self).__init__(max_lines_search, True)
+    self.pattern = pattern
+    self.replacement = replacement
+
+  def process_line(self, line):
+    return re.sub(self.pattern, self.replacement, line)