Build a directory tree, including files, for a given path

dirtree.py

import os

def dirtree(path):
  steps = list(os.walk(path))
  def _tree():
      _, child_dirs, files = steps.pop(0)
      obj = { child: _tree() for child in child_dirs }
      obj['__files__'] = files
      return obj
  return _tree()

# Output Example:
# {
#   "A": {
#     "C": {
#       "D": {
#         "__files__": [
#           "d.txt",
#           "d1.txt"
#         ]
#       },
#       "__files__": [
#         "c.txt"
#       ]
#     },
#     "__files__": [
#       "a.txt",
#       "a2.txt",
#       "a3.txt"
#     ]
#   },
#   "__files__": []
# }