Display Directory

Author: saurabhchris1

Display_Sirectory.py

@@ -0,0 +1,33 @@
+class Solution:
+    def displayDir(self, dirName):
+        res = []
+        arr = self.getNodes(dirName)
+
+        for element in arr:
+            if "lines" in element.keys():
+                out = [element["name"]] + [" | "] + ["file"] + [" | "] + [str(element["lines"])]
+
+            else:
+                out = [element["name"]] + [" | "] + ["dir"] + [" | "] + [str(self.getLines(element["name"]))]
+            res.append("".join(out))
+        return res
+
+    def getLines(self, dirName):
+        arr = self.getNodes(dirName)
+        numLines = 0
+
+        for element in arr:
+            if "lines" in element.keys():
+                numLines += element["lines"]
+            else:
+                numLines += self.getLines(element["name"])
+
+        return numLines
+
+    def getNodes(self, strs):
+
+        dict = {"home": [{"name": "src"}, {"name": "abc"}, {"name": "test.html", "lines": 10}],
+                "src": [{"name": "abc"}, {"name": "A.js", "lines": 4}],
+                "abc": [{"name": "abc.js", "lines": 5}]
+                }
+        return dict[strs]