diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/article.py | 31 |
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) |