mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-20 00:20:47 +00:00
82 lines
2.5 KiB
Python
Executable File
82 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
#
|
||
# Copyright (C) 2024-present ScyllaDB
|
||
#
|
||
#
|
||
# SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.1
|
||
#
|
||
|
||
import argparse
|
||
import sys
|
||
from pathlib import Path
|
||
from typing import Set
|
||
|
||
|
||
def parse_args() -> argparse.Namespace:
|
||
"""Parses command-line arguments."""
|
||
parser = argparse.ArgumentParser(description='Check license headers in files')
|
||
parser.add_argument('--files', required=True, nargs="+", type=Path,
|
||
help='List of files to check')
|
||
parser.add_argument('--license', required=True,
|
||
help='License to check for')
|
||
parser.add_argument('--check-lines', type=int, default=10,
|
||
help='Number of lines to check (default: %(default)s)')
|
||
parser.add_argument('--extensions', required=True, nargs="+",
|
||
help='List of file extensions to check')
|
||
parser.add_argument('--verbose', action='store_true',
|
||
help='Print verbose output (default: %(default)s)')
|
||
return parser.parse_args()
|
||
|
||
|
||
def should_check_file(file_path: Path, allowed_extensions: Set[str]) -> bool:
|
||
return file_path.suffix in allowed_extensions
|
||
|
||
|
||
def check_license_header(file_path: Path, license_header: str, check_lines: int) -> bool:
|
||
try:
|
||
with open(file_path, 'r', encoding='utf-8') as f:
|
||
for _ in range(check_lines):
|
||
line = f.readline()
|
||
if license_header in line:
|
||
return True
|
||
return False
|
||
except (UnicodeDecodeError, StopIteration):
|
||
# Handle files that can't be read as text or have fewer lines
|
||
return False
|
||
|
||
|
||
def main() -> int:
|
||
args = parse_args()
|
||
|
||
if not args.files:
|
||
print("No files to check")
|
||
return 0
|
||
|
||
num_errors = 0
|
||
|
||
for file_path in args.files:
|
||
# Skip non-existent files
|
||
if not file_path.exists():
|
||
continue
|
||
|
||
# Skip files with non-matching extensions
|
||
if not should_check_file(file_path, args.extensions):
|
||
print(f"ℹ️ Skipping file with unchecked extension: {file_path}")
|
||
continue
|
||
|
||
# Check license header
|
||
if check_license_header(file_path, args.license, args.check_lines):
|
||
if args.verbose:
|
||
print(f"✅ License header found in: {file_path}")
|
||
else:
|
||
print(f"❌ Missing license header in: {file_path}")
|
||
num_errors += 1
|
||
|
||
if num_errors > 0:
|
||
sys.exit(1)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|