feat: support customize filename format

Closes #116

Author: Ljzd-PRO

ktoolbox/action/job.py

@@ -11,7 +11,7 @@
 
 from ktoolbox._enum import PostFileTypeEnum, DataStorageNameEnum
 from ktoolbox.action import ActionRet, fetch_creator_posts, FetchInterruptError
-from ktoolbox.action.utils import generate_post_path_name, filter_posts_by_date
+from ktoolbox.action.utils import generate_post_path_name, filter_posts_by_date, generate_filename
 from ktoolbox.api.model import Post, Attachment
 from ktoolbox.configuration import config, PostStructureConfiguration
 from ktoolbox.job import Job, CreatorIndices
@@ -68,7 +68,8 @@ async def create_job_from_post(
                 config.job.block_list
             )
         ):
-            alt_filename = f"{i + 1}{file_path_obj.suffix}" if config.job.sequential_filename else file_path_obj.name
+            basic_filename = f"{i + 1}{file_path_obj.suffix}" if config.job.sequential_filename else file_path_obj.name
+            alt_filename = generate_filename(post, basic_filename)
             jobs.append(
                 Job(
                     path=attachments_path,

ktoolbox/action/utils.py

@@ -8,32 +8,53 @@
 from ktoolbox.configuration import config
 from ktoolbox.job import CreatorIndices
 
-__all__ = ["generate_post_path_name", "filter_posts_by_date", "filter_posts_by_indices"]
+__all__ = ["generate_post_path_name", "generate_filename", "filter_posts_by_date", "filter_posts_by_indices"]
+
+TIME_FORMAT = "%Y-%m-%d"
 
 
 def generate_post_path_name(post: Post) -> str:
     """Generate directory name for post to save."""
     if config.job.post_id_as_path or not post.title:
         return post.id
     else:
-        time_format = "%Y-%m-%d"
         try:
             return sanitize_filename(
                 config.job.post_dirname_format.format(
                     id=post.id,
                     user=post.user,
                     service=post.service,
                     title=post.title,
-                    added=post.added.strftime(time_format) if post.added else "",
-                    published=post.published.strftime(time_format) if post.published else "",
-                    edited=post.edited.strftime(time_format) if post.edited else ""
+                    added=post.added.strftime(TIME_FORMAT) if post.added else "",
+                    published=post.published.strftime(TIME_FORMAT) if post.published else "",
+                    edited=post.edited.strftime(TIME_FORMAT) if post.edited else ""
                 )
             )
         except KeyError as e:
             logger.error(f"`JobConfiguration.post_dirname_format` contains invalid key: {e}")
             exit(1)
 
 
+def generate_filename(post: Post, basic_name: str) -> str:
+    """Generate download filename"""
+    try:
+        return sanitize_filename(
+            config.job.filename_format.format(
+                basic_name,
+                id=post.id,
+                user=post.user,
+                service=post.service,
+                title=post.title,
+                added=post.added.strftime(TIME_FORMAT) if post.added else "",
+                published=post.published.strftime(TIME_FORMAT) if post.published else "",
+                edited=post.edited.strftime(TIME_FORMAT) if post.edited else ""
+            )
+        )
+    except KeyError as e:
+        logger.error(f"`JobConfiguration.filename_format` contains invalid key: {e}")
+        exit(1)
+
+
 def _match_post_date(
         post: Post,
         start_date: Optional[datetime],