1j01/textual-paint
1
1
Fork 0
mirror of https://github.com/1j01/textual-paint.git synced 2024-12-22 14:21:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Reorder None case for mypy

Browse Source 
While Pyright can narrow down the type of format_id from `str | None` to `str` based on `format_id in Image.SAVE` (where `Image.SAVE` is `dict[str, Any]`), mypy does not.
This commit is contained in:
Isaiah Odhner 2023-07-15 21:46:00 -04:00
parent 641a46e368
commit 0a6fb29e10
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/textual_paint/paint.py
View File

@ -832,7 +832,9 @@ class AnsiArtDocument:
def encode_to_format(self, format_id: str | None) -> bytes:
"""Encode the image into the given file format."""
# print("Supported image formats for writing:", Image.SAVE.keys())
if format_id == "ANSI":
if format_id is None:
raise FormatWriteNotSupported(localized_message=_("Unknown file extension.") + "\n\n" + _("To save your changes, use a different filename."))
elif format_id == "ANSI":
# This maybe shouldn't use UTF-8... but there's not a singular encoding for "ANSI art".
return self.get_ansi().encode("utf-8")
elif format_id == "IRC":
@ -848,8 +850,6 @@ class AnsiArtDocument:
return self.get_rich_console_markup().encode("utf-8")
elif format_id in Image.SAVE and format_id not in SAVE_DISABLED_FORMATS:
return self.encode_image_format(format_id)
elif format_id is None:
raise FormatWriteNotSupported(localized_message=_("Unknown file extension.") + "\n\n" + _("To save your changes, use a different filename."))
else:
raise FormatWriteNotSupported(localized_message=_("Cannot write files in %1 format.", format_id) + "\n\n" + _("To save your changes, use a different filename."))