insert_wordpress.py

@@ -15,8 +15,7 @@ class WPimport:
 
     def fromDirectory(self, directory):
         directory = "{0}/archives".format(directory)
-        for item in os.listdir(directory):
-            subdirectory = self._checkDirectory(item)
+        subdirectory = self._checkDirectory([], "{0}".format(directory))
         print(subdirectory)
 
 
@@ -34,10 +33,13 @@ class WPimport:
     
     # Private method
 
-    def _checkDirectory(self, item):
-        if os.path.isdir(item):
-            self._checkDirectory(item)
-        return item
+    def _checkDirectory(self, subdirectory, item):
+        sub = subdirectory
+        for i in os.listdir(item):
+            if os.path.isdir("{0}/{1}".format(item, i)):
+                sub.append("{0}/{1}".format(item, i))
+                subdirectory = self._checkDirectory(sub, "{0}/{1}".format(item, i))
+        return subdirectory
 
     ## Add or update featured media