88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""Count lines of Python code (excluding comments and blank lines)."""
|
|
|
|
import sys
|
|
import ast
|
|
from pathlib import Path
|
|
|
|
|
|
def count_python_loc(filepath: str) -> dict:
|
|
"""
|
|
Count logical lines of Python code.
|
|
|
|
Returns:
|
|
dict with total_lines, code_lines, comment_lines, blank_lines
|
|
"""
|
|
path = Path(filepath)
|
|
|
|
if not path.exists():
|
|
return {"error": f"File not found: {filepath}", "valid": False}
|
|
|
|
try:
|
|
with open(filepath, 'r') as f:
|
|
lines = f.readlines()
|
|
except Exception as e:
|
|
return {"error": str(e), "valid": False}
|
|
|
|
total = len(lines)
|
|
code_lines = 0
|
|
comment_lines = 0
|
|
blank_lines = 0
|
|
|
|
in_docstring = False
|
|
docstring_delim = None
|
|
|
|
for line in lines:
|
|
stripped = line.strip()
|
|
|
|
# Check for docstring start/end
|
|
if not in_docstring and (stripped.startswith('"""') or stripped.startswith("'''")):
|
|
if stripped.count('"""') >= 2 or stripped.count("'''") >= 2:
|
|
# Single line docstring
|
|
comment_lines += 1
|
|
else:
|
|
in_docstring = True
|
|
docstring_delim = '"""' if '"""' in stripped else "'''"
|
|
continue
|
|
elif in_docstring:
|
|
if docstring_delim in stripped:
|
|
in_docstring = False
|
|
comment_lines += 1
|
|
continue
|
|
|
|
if not stripped:
|
|
blank_lines += 1
|
|
elif stripped.startswith('#'):
|
|
comment_lines += 1
|
|
else:
|
|
code_lines += 1
|
|
|
|
return {
|
|
"valid": True,
|
|
"file": filepath,
|
|
"total_lines": total,
|
|
"code_lines": code_lines,
|
|
"comment_lines": comment_lines,
|
|
"blank_lines": blank_lines,
|
|
"sum": code_lines + comment_lines + blank_lines
|
|
}
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print("Usage: py_loc.py <file1.py> [file2.py...]", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
files = sys.argv[1:]
|
|
|
|
for filepath in files:
|
|
result = count_python_loc(filepath)
|
|
if "valid" in result and result["valid"]:
|
|
print(f"{filepath}: {result['code_lines']} code, {result['comment_lines']} comments, {result['blank_lines']} blank, {result['total_lines']} total")
|
|
else:
|
|
print(f"ERROR: {filepath} - {result.get('error', 'unknown')}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |