Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 26 additions & 0 deletions core/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -359,6 +359,32 @@ class _Default:


class SafeFormatter(Formatter):
_FIELD_PATTERN = re.compile(r"(?<!{)\{[^{}]+\}(?!})")
_UNMATCHED_BRACE_ERRORS = {
"Single '{' encountered in format string",
"Single '}' encountered in format string",
"expected '}' before end of string",
"unexpected '{' in field name",
}

def format(self, format_string, /, *args, **kwargs):
try:
return super().format(format_string, *args, **kwargs)
except ValueError as exc:
if str(exc) not in self._UNMATCHED_BRACE_ERRORS:
raise

# A malformed literal brace makes Formatter reject the entire message.
# Format complete fields individually so the remaining text can still
# be sent as written.
def format_field(match):
try:
return Formatter.format(self, match.group(), *args, **kwargs)
except ValueError:
return match.group()

return self._FIELD_PATTERN.sub(format_field, format_string)

def get_field(self, field_name, args, kwargs):
first, rest = _string.formatter_field_name_split(field_name)

Expand Down
Loading