Add sizes to file entries (fixes #5)

bento-platform · Jan 28, 2020 · 973c7b4 · 973c7b4
1 parent d3ee0c6
commit 973c7b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chord_drop_box_service/app.py b/chord_drop_box_service/app.py
@@ -54,7 +54,9 @@ def recursively_build_directory_tree(directory, level=0):
                   "path": os.path.abspath(os.path.join(directory, entry)),
                   "contents": recursively_build_directory_tree(os.path.join(directory, entry), level=level+1)}
                  if os.path.isdir(os.path.join(directory, entry))
-                 else {"name": entry, "path": os.path.abspath(os.path.join(directory, entry))}
+                 else {"name": entry,
+                       "path": os.path.abspath(os.path.join(directory, entry)),
+                       "size": os.path.getsize(os.path.join(directory, entry))}
                  for entry in os.listdir(directory)
                  if (level < TRAVERSAL_LIMIT or not os.path.isdir(os.path.join(directory, entry))) and entry[0] != ".")