License check fix & cleanup (#10)

Author: traut

cortado/cli.py

@@ -1,19 +1,23 @@
+# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+# or more contributor license agreements. Licensed under the Elastic License
+# 2.0; you may not use this file except in compliance with the Elastic License
+# 2.0.
+
 import logging
 import sys
 
 import structlog
 import typer
 import json
 
-from pathlib import Path
 from collections import Counter
 
 from rich import box
 from rich.table import Table
 from rich.console import Console
 from rich.text import Text
 
-from cortado import mapping, rules
+from cortado import rules
 from cortado.rules import RuleMaturity, RuleRelease
 from cortado.rtas import get_registry, HashRta
 from cortado.utils import configure_logging
@@ -70,21 +74,6 @@ def print_rtas(as_json: bool = False):
     console.print(table)
 
 
-@app.command()
-def generate_mapping(mapping_file: Path = mapping.DEFAULT_MAPPING_FILE):
-    """
-    Generate a mapping file that contains rule to RTA relations.
-    """
-    _log = log.bind(path=str(mapping_file))
-    _log.info("Generating rule-to-RTA mapping file")
-
-    data = mapping.generate_mapping()
-    with open(mapping_file, "w") as f:
-        json.dump(data, f, sort_keys=True, indent=4)
-
-    _log.info("Mapping file saved")
-
-
 @app.command()
 def get_coverage(
     rules_glob: Annotated[str | None, typer.Option(help="A path glob that resolves into rule paths")] = None,

cortado/main.py

@@ -1,3 +1,8 @@
+# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+# or more contributor license agreements. Licensed under the Elastic License
+# 2.0; you may not use this file except in compliance with the Elastic License
+# 2.0.
+
 import glob
 import tomli
 import structlog

cortado/mapping.py

@@ -1,3 +1,8 @@
+# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+# or more contributor license agreements. Licensed under the Elastic License
+# 2.0; you may not use this file except in compliance with the Elastic License
+# 2.0.
+
 import logging
 from typing import Any
 

cortado/rules.py

@@ -1,4 +1,4 @@
-FILE_GLOB=$1
+DIR_PATH=$1
 
 LICENSE_BODY_FILE="scripts/license-header.txt"
 LICENSE_LINES=$(wc -l $LICENSE_BODY_FILE | xargs | cut -f1 -d ' ')
@@ -9,7 +9,11 @@ rc=0
 counter_failed=0
 counter_all=0
 
-for x in $FILE_GLOB; do  
+echo "Checking .py files in $DIR_PATH"
+
+for x in $(find $DIR_PATH -name "*.py"); do  
+
+    # echo "Checking $x"
 
     counter_all=$((counter_all+1))