Kaynağa Gözat

Add script to add missing copyright headers

Matthias Reso 1 yıl önce
ebeveyn
işleme
f44ccffba9
1 değiştirilmiş dosya ile 17 ekleme ve 0 silme
  1. 17 0
      scripts/check_copyright_header.py

+ 17 - 0
scripts/check_copyright_header.py

@@ -0,0 +1,17 @@
+import re
+from pathlib import Path
+
+WORK_DIR = Path(__file__).parents[1]
+PATTERN = "(Meta Platforms, Inc. and affiliates)|(Facebook, Inc(\.|,)? and its affiliates)|([0-9]{4}-present(\.|,)? Facebook)|([0-9]{4}(\.|,)? Facebook)"
+
+HEADER = """# Copyright (c) Meta Platforms, Inc. and affiliates.
+# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.\n\n"""
+
+if __name__ == "__main__":
+    for ext in ["*.py", "*.sh"]:
+        for file in WORK_DIR.rglob(ext):
+            text = file.read_text()
+            if not re.search(PATTERN, text):
+                text = HEADER + text
+                file.write_text(text)
+