65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
import pathlib
|
|
|
|
|
|
def sizeof_fmt(num: int, suffix='B') -> str:
|
|
"""Returns a human-readable size.
|
|
|
|
Source: http://stackoverflow.com/a/1094933/875379
|
|
"""
|
|
|
|
for unit in ['', 'Ki', 'Mi', 'Gi', 'Ti', 'Pi', 'Ei', 'Zi']:
|
|
if abs(num) < 1024:
|
|
return '%.1f %s%s' % (num, unit, suffix)
|
|
num /= 1024
|
|
|
|
return '%.1f Yi%s' % (num, suffix)
|
|
|
|
|
|
def find_in_path(path: pathlib.Path, filename: str) -> pathlib.Path:
|
|
"""Performs a breadth-first search for the filename.
|
|
|
|
Returns the path that contains the file, or None if not found.
|
|
"""
|
|
|
|
import collections
|
|
|
|
# Be lenient on our input type.
|
|
if isinstance(path, str):
|
|
path = pathlib.Path(path)
|
|
|
|
if not path.exists():
|
|
return None
|
|
assert path.is_dir()
|
|
|
|
to_visit = collections.deque([path])
|
|
while to_visit:
|
|
this_path = to_visit.popleft()
|
|
|
|
for subpath in this_path.iterdir():
|
|
if subpath.is_dir():
|
|
to_visit.append(subpath)
|
|
continue
|
|
|
|
if subpath.name == filename:
|
|
return subpath
|
|
|
|
return None
|