py_script/clean_vs.py at master · zhoukaidev/py_script · 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
import os

__author__ = 'KaiZhou'

suffix = ('.txt',
        '.ilk',
        '.pdb',
        '.iobj',
        '.ipdb',
        '.exe',
        '.user',
        '.log',
        '.pch',
        '.tlog',
        '.db',
        '.opendb',
        '.ipch',
        '.suo',
        '.obj',
        '.lastbuildstate',
        '.idb')

def _getDirs(path='.'):
    dirs =[os.path.join(path,x) for x in os.listdir(path)]
    return dirs

def _containExt(ext):
    for i in suffix:
        if i == ext:
            return True
    return False

def _RemoveTemporaryFile(path='.'):
    dirs = _getDirs(path)
    for i in dirs:
        if os.path.isdir(i):
            _RemoveTemporaryFile(i)
        else:
           if _containExt(os.path.splitext(i)[1]):
                os.remove(i)
                print("start remove file:%s" % i)

def _RemoveEmptyDirectory(path='.'):
    dirs = _getDirs(path)
    for i in dirs:
        if os.path.isdir(i):
            if len(os.listdir(i)) == 0:
                os.removedirs(i)
                print('start remove directory:%s'% i)
            else:
                _RemoveEmptyDirectory(i)

def CleanBuildTree(path='.'):
    _RemoveTemporaryFile(path)
    _RemoveEmptyDirectory(path)


if __name__=='__main__':
    CleanBuildTree()