updates

Author: ehanson8

README.md

@@ -13,6 +13,9 @@ Based on the file path entered and file extension specified, extracts the names
 #### [extractFolderNames.py](/extractFilenames.py)
 Based on the file path entered, extracts the names of all nested folders and writes them to a CSV file.
 
+#### [generateChecksumsForFilesInDirectory.py](/generateChecksumsForFilesInDirectory.py)
+Based on the file path entered, creates a CSV of all files in that directory and the files' MD5 checksums.
+
 #### [renameDirectories.py](/renameDirectories.py)
 Based on the file path entered, renames all nested folders according to a CSV file named 'FolderNames.csv' with the columns named 'oldFolder' and 'newFolder,' provided that there is a match between the nested folders and the names in the 'oldFolder' column. The script writes a log of all folders that were renamed, including their old name and their new name.
 

generateChecksumsForFilesInDirectory.py

@@ -0,0 +1,28 @@
+import hashlib
+import csv
+import argparse
+import os
+
+parser = argparse.ArgumentParser()
+parser.add_argument('-d', '--directory', help='the directory of the files. optional - if not provided, the script will ask for input')
+args = parser.parse_args()
+
+if args.directory:
+    directory = args.directory
+else:
+    directory = raw_input('Enter directory (C:/Test/): ')
+
+fileList = []
+for root, dirs, files in os.walk(directory, topdown=True):
+    for file in files:
+        fileList.append(os.path.join(root, file).replace('\\','/'))
+
+f=csv.writer(open('fileChecksums.csv', 'wb'))
+f.writerow(['fileName']+['checksum'])
+
+for file in fileList:
+    fileName = file[file.rindex('/')+1:]
+    print fileName
+    file = open(file,'rb').read()
+    checksum = hashlib.md5(file).hexdigest()
+    f.writerow([fileName]+[checksum])

renameFiles.py

@@ -6,18 +6,18 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument('-d', '--directory', help='the directory of the files to be renamed. optional - if not provided, the script will ask for input')
-parser.add_argument('-f', '--fileName', help='the CSV file of name changes. optional - if not provided, the script will ask for input')
+parser.add_argument('-f', '--fileNameCSV', help='the CSV file of name changes. optional - if not provided, the script will ask for input')
 parser.add_argument('-m', '--makeChanges', help='Enter "true" to if the script should actually rename the files (otherwise, it will only create a log of the expected file name changes). optional - if not provided, the script will to "false"')
 args = parser.parse_args()
 
 if args.directory:
     directory = args.directory
 else:
     directory = raw_input('Enter the directory of the files to be renamed: ')
-if args.fileName:
-    fileName = args.fileName
+if args.fileNameCSV:
+    fileNameCSV = args.fileNameCSV
 else:
-    fileName = raw_input('Enter the CSV file of name changes (including \'.csv\'): ')
+    fileNameCSV = raw_input('Enter the CSV file of name changes (including \'.csv\'): ')
 if args.makeChanges:
     makeChanges = args.makeChanges
 else:
@@ -28,7 +28,7 @@
 f.writerow(['oldFilename']+['newFilename'])
 for root, dirs, files in os.walk(directory, topdown=True):
     for file in files:
-        with open(nameChangeFile) as csvfile:
+        with open(fileNameCSV) as csvfile:
             reader = csv.DictReader(csvfile)
             for row in reader:
                 oldFilename = row['file']
@@ -38,7 +38,7 @@
                     oldPath = os.path.join(root,file)
                     newPath = os.path.join(root,newFilename)
                     f.writerow([oldPath]+[newPath])
-                    if makeChanges == 'true'
+                    if makeChanges == 'true':
                         os.rename(oldPath,newPath)
                     else:
                         print 'log of expected file name changes created only, no files renamed'