PublicLibrary/ProjectSummary.py at master · GitCodeAllen/PublicLibrary · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
# -*- coding: utf-8 -*-

import os
import sys
import time
import threading

#需要统计行数的文件后缀
Suffix = {".c",".cpp",".h",".hpp",".cxx"}
#目录路径
Path = ""

if len(sys.argv) > 1:
    Path = sys.argv[1]

if not Path:
    print("Path is empty")
    exit()

if not os.path.isdir(Path):
    print("Path is not exists")
    exit()

def CountFileLines(FileName):
    count=0
    fp=open(FileName,"r")
    last_chr='\n'
    while 1:
        buffer=fp.read(8*1024*1024)
        if not buffer:
            break
        count+=buffer.count('\n')
        last_chr=buffer[-1]
    fp.close()
    if last_chr != '\n':
        count+=1
    return count

def listdir(dir,Targets,isSearchFiles = False):
    files = os.listdir(dir)
    for file in files:
        filepath = os.path.join(dir,file)
        if os.path.isdir(filepath):
            listdir(filepath,Targets,isSearchFiles)
        elif os.path:
            suffix = os.path.splitext(file)[1]
            if Targets.has_key(suffix):
                if not Targets[suffix].has_key("TotalFiles"):
                    Targets[suffix]["TotalFiles"] = 0
                Targets[suffix]["TotalFiles"] += 1
                if isSearchFiles:
                    continue
                lines = CountFileLines(filepath)
                if not Targets[suffix].has_key("TotalLines"):
                    Targets[suffix]["TotalLines"] = 0
                Targets[suffix]["TotalLines"] += lines
                if not Targets[suffix].has_key("TheMaxLines"):
                    Targets[suffix]["TheMaxLines"] = 0
                if Targets[suffix]["TheMaxLines"] < lines:
                    Targets[suffix]["TheMaxLines"] = lines

def Count(Targets):
    files = 0
    lines = 0
    for suffix_val in Targets.keys():
        if not Targets[suffix_val]:
            continue
        for suffix_item in Targets[suffix_val].keys():
            if suffix_item == "TotalFiles":
                files += Targets[suffix_val][suffix_item]
            if suffix_item == "TotalLines":
                lines += Targets[suffix_val][suffix_item]
    return (files,lines)

def DynamicShow(ShowStr):
    sys.stdout.write(('{:<20}').format(ShowStr)+'\r')
    sys.stdout.flush()

def EndDynamicShow():
    print('')

def ShowResults(Targets):
    for suffix_val in Targets.keys():
        if not Targets[suffix_val]:
            continue
        print(suffix_val+":")
        for suffix_item in Targets[suffix_val].keys():
            print("\t"+suffix_item+"\t:"+"{:,}".format(Targets[suffix_val][suffix_item]))
        print('\n')

def ShowSummary(Targets):
    result = Count(Targets)
    print("Total:")
    print("\tFiles\t:"+"{:,}".format(result[0]))
    print("\tLines\t:"+"{:,}".format(result[1]))

TargetsCount = {}
Targets = {}
for val in Suffix:
    TargetsCount[val] = {}
    Targets[val] = {}

print('Begin searching files')
listdir(Path,TargetsCount,True)
count_result = Count(TargetsCount)
print('Find ' + str(count_result[0]) + ' files')
listdir(Path,Targets)
print('\nResult:')
ShowResults(Targets)
ShowSummary(Targets)