WIP Add extension agnostic functionality

Author: NYPD

phodupe/__main__.py

@@ -7,7 +7,7 @@ def main():
     destination1 = sys.argv[1]
     destination2 = sys.argv[2]
 
-    dupe_files = DupeFinder.getDuplicateFileNames(destination1, destination2)
+    dupe_files = DupeFinder.getDuplicateFileNames(destination1, destination2, False)
 
     if len(dupe_files) is 0:
         print('No dupe files found!')

phodupe/dupe_finder.py

@@ -1,4 +1,5 @@
 import os
+import pathlib
 
 
 class DupeFinder:
@@ -7,7 +8,7 @@ class DupeFinder:
     """
 
     @staticmethod
-    def getDuplicateFileNames(directory1, directory2):
+    def getDuplicateFileNames(directory1, directory2, sameExtension = True):
         """
         Return a list of duplicate file names in both directories
 
@@ -24,14 +25,34 @@ def getDuplicateFileNames(directory1, directory2):
             A list of the duplicate file names between both directories. If no
             duplicate files are found, an empty list is returned
         """
-        directory1Files = os.listdir(directory1)
-        directory2Files = os.listdir(directory2)
+        directory1Path = pathlib.Path(directory1)
+        directory2Path = pathlib.Path(directory2)
+
+        directory1Files = os.listdir(directory1Path)
+        directory2Files = os.listdir(directory2Path)
+
+        directory1FilesNoExt = {}
+        directory2FilesNoExt = {}
+
+        if sameExtension == False:
+            for i, file in enumerate(directory1Files):
+                purePath = pathlib.Path(directory1 + '/' + file)
+                directory1FilesNoExt[purePath.stem] = purePath.suffix
+            for i, file in enumerate(directory2Files):
+                purePath = pathlib.Path(directory2 + '/' + file)
+                directory2FilesNoExt[purePath.stem] = purePath.suffix
 
         dupeFiles = []
 
-        for file in directory1Files:
-            if(file in directory2Files):
-                dupeFiles.append(file)
+        if sameExtension:
+            for file in directory1Files:
+                if(file in directory2Files):
+                    dupeFiles.append(file)
+        else:
+            for fileName, ext in directory1FilesNoExt.items():
+                if fileName in directory2FilesNoExt:
+                    dupeFiles.append(fileName + ext)
+        //We need to delete both extensions
 
         return dupeFiles