article thumbnail file extension agnostic

1 files changed, 22 insertions, 9 deletions

diff --git a/src/article.py b/src/article.py
index afca0df..20e1c21 100644
--- a/src/article.py
+++ b/src/article.py
@@ -55,20 +55,22 @@ class Article(WebPage):
     source: str
     html: str
     modified: datetime.datetime
+    thumbnail_path: typing.Union[None, str] = None
     folder_path: typing.Union[None, str] = None
 
     @property
     def pretty_print(self) -> str:
         return f'''
-Name:        {self.name}
-Metadata:    {self.metadata.pretty_print}
-Web dir:     {self.web_dir}
-URL:         {self.url}
-Source path: {self.source_path}
-Folder path: {self.folder_path}
-Modified:    {self.modified}
-HTML:        {truncate(self.html)}
-Source:      {truncate(self.source)}'''
+Name:           {self.name}
+Metadata:       {self.metadata.pretty_print}
+Web dir:        {self.web_dir}
+URL:            {self.url}
+Source path:    {self.source_path}
+Thumbnail path: {self.thumbnail_path}
+Folder path:    {self.folder_path}
+Modified:       {self.modified}
+HTML:           {truncate(self.html)}
+Source:         {truncate(self.source)}'''
 
 
 @dataclasses.dataclass
@@ -115,6 +117,17 @@ class ArticleGenerator:
             article_args['folder_path'] = article_folder_name
             dir_basename = os.path.basename(article_folder_name)
             article_args['name'] = dir_basename
+
+            # Look for a thumbnail (file extension agnostic)
+            thumbnail_paths = glob.glob(f'{article_args["folder_path"]}/Thumbnail*')
+
+            if len(thumbnail_paths) > 1:
+                logging.warning(f"More than one thumbnail found, using: {thumbnail_paths[0]}")
+
+            if len(thumbnail_paths) == 0:
+                logging.warning("No thumbnail image")
+            else:
+                article_args['thumbnail_path'] = '/'.join(pathlib.Path(thumbnail_paths[0]).parts[2:])
         else:
             # Article one file
             article_args['name'] = os.path.basename(filename)