Как получить Python для изящного форматирования Нет и несуществующие поля

Если я пишу в Python:

data = {'n': 3, 'k': 3.141594, 'p': {'a': 7, 'b': 8}}
print('{n}, {k:.2f}, {p[a]}, {p[b]}'.format(**data))
del data['k']
data['p']['b'] = None
print('{n}, {k:.2f}, {p[a]}, {p[b]}'.format(**data))

Я получаю:

3, 3.14, 7, 8
Traceback (most recent call last):
  File "./funky.py", line 186, in <module>
    print('{n}, {k:.2f}, {p[a]}, {p[b]}'.format(**data))
KeyError: 'k'

Вместо сообщения об ошибке как я могу заставить Python более грациозно отформатировать None и несуществующие поля?

Чтобы привести пример, я хотел бы видеть в выводе нечто большее:

3, 3.14, 7, 8
3, ~, 7, ~

В идеале, конечно, я хотел бы иметь возможность указать строку, используемую вместо этих отсутствующих значений.

Ответ 1

Рекомендация в PEP 3101 заключается в подклассе Formatter:

import string
class PartialFormatter(string.Formatter):
    def __init__(self, missing='~~', bad_fmt='!!'):
        self.missing, self.bad_fmt=missing, bad_fmt

    def get_field(self, field_name, args, kwargs):
        # Handle a key not found
        try:
            val=super(PartialFormatter, self).get_field(field_name, args, kwargs)
            # Python 3, 'super().get_field(field_name, args, kwargs)' works
        except (KeyError, AttributeError):
            val=None,field_name 
        return val 

    def format_field(self, value, spec):
        # handle an invalid format
        if value==None: return self.missing
        try:
            return super(PartialFormatter, self).format_field(value, spec)
        except ValueError:
            if self.bad_fmt is not None: return self.bad_fmt   
            else: raise

fmt=PartialFormatter()
data = {'n': 3, 'k': 3.141594, 'p': {'a': '7', 'b': 8}}
print(fmt.format('{n}, {k:.2f}, {p[a]}, {p[b]}', **data))
# 3, 3.14, 7, 8
del data['k']
data['p']['b'] = None
print(fmt.format('{n}, {k:.2f}, {p[a]:.2f}, {p[b]}', **data))
# 3, ~~, !!, ~~

Как установлено, он будет печатать ~~, если поле или атрибут не найдено, и !!, если используется недопустимый формат, учитывая значение поля. (Просто используйте None для аргумента ключевого слова bad_fmt, если вы хотите, чтобы значение ошибки значения повышалось.)

Чтобы обрабатывать недостающие ключи, вам нужно подклассифицировать как get_field, чтобы поймать KeyError или AttributeError и format_field, чтобы вернуть значение по умолчанию для отсутствующего ключа.

Поскольку вы ловите ошибки format_field, вы также можете поймать поле плохого формата, перехватив ValueError из суперкласса.

Ответ 2

Метод str.format() не дает вам прямого метода обработки отсутствующих ключей или замены значений.

Вы можете добавить слой косвенности; передать в отображении, которое обрабатывает отсутствующие значения и значения None, и изменить формат для использования только этого аргумента:

class PlaceholderFormatValue():
    def __format__(self, spec):
        return '~'
    def __getitem__(self, name):
        # handle further nested item access
        return self

class formatting_dict(dict):
    def __getitem__(self, name):
        value = self.get(name)
        if isinstance(value, dict):
            # rewrap nested dictionaries to handle missing nested keys
            value = type(self)(value)
        return value if value is not None else PlaceholderFormatValue()

print('{0[n]}, {0[k]:.2f}, {0[p][a]}, {0[p][b]}'.format(formatting_dict(data)))

Теперь все слоты ссылаются на позиционный аргумент 0, который обрабатывается как словарь, но ключевые поиски всегда успешны, и обе пропущенные значения и None заменяются значением-заполнителем.

Здесь PlaceholderFormatValue() гарантирует, что независимо от того, что дает спецификация формата, значение может быть интерполировано в формате. Это делает работу {0[k]:.2f}, например.

Обернув любые значения dict и имея доступ к элементу управления PlaceholderFormatValue, вышесказанное может также обрабатывать отказ для обеспечения вложенных ключей или целых словарей:

>>> data = {'n': 3, 'k': 3.141594, 'p': {'a': 7, 'b': 8}}
>>> del data['k']
>>> data['p']['b'] = None
>>> print('{0[n]}, {0[k]:.2f}, {0[p][a]}, {0[p][b]}'.format(formatting_dict(data)))
3, ~, 7, ~
>>> del data['p']['a']
>>> print('{0[n]}, {0[k]:.2f}, {0[p][a]}, {0[p][b]}'.format(formatting_dict(data)))
3, ~, ~, ~
>>> del data['p']
>>> print('{0[n]}, {0[k]:.2f}, {0[p][a]}, {0[p][b]}'.format(formatting_dict(data)))
3, ~, ~, ~

Ответ 3

Если вы можете сделать форматирование отдельно, вы можете использовать Template.safe_substitute, который изящно обрабатывает отсутствующие значения:

>>> from string import Template
>>> t = Template("$a $b $c")
>>> t.safe_substitute(a=3)
'3 $b $c'