From 5c4f1a87d2cce009ae9accbc50e5705c9b22bfc4 Mon Sep 17 00:00:00 2001
From: acite <1498045907@qq.com>
Date: Tue, 9 Sep 2025 14:09:08 +0800
Subject: [PATCH] [feat] update-video.py

---
 .idea/.idea.Abyss/.idea/workspace.xml |  17 +---
 Abyss/Toolkits/update-video.py        | 115 ++++++++++++++++++++------
 2 files changed, 94 insertions(+), 38 deletions(-)

diff --git a/.idea/.idea.Abyss/.idea/workspace.xml b/.idea/.idea.Abyss/.idea/workspace.xml
index 19883db..fec12a1 100644
--- a/.idea/.idea.Abyss/.idea/workspace.xml
+++ b/.idea/.idea.Abyss/.idea/workspace.xml
@@ -10,21 +10,8 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="bf317275-3039-49bb-a475-725a800a0cce" name="Changes" comment="">
-      <change afterPath="$PROJECT_DIR$/Abyss/Toolkits/image-creator.py" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/Abyss/Toolkits/image-sum.py" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/Abyss/Toolkits/update-tags.py" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/Abyss/Toolkits/update-video.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/.idea/.idea.Abyss/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/.idea.Abyss/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss.sln.DotSettings.user" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss.sln.DotSettings.user" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Controllers/Media/ImageController.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Controllers/Media/ImageController.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Controllers/Media/VideoController.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Controllers/Media/VideoController.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Services/ConfigureService.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Services/ConfigureService.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Services/ResourceService.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Services/ResourceService.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Services/UserService.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Services/UserService.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Components/Tools/NaturalStringComparer.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Components/Tools/NaturalStringComparer.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Model/Bookmark.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Model/Bookmark.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Model/Comic.cs" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Model/Comic.cs" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Abyss/Properties/launchSettings.json" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Properties/launchSettings.json" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/Abyss/Toolkits/update-video.py" beforeDir="false" afterPath="$PROJECT_DIR$/Abyss/Toolkits/update-video.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -92,7 +79,7 @@
     "RunOnceActivity.TerminalTabsStorage.copyFrom.TerminalArrangementManager.252": "true",
     "RunOnceActivity.git.unshallow": "true",
     "XThreadsFramesViewSplitterKey": "0.30266345",
-    "git-widget-placeholder": "dev-task",
+    "git-widget-placeholder": "main",
     "last_opened_file_path": "/storage/Images/31/summary.json",
     "node.js.detected.package.eslint": "true",
     "node.js.detected.package.tslint": "true",
diff --git a/Abyss/Toolkits/update-video.py b/Abyss/Toolkits/update-video.py
index a72d794..2b509ff 100644
--- a/Abyss/Toolkits/update-video.py
+++ b/Abyss/Toolkits/update-video.py
@@ -9,7 +9,7 @@ def get_video_duration(video_path):
     """Get video duration in milliseconds using ffprobe"""
     try:
         cmd = [
-            'ffprobe', 
+            'ffprobe',
             '-v', 'error',
             '-show_entries', 'format=duration',
             '-of', 'default=noprint_wrappers=1:nokey=1',
@@ -44,15 +44,15 @@ def create_thumbnails(video_path, gallery_path, num_thumbnails=10):
     except (subprocess.CalledProcessError, ValueError) as e:
         print(f"Could not get duration for '{video_path}': {e}. Skipping thumbnail creation.")
         return
-    
+
     if duration <= 0:
         print(f"Warning: Invalid video duration for '{video_path}'. Skipping thumbnail creation.")
         return
 
     interval = duration / (num_thumbnails + 1)
-    
+
     print(f"Generating {num_thumbnails} thumbnails for {video_path.name}...")
-    
+
     for i in range(num_thumbnails):
         timestamp = (i + 1) * interval
         output_thumbnail_path = gallery_path / f"{i}.jpg"
@@ -68,6 +68,45 @@ def create_thumbnails(video_path, gallery_path, num_thumbnails=10):
         except subprocess.CalledProcessError as e:
             print(f"  Error extracting thumbnail {i}.jpg: {e}")
 
+def create_cover(video_path, output_path, time_percent):
+    """
+    Creates a cover image from a video at a specified time percentage.
+    """
+    try:
+        subprocess.run(['ffmpeg', '-version'], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        print("Error: ffmpeg is not installed or not in your PATH. Cannot create cover.")
+        return
+
+    try:
+        duration_cmd = [
+            'ffprobe', '-v', 'error', '-show_entries', 'format=duration',
+            '-of', 'default=noprint_wrappers=1:nokey=1', str(video_path)
+        ]
+        result = subprocess.run(duration_cmd, capture_output=True, text=True, check=True)
+        duration = float(result.stdout)
+    except (subprocess.CalledProcessError, ValueError) as e:
+        print(f"Could not get duration for '{video_path}': {e}. Cannot create cover.")
+        return
+
+    if duration <= 0:
+        print(f"Warning: Invalid video duration for '{video_path}'. Cannot create cover.")
+        return
+
+    timestamp = duration * time_percent
+
+    ffmpeg_cmd = [
+        'ffmpeg', '-ss', str(timestamp), '-i', str(video_path),
+        '-vframes', '1', str(output_path), '-y'
+    ]
+
+    print(f"Creating cover image from video at {timestamp:.2f} seconds...")
+    try:
+        subprocess.run(ffmpeg_cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        print(f"Cover image created at {output_path}")
+    except subprocess.CalledProcessError as e:
+        print(f"Error creating cover image: {e}")
+
 def update_summary(base_path, name_input=None, author_input=None):
     """
     Updates the summary.json file for a given path.
@@ -87,7 +126,7 @@ def update_summary(base_path, name_input=None, author_input=None):
         "like": 0,
         "author": author_input if author_input is not None else "anonymous"
     }
-    
+
     # Load existing summary if available
     if summary_path.exists():
         try:
@@ -99,13 +138,13 @@ def update_summary(base_path, name_input=None, author_input=None):
                         default_summary[key] = existing_data[key]
         except json.JSONDecodeError:
             print("Warning: Invalid JSON in summary.json, using defaults")
-    
+
     # Update duration from video file
     if video_path.exists():
         default_summary["duration"] = get_video_duration(video_path)
     else:
         print(f"Warning: video.mp4 not found at {video_path}")
-    
+
     # Update gallery from directory
     if gallery_path.exists() and gallery_path.is_dir():
         gallery_files = []
@@ -116,11 +155,11 @@ def update_summary(base_path, name_input=None, author_input=None):
         default_summary["gallery"] = gallery_files
     else:
         print(f"Warning: gallery directory not found at {gallery_path}")
-    
+
     # Write updated summary
     with open(summary_path, 'w', encoding='utf-8') as f:
         json.dump(default_summary, f, indent=4, ensure_ascii=False)
-    
+
     print(f"Summary updated successfully at {summary_path}")
 
 def find_next_directory(base_path):
@@ -129,7 +168,7 @@ def find_next_directory(base_path):
     for item in base_path.iterdir():
         if item.is_dir() and item.name.isdigit():
             existing_dirs.add(int(item.name))
-    
+
     next_num = 1
     while next_num in existing_dirs:
         next_num += 1
@@ -141,8 +180,9 @@ def main():
         print("Commands:")
         print("  -u <path>              Update the summary.json in the specified path.")
         print("  -a <video_file> <path> Add a new video project in a new directory under the specified path.")
+        print("  -c <path> <time>       Create a cover image from the video in the specified path at a given time percentage (0.0-1.0).")
         sys.exit(1)
-        
+
     command = sys.argv[1]
 
     if command == '-u':
@@ -159,7 +199,7 @@ def main():
         if len(sys.argv) != 4:
             print("Usage: python script.py -a <video_file> <path>")
             sys.exit(1)
-        
+
         video_source_path = Path(sys.argv[2])
         base_path = Path(sys.argv[3])
 
@@ -170,37 +210,66 @@ def main():
         if not base_path.is_dir():
             print(f"Error: Base path not found or is not a directory: {base_path}")
             sys.exit(1)
-        
+
         # Find a new directory name (e.g., "1", "2", "3")
         new_dir_name = find_next_directory(base_path)
         new_project_path = base_path / new_dir_name
-        
+
         # Create the new project directory and the gallery subdirectory
         new_project_path.mkdir(exist_ok=True)
         gallery_path = new_project_path / "gallery"
         gallery_path.mkdir(exist_ok=True)
         print(f"New project directory created at {new_project_path}")
-        
+
         # Copy video file to the new directory
         shutil.copy(video_source_path, new_project_path / "video.mp4")
         print(f"Video copied to {new_project_path / 'video.mp4'}")
 
-        # --- 新增功能：自动生成缩略图 ---
+        # Auto-generate thumbnails
         video_dest_path = new_project_path / "video.mp4"
         create_thumbnails(video_dest_path, gallery_path)
-        # ------------------------------------
 
-        # Get user input for name and author
-        video_name = input("Enter the video name: ")
-        video_author = input("Enter the author's name: ")
+        # Get user input for name and author, with a prompt for default values
+        print("\nEnter the video name (press Enter to use the original filename):")
+        video_name = input(f"Video Name [{video_source_path.stem}]: ")
+        if not video_name:
+            video_name = video_source_path.stem
 
-        # Update the summary with user input
+        print("\nEnter the author's name (press Enter to use 'Anonymous'):")
+        video_author = input("Author Name [Anonymous]: ")
+        if not video_author:
+            video_author = "Anonymous"
+
+        # Update the summary with user input or default values
         update_summary(new_project_path, name_input=video_name, author_input=video_author)
 
+    elif command == '-c':
+        if len(sys.argv) != 4:
+            print("Usage: python script.py -c <path> <time>")
+            sys.exit(1)
+
+        base_path = Path(sys.argv[2])
+        video_path = base_path / "video.mp4"
+        cover_path = base_path / "cover.jpg"
+
+        try:
+            time_percent = float(sys.argv[3])
+            if not 0.0 <= time_percent <= 1.0:
+                raise ValueError
+        except ValueError:
+            print("Error: Time value must be a number between 0.0 and 1.0.")
+            sys.exit(1)
+
+        if not video_path.exists() or not video_path.is_file():
+            print(f"Error: video.mp4 not found at {video_path}")
+            sys.exit(1)
+
+        create_cover(video_path, cover_path, time_percent)
+
     else:
-        print("Invalid command. Use -u or -a.")
+        print("Invalid command. Use -u, -a, or -c.")
         print("Usage: python script.py <command> [arguments]")
         sys.exit(1)
 
 if __name__ == "__main__":
-    main()
+    main()
\ No newline at end of file