summaryrefslogtreecommitdiffstats
path: root/misc/scripts/copyright_headers.py
diff options
context:
space:
mode:
Diffstat (limited to 'misc/scripts/copyright_headers.py')
-rwxr-xr-xmisc/scripts/copyright_headers.py99
1 files changed, 50 insertions, 49 deletions
diff --git a/misc/scripts/copyright_headers.py b/misc/scripts/copyright_headers.py
index a5e2f0c05d..2b1201b3c0 100755
--- a/misc/scripts/copyright_headers.py
+++ b/misc/scripts/copyright_headers.py
@@ -1,6 +1,7 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
+import os
import sys
header = """\
@@ -35,61 +36,61 @@ header = """\
/**************************************************************************/
"""
-fname = sys.argv[1]
+if len(sys.argv) < 2:
+ print("Invalid usage of copyright_headers.py, it should be called with a path to one or multiple files.")
+ sys.exit(1)
-# Handle replacing $filename with actual filename and keep alignment
-fsingle = fname.strip()
-if fsingle.find("/") != -1:
- fsingle = fsingle[fsingle.rfind("/") + 1 :]
-rep_fl = "$filename"
-rep_fi = fsingle
-len_fl = len(rep_fl)
-len_fi = len(rep_fi)
-# Pad with spaces to keep alignment
-if len_fi < len_fl:
- for x in range(len_fl - len_fi):
- rep_fi += " "
-elif len_fl < len_fi:
- for x in range(len_fi - len_fl):
- rep_fl += " "
-if header.find(rep_fl) != -1:
- text = header.replace(rep_fl, rep_fi)
-else:
- text = header.replace("$filename", fsingle)
-text += "\n"
+for f in sys.argv[1:]:
+ fname = f
-# We now have the proper header, so we want to ignore the one in the original file
-# and potentially empty lines and badly formatted lines, while keeping comments that
-# come after the header, and then keep everything non-header unchanged.
-# To do so, we skip empty lines that may be at the top in a first pass.
-# In a second pass, we skip all consecutive comment lines starting with "/*",
-# then we can append the rest (step 2).
+ # Handle replacing $filename with actual filename and keep alignment
+ fsingle = os.path.basename(fname.strip())
+ rep_fl = "$filename"
+ rep_fi = fsingle
+ len_fl = len(rep_fl)
+ len_fi = len(rep_fi)
+ # Pad with spaces to keep alignment
+ if len_fi < len_fl:
+ for x in range(len_fl - len_fi):
+ rep_fi += " "
+ elif len_fl < len_fi:
+ for x in range(len_fi - len_fl):
+ rep_fl += " "
+ if header.find(rep_fl) != -1:
+ text = header.replace(rep_fl, rep_fi)
+ else:
+ text = header.replace("$filename", fsingle)
+ text += "\n"
-fileread = open(fname.strip(), "r")
-line = fileread.readline()
-header_done = False
+ # We now have the proper header, so we want to ignore the one in the original file
+ # and potentially empty lines and badly formatted lines, while keeping comments that
+ # come after the header, and then keep everything non-header unchanged.
+ # To do so, we skip empty lines that may be at the top in a first pass.
+ # In a second pass, we skip all consecutive comment lines starting with "/*",
+ # then we can append the rest (step 2).
-while line.strip() == "": # Skip empty lines at the top
- line = fileread.readline()
+ with open(fname.strip(), "r", encoding="utf-8") as fileread:
+ line = fileread.readline()
+ header_done = False
-if line.find("/**********") == -1: # Godot header starts this way
- # Maybe starting with a non-Godot comment, abort header magic
- header_done = True
+ while line.strip() == "" and line != "": # Skip empty lines at the top
+ line = fileread.readline()
-while not header_done: # Handle header now
- if line.find("/*") != 0: # No more starting with a comment
- header_done = True
- if line.strip() != "":
- text += line
- line = fileread.readline()
+ if line.find("/**********") == -1: # Godot header starts this way
+ # Maybe starting with a non-Godot comment, abort header magic
+ header_done = True
-while line != "": # Dump everything until EOF
- text += line
- line = fileread.readline()
+ while not header_done: # Handle header now
+ if line.find("/*") != 0: # No more starting with a comment
+ header_done = True
+ if line.strip() != "":
+ text += line
+ line = fileread.readline()
-fileread.close()
+ while line != "": # Dump everything until EOF
+ text += line
+ line = fileread.readline()
-# Write
-filewrite = open(fname.strip(), "w")
-filewrite.write(text)
-filewrite.close()
+ # Write
+ with open(fname.strip(), "w", encoding="utf-8", newline="\n") as filewrite:
+ filewrite.write(text)