Include files in the root directory for license check

dtaniwaki · Apr 13, 2019 · f9a8212 · f9a8212
1 parent 5ace80e
commit f9a8212
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/scripts/license/add.py b/scripts/license/add.py
@@ -24,10 +24,15 @@
 
 PROJECT_ROOT = subprocess.check_output(
     "git rev-parse --show-toplevel".split()).strip().decode("utf-8")
-target_dirs = [os.path.join(PROJECT_ROOT, d) for d in os.listdir(PROJECT_ROOT) if d != "vendor"]
+target_dirs = [os.path.join(PROJECT_ROOT, d) for d in os.listdir(PROJECT_ROOT)
+               if d != "vendor" and os.path.isdir(os.path.join(PROJECT_ROOT, d))]
 
 
 def main(verbose=False):
+    add(Path(PROJECT_ROOT).glob("*.go"), license_header("//"), verbose)
+    add(Path(PROJECT_ROOT).glob("*.py"), license_header("#"), verbose)
+    add(Path(PROJECT_ROOT).glob("*.sh"), license_header("#"), verbose)
+
     for d in target_dirs:
         add(Path(d).glob("**/*_k8s.go"), license_header("//", modification=True), verbose)
         add([p for p in Path(d).glob("**/*.go")

diff --git a/scripts/license/check.py b/scripts/license/check.py
@@ -24,12 +24,17 @@
 
 PROJECT_ROOT = subprocess.check_output(
     "git rev-parse --show-toplevel".split()).strip().decode("utf-8")
-target_dirs = [os.path.join(PROJECT_ROOT, d) for d in os.listdir(PROJECT_ROOT) if d != "vendor"]
+target_dirs = [os.path.join(PROJECT_ROOT, d) for d in os.listdir(PROJECT_ROOT)
+               if d != "vendor" and os.path.isdir(os.path.join(PROJECT_ROOT, d))]
 
 
 def main(verbose=False):
     ok = True
 
+    ok &= check(Path(PROJECT_ROOT).glob("*.go"), license_header("//"), verbose)
+    ok &= check(Path(PROJECT_ROOT).glob("*.py"), license_header("#"), verbose)
+    ok &= check(Path(PROJECT_ROOT).glob("*.sh"), license_header("#"), verbose)
+
     for d in target_dirs:
         ok &= check(Path(d).glob("**/*_k8s.go"), license_header("//", modification=True), verbose)
         ok &= check([p for p in Path(d).glob("**/*.go")