mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 19:40:18 +00:00
23 lines
585 B
Python
23 lines
585 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'):
|
|
_i = get_loc(os.path.join(root, file))
|
|
print(f"{file}: {_i}")
|
|
if file.startswith('_'):
|
|
loc_ex += _i
|
|
else:
|
|
loc += _i
|
|
return f'{loc} (+{loc_ex})'
|
|
|
|
print(get_loc_for_dir('src')) |