pocketpy/scripts/loc.py
blueloveTH f3ac21ccc2 ...
2023-07-02 01:39:24 +08:00

27 lines
671 B
Python

import os
def get_loc(path):
loc = 0
with open(path, "rt", encoding='utf-8') as f:
loc += len(f.readlines())
return loc
def get_loc_for_dir(path):
loc = 0
loc_ex = 0
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith('.h') or file.endswith('.cpp'):
_i = get_loc(os.path.join(root, file))
print(f"{file}: {_i}")
if file.startswith('_'):
loc_ex += _i
else:
loc += _i
return f'{path}: {loc} (+{loc_ex})'
print(get_loc_for_dir('include/pocketpy'))
print()
print(get_loc_for_dir('src'))