Limit max images per item for to image feed

sekai-soft · Dec 24, 2024 · 918fe72 · 918fe72
1 parent 24a819b
commit 918fe72
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/rss_lambda/test_to_image_feed.py b/rss_lambda/test_to_image_feed.py
@@ -14,12 +14,17 @@ def test_to_image_feed(self):
             '<p>also some random texts<br>but without images haha 2222</p> ',
             '<p>also some random texts but with images hahahaha</p><img src="https://nitter.example.com/twitter_handle/pic/pic1.jpg" /><img src="https://nitter.example.com/twitter_handle/pic/pic2.jpg" />',
             '<p>also some random texts but with images hahahaha</p><img src="https://nitter.example.com/twitter_handle/pic/pic3.jpg" />',
+            '<p>also some random texts but with a lot of images hahahaha</p><img src="https://nitter.example.com/twitter_handle/pic/pic4.jpg" /><img src="https://nitter.example.com/twitter_handle/pic/pic5.jpg" /><img src="https://nitter.example.com/twitter_handle/pic/pic6.jpg" /><img src="https://nitter.example.com/twitter_handle/pic/pic7.jpg" /><img src="https://nitter.example.com/twitter_handle/pic/pic8.jpg" />',
         ])
         self.assertEqual(
             to_image_feed(rss_text),
             nitter_rss20_response([
                 '<img src="https://nitter.example.com/twitter_handle/pic/pic1.jpg"></img>',
                 '<img src="https://nitter.example.com/twitter_handle/pic/pic2.jpg"></img>',
                 '<img src="https://nitter.example.com/twitter_handle/pic/pic3.jpg"></img>',
+                '<img src="https://nitter.example.com/twitter_handle/pic/pic4.jpg"></img>',
+                '<img src="https://nitter.example.com/twitter_handle/pic/pic5.jpg"></img>',
+                '<img src="https://nitter.example.com/twitter_handle/pic/pic6.jpg"></img>',
+                '<img src="https://nitter.example.com/twitter_handle/pic/pic7.jpg"></img>',
             ])
         )
diff --git a/rss_lambda/to_image_feed/to_image_feed.py b/rss_lambda/to_image_feed/to_image_feed.py
@@ -3,6 +3,8 @@
 from ..utils.process_rss_text import ParsedRssText, process_rss_text
 from ..utils.image_utils import extract_images_from_description
 
+MAX_IMAGES_PER_ITEM = 4
+
 def to_image_feed(rss_text: str) -> str:
     def processor(parsed_rss_text: ParsedRssText):
         root = parsed_rss_text.root
@@ -28,7 +30,7 @@ def handle_item(item):
             images = extract_images_from_description(item, root.nsmap)
             if not images:
                 return
-            for image in images:
+            for image in images[: MAX_IMAGES_PER_ITEM]:
                 handle_image(item, image)
 
         for item in items: