Extras.install.py 1.61 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
"""Recursively copy a directory but skip undesired files and
directories (CVS, backup files, pyc files, etc)"""

import sys
import os
import shutil

verbose = 1
debug = 0

def isclean(name):
12 13 14
    if name == 'CVS': return 0
    if name == '.cvsignore': return 0
    if name == '.DS_store': return 0
15
    if name == '.svn': return 0
16 17 18 19 20 21 22
    if name.endswith('~'): return 0
    if name.endswith('.BAK'): return 0
    if name.endswith('.pyc'): return 0
    if name.endswith('.pyo'): return 0
    if name.endswith('.orig'): return 0
    return 1

23
def copycleandir(src, dst):
24 25 26 27
    for cursrc, dirs, files in os.walk(src):
        assert cursrc.startswith(src)
        curdst = dst + cursrc[len(src):]
        if verbose:
28
            print("mkdir", curdst)
29 30 31 32 33 34
        if not debug:
            if not os.path.exists(curdst):
                os.makedirs(curdst)
        for fn in files:
            if isclean(fn):
                if verbose:
35
                    print("copy", os.path.join(cursrc, fn), os.path.join(curdst, fn))
36 37 38 39
                if not debug:
                    shutil.copy2(os.path.join(cursrc, fn), os.path.join(curdst, fn))
            else:
                if verbose:
40
                    print("skipfile", os.path.join(cursrc, fn))
41 42 43
        for i in range(len(dirs)-1, -1, -1):
            if not isclean(dirs[i]):
                if verbose:
44
                    print("skipdir", os.path.join(cursrc, dirs[i]))
45 46
                del dirs[i]

47
def main():
48 49 50 51 52
    if len(sys.argv) != 3:
        sys.stderr.write("Usage: %s srcdir dstdir\n" % sys.argv[0])
        sys.exit(1)
    copycleandir(sys.argv[1], sys.argv[2])

53
if __name__ == '__main__':
54
    main()