1effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch# Copyright 2014 The Chromium Authors. All rights reserved. 2effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch# Use of this source code is governed by a BSD-style license that can be 3effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch# found in the LICENSE file. 4effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch 5effb81e5f8246d0db0270817048dc992db66e9fbBen Murdochimport os 6effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch 7effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch 8effb81e5f8246d0db0270817048dc992db66e9fbBen Murdochdef GetRecursiveDiskUsage(path): 9effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch """Returns the disk usage in bytes of |path|. Similar to `du -sb |path|`.""" 10effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch running_size = os.path.getsize(path) 11effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch if os.path.isdir(path): 12effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch for root, dirs, files in os.walk(path): 13effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch running_size += sum([os.path.getsize(os.path.join(root, f)) 14effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch for f in files + dirs]) 15effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch return running_size 16effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch 17