-
Notifications
You must be signed in to change notification settings - Fork 25
/
count.py
42 lines (39 loc) · 1.22 KB
/
count.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#coding=utf-8
import re
import os
def get_file_path(root_path, file_list, dir_list):
# 获取该目录下所有的文件名称和目录名称
dir_or_files = os.listdir(root_path)
for dir_file in dir_or_files:
# 获取目录或者文件的路径
dir_file_path = os.path.join(root_path, dir_file)
# 判断该路径为文件还是路径
if os.path.isdir(dir_file_path):
dir_list.append(dir_file_path)
# 递归获取所有文件和目录的路径
get_file_path(dir_file_path, file_list, dir_list)
else:
file_list.append(dir_file_path)
if __name__ == "__main__":
# 根目录路径
root_path = "."
# 用来存放所有的文件路径
file_list = []
# 用来存放所有的目录路径
dir_list = []
get_file_path(root_path, file_list, dir_list)
#print(file_list)
#print(dir_list)
sollist = list()
for i in file_list:
if i.endswith(".sol"):
sollist.append(i)
for i in sollist:
print(i)
numline = 0
for file in sollist:
f = open(file, "r")
numline += len(f.readlines())
f.close()
print(numline, len(sollist))
print(numline / len(sollist))