matplotlib 이용해 파일 사용 흔적 확인

프로그램의 실행흔적을 확인하고 싶을 때 여러가지 방법이 있겠지만

참고 : http://moaimoai.tistory.com/136

그 프로그램이 생성하는 파일로 해당 프로그램의 사용흔적을 추정할 수 있다.

이는 파이썬을 통해 쉽게 추적이 가능하며 본 프로그램에서는 추적뿐만 아니라 그래프로 쉽게 표현해주는 기능을 추가하였다.

지금은 파일 확장자로 파일을 탐색하지만, 더 정확하게 파일이 가지는 고유 시그니처를 통해 파일의 확장자를 구별할 수 있다.

아래 코드에서 확장자를 더 추가하고 싶으면 

extList = [".hwp", ".doc", ".pdf", ".txt"] <--이부분을 수정하면 된다.

#FileTypeGraph.py from FileTree import FileTree import csv import matplotlib.pyplot as plt from matplotlib.dates import YearLocator, MonthLocator, DayLocator, DateFormatter import numpy as np import datetime import sys, os if __name__=="__main__": if len(sys.argv) != 2: print "Usage : FileTimeGraph [Input File Path]" sys.exit() if not os.path.isdir(sys.argv[1]) : print "[Error] can't open the file" sys.exit() #############setting############# #color : #b: blue #g: green #r: red #c: cyan #m: magenta #y: yellow #k: black #w: white #style : s, o + colorList = ['r', 'b', 'g', 'c', 'm', 'y', 'k'] extList = [".hwp", ".doc", ".pdf", ".txt"] extList.sort() extDict = {} for i in range(len(extList)) : extDict[extList[i]] = (i, 'o', colorList[i%len(colorList)]) #############File Search############# filename = "result" fileTree = FileTree(sys.argv[1], extDict , filename) fileTree.startTree() fileTree = None #############ploting############# def plot_kind( k, m, c ): csvName = filename + "_" + k.replace(".","") + ".csv" csvfile = open( csvName, 'rb' ) rowList = csv.reader(csvfile, delimiter=',', quotechar='\n') for row in rowList: ax.scatter(row[0], row[1], marker=m, color=c, label=k ) fig, ax = plt.subplots() params = { 'axes.labelsize': 15, 'text.fontsize': 10, 'legend.fontsize': 15, 'xtick.labelsize': 12, 'ytick.labelsize': 12, } plt.xlabel("Time Stamp", fontsize = 15) plt.ylabel("File Extension", fontsize = 15) plt.xticks(rotation=90) years = YearLocator() months = MonthLocator() days = DayLocator() yearsFmt = DateFormatter('%Y-%m-%d') plt.yticks(np.arange(len(extDict.keys())), sorted(extDict.keys())) for ext, status in extDict.items() : plot_kind( ext, status[1], status[2] ) # format the ticks ax.xaxis.set_major_locator(months) ax.xaxis.set_major_formatter(yearsFmt) ax.xaxis.set_minor_locator(days) ax.autoscale_view() ax.fmt_xdata = yearsFmt ax.grid(True) plt.rcParams.update(params) plt.savefig('result.png') plt.show()

지금은 가장 최근 수정한 시간을 참고하고 있다. 따라서 기준이 되는 시간을 변경하기 위해선

self.csvWriter[ext].writerow({'mTime': fileInfo.getMtime(), 'ext': self.extDict[ext][0] }) 

위 코드에서 fileInfo.getMtime()이 호출하는 함수를 수정하면 된다.

getAtime() : 접근 시간

getMtime() : 수정 시간

getCtime() : 생성 시간

#FileTree.py import sys import os import csv from FileInformation import FileInformation reload(sys) sys.setdefaultencoding('MS949') class FileTree : def __init__(self, rootDir, extDict, result): self.rootDir = u"" +rootDir self.extDict = extDict self.csvWriter = {} self.csvFileList = [] for ext in extDict.keys() : filename = result + "_" + ext.replace(".","") + ".csv" if os.path.isfile(filename) : os.remove(filename) csvfile = open(filename, 'ab') self.csvFileList.append(csvfile) fieldnames = ['mTime', 'ext'] self.csvWriter[ext] = csv.DictWriter(csvfile, fieldnames=fieldnames) #self.csvWriter[ext].writeheader() def __del__(self) : for csvfile in self.csvFileList : csvfile.close() def startTree(self): for dirName, subdirList, fileList in os.walk(self.rootDir): for fname in fileList: fileInfo = FileInformation(dirName+ "\\" + fname) for ext in self.extDict.keys() : if fileInfo.checkExt(ext) : self.csvWriter[ext].writerow({'mTime': fileInfo.getMtime(), 'ext': self.extDict[ext][0] }) print "complete"

마지막으로 아래는 파일의 정보를 가져오는 코드이다.

#FileInformation.py import os.path, time import matplotlib.dates as md import matplotlib.dates as dates class FileInformation : def __init__(self, filePath) : self.filePath = filePath; self.setFileTime() def setFileTime(self) : self.ctime = dates.datestr2num(time.strftime('%d/%m/%Y', time.gmtime(os.path.getctime(self.filePath)))) self.mtime = dates.datestr2num(time.strftime('%d/%m/%Y', time.gmtime(os.path.getmtime(self.filePath)))) self.atime = dates.datestr2num(time.strftime('%d/%m/%Y', time.gmtime(os.path.getatime(self.filePath)))) self.ext = os.path.splitext(self.filePath)[1] def checkExt(self, ext): if self.ext == ext : return True else : return False def getCtime(self): return self.ctime def getMtime(self): return self.mtime def getAtime(self): return self.atime

위의 3개 코드를 이용해 프로그램을 실행한다.

실행 예제는 다음과 같다.

FileTimeGraph.py "C:\"

참고로 윈도우에서는 matplotlib의 설치가 어려우니 파이썬 대신 아나콘다를 설치해서 쓰기를 추천한다.