Update multiple licence year ends in one run - close #94

pre_commit_hooks/insert_license.py

@@ -396,14 +396,15 @@ def try_update_year_range(
     :return: source file contents and a flag indicating update
     """
     current_year = datetime.now().year
+    changed = False
     for i in range(license_header_index, license_header_index + license_length):
         updated = try_update_year(
             src_file_content[i], src_filepath, current_year, introduce_range=True
         )
         if updated:
             src_file_content[i] = updated
-            return src_file_content, True
-    return src_file_content, False
+            changed = True
+    return src_file_content, changed
 
 
 def license_found(

tests/insert_license_test.py

@@ -425,6 +425,20 @@ def _convert_line_ending(file_path, new_line_endings):
                     ),
                 ),
             ),
+            product(
+                ("LICENSE_with_multiple_year_ranges.txt",),
+                ("\n",),
+                (
+                    (
+                        "module_with_multiple_stale_years_in_license.py",
+                        "#",
+                        "module_with_multiple_years_in_license.py",
+                        "",
+                        True,
+                        ["--use-current-year"],
+                    ),
+                ),
+            ),
         ),
     ),
 )

tests/resources/LICENSE_with_multiple_year_ranges.txt

@@ -0,0 +1,5 @@
+Copyright (C) 2016-2017 Teela O'Malley
+
+Licensed under the Apache License, Version 2.0 (the "License");
+
+@copyright 2012-2015 Teela O'Malley

tests/resources/module_with_multiple_stale_years_in_license.py

@@ -0,0 +1,8 @@
+# Copyright (C) 2012-2015 Teela O'Malley
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+#
+# @copyright 2012-2015 Teela O'Malley
+
+import sys
+sys.stdout.write("FOO")

tests/resources/module_with_multiple_years_in_license.py

@@ -0,0 +1,8 @@
+# Copyright (C) 2012-2017 Teela O'Malley
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+#
+# @copyright 2012-2017 Teela O'Malley
+
+import sys
+sys.stdout.write("FOO")