fix: News post images (#2)

Computer-Science-Course · Nov 19, 2024 · 204cf63 · 204cf63
1 parent c3f5541
commit 204cf63
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/back/raspa_tacho/models/news_info.py b/back/raspa_tacho/models/news_info.py
@@ -7,21 +7,24 @@ def __init__(
         self,
         datetime,
         title,
-        src_image,
+        cover_image_src,
+        post_images_srcs,
         paragraphs,
         related_links,
     ):
         self.datetime = datetime
         self.title = title
-        self.src_image = src_image
+        self.cover_image_src = cover_image_src
+        self.post_images_srcs = post_images_srcs
         self.paragraphs = paragraphs
         self.related_links = related_links
 
     def __str__(self):
         return '\n'.join([
             f'Datetime: {self.datetime}',
             f'Title: {self.title}',
-            f'Image src: {self.src_image}',
+            f'Cover image src: {self.cover_image_src}',
+            f'Post images srcs: {self.post_images_srcs}',
             '\n\n'.join(self.paragraphs),
             f'Related links: {self.related_links}'
         ])
diff --git a/back/raspa_tacho/services/tome/news.py b/back/raspa_tacho/services/tome/news.py
@@ -21,7 +21,8 @@ def info(self) -> NewsInfo:
         """
         raw_datetime = self.content.select('header ul .meta-date')[0].text
         title = self.content.select('.page-title')[0].text
-        # src_image = self.content.select('.panel .panel-body .container-fluid .row img')[0]['src']
+        cover_image_src = self.content.select('.hero-section img')[0]['src']
+        post_images_srcs = [image['src'] for image in self.content.select('.ct-container > article img')]
         raw_paragraphs = self.content.select('article p')
         raw_related_link_anchors = self.content.select('.panel .panel-body .panel-group .panel-default > a')
         raw_related_link_titles = self.content.select('.panel .panel-body .panel-group .panel-default > a p')
@@ -38,9 +39,10 @@ def info(self) -> NewsInfo:
         return NewsInfo(
             str(datetime_object),
             title,
-            '',
+            cover_image_src,
+            post_images_srcs,
             paragraphs,
-            related_links
+            related_links,
         )
 
     @staticmethod