# ##### 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 def pyside_cache(propname): """Decorator, stores the result of the decorated callable in Python-managed memory. This is to work around the warning at https://www.blender.org/api/blender_python_api_master/bpy.props.html#bpy.props.EnumProperty """ if callable(propname): raise TypeError('Usage: pyside_cache("property_name")') def decorator(wrapped): """Stores the result of the callable in Python-managed memory. This is to work around the warning at https://www.blender.org/api/blender_python_api_master/bpy.props.html#bpy.props.EnumProperty """ import functools @functools.wraps(wrapped) # We can't use (*args, **kwargs), because EnumProperty explicitly checks # for the number of fixed positional arguments. def wrapper(self, context): result = None try: result = wrapped(self, context) return result finally: rna_type, rna_info = getattr(self.bl_rna, propname) rna_info['_cached_result'] = result return wrapper return decorator def redraw(self, context): context.area.tag_redraw()