45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from pathlib import Path
|
||
from typing import Dict
|
||
|
||
|
||
def count_python_lines(project_path: str = ".", exclude: str = ".venv") -> Dict[str, int]:
|
||
"""
|
||
Подсчитывает количество строк кода во всех Python (.py) файлах проекта.
|
||
|
||
Args:
|
||
project_path (str): Путь к директории проекта. По умолчанию — текущая папка.
|
||
exclude (str): Имя директории, которую нужно исключить (например, .venv).
|
||
|
||
Returns:
|
||
Dict[str, int]: Словарь с двумя ключами:
|
||
- "files": количество найденных файлов .py
|
||
- "lines": общее количество строк во всех .py файлах
|
||
"""
|
||
root: Path = Path(project_path)
|
||
total_lines: int = 0
|
||
file_count: int = 0
|
||
|
||
# Рекурсивный обход всех файлов
|
||
for file_path in root.rglob("*.py"):
|
||
# Игнорируем .venv и любые подпапки внутри него
|
||
if exclude in file_path.parts:
|
||
continue
|
||
|
||
try:
|
||
with file_path.open("r", encoding="utf-8") as f:
|
||
# Считаем количество строк в файле
|
||
line_count = sum(1 for _ in f)
|
||
total_lines += line_count
|
||
file_count += 1
|
||
except (UnicodeDecodeError, PermissionError):
|
||
# Иногда могут встретиться битые файлы или без прав доступа
|
||
continue
|
||
|
||
return {"files": file_count, "lines": total_lines}
|
||
|
||
|
||
if __name__ == "__main__":
|
||
stats = count_python_lines()
|
||
print(f"📊 Найдено файлов: {stats['files']}")
|
||
print(f"📄 Всего строк кода: {stats['lines']}")
|