【趋势科技实习录】整理文件的脚本

本文链接：https://blog.youkuaiyun.com/u010951938/article/details/45076117

本文提供了一个脚本，用于在指定目录中搜索特定文件，并将找到的文件路径追加到日志文件中。脚本支持32位和64位系统，能够高效地收集和整理文件路径。

自己写的脚本：

import os
import shutil
import sys
def SearchDirFile(path, src):
    for root , dirs, files in os.walk(path, True):
        if - 1 != root.find(src):  
            print root  
        for item in files:
             path = os.path.join(root, item)
             if - 1 != path.find(src):         
                yield path

if __name__ == '__main__':
    if len(sys.argv)<2:
        print 'Use: %s ' % sys.argv[0]
        sys.exit(1)
    if sys.argv[1]=='32':
        matchs = list(SearchDirFile("/Users/mac/DCERAT","TotoalResult32"))
        if os.path.exists("/Users/mac/DCERAT/BoundaryTotalResult32.log"):
            os.remove("/Users/mac/DCERAT/BoundaryTotalResult32.log")
        for match in matchs:
            inputFile = open(match, "r")
            outputFile = open("/Users/mac/DCERAT/BoundaryTotalResult32.log", "a");
            outputFile.write(match);
            outputFile.write('\n');
            allLines =  inputFile.readlines();
            for eachLine in allLines:
                outputFile.write(eachLine);
            outputFile.write('\n\n\n');
            inputFile.close();
        outputFile.close();
    else:
        matchs = list(SearchDirFile("/Users/mac/DCERAT","TotoalResult64"))
        if os.path.exists("/Users/mac/DCERAT/BoundaryTotalResult64.log"):
            os.remove("/Users/mac/DCERAT/BoundaryTotalResult64.log")
        for match in matchs:
            inputFile = open(match, "r")
            outputFile = open("/Users/mac/DCERAT/BoundaryTotalResult64.log", "a");
            outputFile.write(match);
            outputFile.write('\n');
            allLines =  inputFile.readlines();
            for eachLine in allLines:
                outputFile.write(eachLine);
            outputFile.write('\n\n\n');
            inputFile.close();
        outputFile.close();

windows效果图：