diff --git a/py34/post.py b/py34/post.py index d0e4b6e..3a725c5 100644 --- a/py34/post.py +++ b/py34/post.py @@ -1,5 +1,5 @@ from .scraper import scraper -from .url import ImageURL, ThumbnailURL, IMAGE_FORMATS +from .url import ImageURL, ThumbnailURL, IMAGE_FORMATS, VIDEO_FORMATS from io import BytesIO from PIL import Image from PIL.ImageFile import ImageFile @@ -57,7 +57,7 @@ class Post: def get_image_data(self) -> bytes: if self._image_data is not None: return self._image_data - for ext in IMAGE_FORMATS: + for ext in VIDEO_FORMATS + IMAGE_FORMATS: try: self._image_data = scraper.get(ImageURL(self.image_dir, self.image_id, ext), retry=False) self._image_format = ext