py/lumberjack/list_dir_as_yaml.py

#!/usr/bin/python
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""
Directory lister is a ls-like wrapper to return listed files in yaml format.
The output of this tool will be a dictionary where key is file name and value
is the file size.

  ./list_dir.py <directory>
"""
import os
import sys
import yaml

if __name__ == '__main__':
  if len(sys.argv[1:]) != 1:
    raise ValueError('Accept only single argument as directory name')
  dir_path = sys.argv[1]

  files = dict()
  for file_path in os.listdir(dir_path):
    full_path = os.path.join(dir_path, file_path)
    if not os.path.isfile(full_path):
      continue
    files[file_path] = os.path.getsize(full_path)
  print yaml.safe_dump(files, canonical=True)