diff options
Diffstat (limited to 'lib/find_stats.py')
-rw-r--r-- | lib/find_stats.py | 59 |
1 files changed, 0 insertions, 59 deletions
diff --git a/lib/find_stats.py b/lib/find_stats.py deleted file mode 100644 index effb5b7..0000000 --- a/lib/find_stats.py +++ /dev/null @@ -1,59 +0,0 @@ -import os -from pathlib import Path - -from .stats import Statistics - -IGNORED_FOLDERS = [ - 'venv', -] - - -def find_all_files(folder: Path, verbose: bool = False): - def gen(): - for subpath, subfolders, filenames in os.walk(str(folder)): - subfolders[:] = [subfolder for subfolder in subfolders if - not (subfolder.startswith('.') or subfolder.lower() in IGNORED_FOLDERS)] - current_path = Path(subpath) - for file in filenames: - file_path = current_path / file - if not file.startswith('.'): - yield file_path - if verbose: - print('Scanning', file_path) - else: - if verbose: - print('Skipping', file_path) - - return list(gen()) - - -def find_stats_for_file(filename: Path, name: str, stats: Statistics): - extension = filename.name.split('.')[-1] - try: - for line in filename.read_text().split('\n'): - spaces = False - tabs = False - while len(line) > 0 and line[0] in [' ', '\t']: - if line[0] == ' ': - spaces = True - if line[0] == '\t': - tabs = True - line = line[1:] - if spaces and tabs: - stats.add_mixed_line(extension=extension, - filename=name) - elif spaces: - stats.add_spaces(extension=extension) - elif tabs: - stats.add_tabs(extension=extension) - except (UnicodeDecodeError, OSError): - pass - - -def find_stats(folder: Path, verbose: bool = False) -> Statistics: - folder = folder.resolve() - files = find_all_files(folder, verbose=verbose) - stats = Statistics() - for file in files: - find_stats_for_file(folder / file, file, stats) - return stats |