I have this script that makes manipulations on a directory, it maps the directory and sub-directories and gives me a list that every item in the list is [path, md5].
when I do it to one directory it works fine, but when i run it on two directories, the seconds list contains all the first list data.
this is the source code:
import os
import hashlib
import platform
import sys
import argparse
import HTML
class Map(object):
#init function
def __init__(self,param,out_path):
self.param_list = param
self.slash = self.slash_by_os()
self.result_list = []
self.os = ""
self.index = 0
self.html_out_path = out_path
def export_to_HTML(self):
html_code = HTML.table(self.result_list,header_row=self.param_list)
f = open(self.html_out_path,'w')
f.write(html_code + "<p>\n")
f.close()
def calc_md5(self,file_path):
hash = hashlib.md5()
with open(file_path, 'rb') as file_to_check:
for chunk in iter(lambda: file_to_check.read(4096), ''):
hash.update(chunk)
return hash.hexdigest()
def slash_by_os(self):
general_id = platform.system()
actual_os = ""
if general_id == "Darwin" or general_id == "darwin":
actual_os = "UNIX"
elif general_id == "Linux" or general_id == "linux":
actual_os = "UNIX"
elif general_id == "SunOS":
actual_os = "UNIX"
elif general_id == "Windows" or general_id == "windows":
actual_os = "WIN"
else:
actual_os = general_id
if actual_os == "UNIX":
return '/'
elif actual_os == "WIN":
return '\\'
else:
return '/'
self.os = actual_os
def what_to_do(self,new_dir):
act = []
act.append(new_dir[:-1])
for param in self.param_list:
if param == "md5":
x = self.calc_md5(new_dir[:-1])
act.append(x)
return act
def list_of_files(self ,dir_name ,traversed = [], results = []):
dirs = os.listdir(dir_name)
if dirs:
for f in dirs:
new_dir = dir_name + f + self.slash
if os.path.isdir(new_dir) and new_dir not in traversed:
traversed.append(new_dir)
self.list_of_files(new_dir, traversed, results)
else:
try:
act = self.what_to_do(new_dir)
results.append(act)
except Exception as e :
print "%s excepted %s, couldent read" % (new_dir,e)
self.result_list = results
return results
def parse_args():
desc = "DirMap 1.0"
parser = argparse.ArgumentParser(description=desc)
parser.add_argument('-p1','--ogpath', help='Path To Original Directory', required=True)
parser.add_argument('-p2','--modpath', help='Path To Modified Directory', required=True)
args = vars(parser.parse_args())
params = ['path','md5']
return args,params
def main():
args , params = parse_args()
og_dir_path = args['ogpath']
og_map = Map(params,"og.html")
og_list = og_map.list_of_files(og_dir_path)
og_map.export_to_HTML()
mod_dir_path =args['modpath']
mod_map = Map(params,"mod.html")
mod_list = mod_map.list_of_files(mod_dir_path)
mod_map.export_to_HTML()
main()
Any Help ???