Mastodon
/
Toot-Mastodon-CLI-TUI-client-Linux
mirror of https://github.com/ihabunek/toot
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #484 from AnonymouX47/improve-image-support 

Improve Image Support
This commit is contained in:
Daniel Schwarz 2024-06-05 17:59:40 -04:00 committed by GitHub
parent b97a995dc4 1ea2e29e25
commit c622cbbccd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
toot/tui/images.py
View File

@ -10,6 +10,9 @@ try:
from term_image import disable_queries # prevent phantom keystrokes from term_image import disable_queries # prevent phantom keystrokes
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
_IMAGE_PIXEL_FORMATS = frozenset({'kitty', 'iterm'})
_ImageCls = None
TuiScreen = UrwidImageScreen TuiScreen = UrwidImageScreen
disable_queries() disable_queries()
@ -17,17 +20,24 @@ try:
return True return True
def can_render_pixels(image_format): def can_render_pixels(image_format):
return image_format in ['kitty', 'iterm'] return image_format in _IMAGE_PIXEL_FORMATS
def get_base_image(image, image_format) -> BaseImage: def get_base_image(image, image_format) -> BaseImage:
# we don't autodetect kitty, iterm; we choose based on option switches # we don't autodetect kitty, iterm; we choose based on option switches
BaseImage.forced_support = True
if image_format == 'kitty': global _ImageCls
return KittyImage(image)
elif image_format == 'iterm': if not _ImageCls:
return ITerm2Image(image) _ImageCls = (
else: KittyImage
return BlockImage(image) if image_format == 'kitty'
else ITerm2Image
if image_format == 'iterm'
else BlockImage
)
_ImageCls.forced_support = True
return _ImageCls(image)
def resize_image(basewidth: int, baseheight: int, img: Image.Image) -> Image.Image: def resize_image(basewidth: int, baseheight: int, img: Image.Image) -> Image.Image:
if baseheight and not basewidth: if baseheight and not basewidth: