"""Generic utilities."""
from typing import Iterable, List
[docs]def class_name(obj):
return obj.__class__.__name__
[docs]def check_arg(condition: bool, msg: str = ''):
if not condition:
raise ValueError(msg)
[docs]def indent_lines(lines: Iterable[str], width=2) -> List[str]:
spaces = ' ' * width
return [spaces + line for line in lines]
[docs]def make_repr(obj, *args, _line_len: int = 60, _indent: int = 2, **kwargs) -> str:
"""
Generate repr(obj).
:param obj:
object to represent
:param args:
positional arguments in the repr
:param _line_len:
if the repr length exceeds this, arguments will be on their own line;
if negative, the repr will be in a single line regardless of its length
:param _indent:
indentation width of arguments in case they are shown in their own line
:param kwargs:
keyword arguments in the repr
:return: str
"""
cls_name = obj.__class__.__name__
arglist = [
*(repr(arg) for arg in args),
*(f'{key}={value!r}' for key, value in kwargs.items()),
]
len_arglist = sum(len(s) for s in arglist)
total_len = len(cls_name) + len_arglist + 2 * len(arglist)
if 0 <= _line_len < total_len:
lines = indent_lines(arglist, width=_indent)
args_text = ',\n'.join(lines)
return f'{cls_name}(\n{args_text}\n)'
else:
args_text = ', '.join(arglist)
return f'{cls_name}({args_text})'
[docs]def make_one_line_repr(obj, *args, **kwargs):
return make_repr(obj, *args, _line_len=-1, **kwargs)
[docs]def pluralize(
count: int, zero: str = '', one: str = '', many: str = '',
) -> str:
if count == 0 and zero:
return zero
if count == 1 and one:
return one
return many.format(count=count)