ytcollector 1.0.4__tar.gz → 1.0.5__tar.gz

{ytcollector-1.0.4 → ytcollector-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.0.4
+Version: 1.0.5
 Summary: SBS 데이터셋 수집기
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{ytcollector-1.0.4 → ytcollector-1.0.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ytcollector"
-version = "1.0.4"
+version = "1.0.5"
 description = "SBS 데이터셋 수집기"
 readme = "README.md"
 requires-python = ">=3.9"

{ytcollector-1.0.4 → ytcollector-1.0.5}/ytcollector/__init__.py

@@ -3,7 +3,7 @@ SBS Dataset Collector - YouTube 영상 수집 및 YOLO-World 검증 파이프라
 """
 from pathlib import Path
 
-__version__ = "1.0.4"
+__version__ = "1.0.5"
 __author__ = "SBS Dataset Team"
 
 # Package root directory

{ytcollector-1.0.4 → ytcollector-1.0.5}/ytcollector/cli.py

@@ -48,24 +48,38 @@ def run_download(args):
     
     base_dir = Path(args.dir) if args.dir else Path.cwd()
     
-    # 파일 경로: video/{task}/youtube_url.txt
-    txt_file = get_url_file_path(base_dir, args.task)
+    # argparse에서 nargs='+'로 받아오면 args.task는 항상 리스트
+    tasks = args.task if isinstance(args.task, list) else [args.task]
     
-    if not txt_file.exists():
-        logger.error(f"URL file not found: {txt_file}")
-        logger.info("Run 'downloader init' first to create project structure")
-        return
-    
-    logger.info(f"Starting{' fast' if args.fast else ''} download for task: {args.task}")
-    
-    if args.fast:
-        from .downloader import download_from_txt_parallel
-        results = download_from_txt_parallel(txt_file, args.task, base_dir, max_count=args.count)
-    else:
-        results = download_from_txt(txt_file, args.task, base_dir, max_count=args.count)
+    total_success = 0
+    total_processed = 0
     
-    success_count = sum(1 for r in results if r.get('success'))
-    print(f"✓ Download complete: {success_count}/{len(results)} successful")
+    for task in tasks:
+        logger.info(f"=== Processing Task: {task} ===")
+        
+        # 파일 경로: video/{task}/youtube_url.txt
+        txt_file = get_url_file_path(base_dir, task)
+        
+        if not txt_file.exists():
+            logger.error(f"URL file not found: {txt_file}")
+            logger.info(f"Skipping {task}. Run 'ytcollector init' first.")
+            continue
+        
+        logger.info(f"Starting{' fast' if args.fast else ''} download for task: {task}")
+        
+        if args.fast:
+            from .downloader import download_from_txt_parallel
+            results = download_from_txt_parallel(txt_file, task, base_dir, max_count=args.count)
+        else:
+            results = download_from_txt(txt_file, task, base_dir, max_count=args.count)
+        
+        success_count = sum(1 for r in results if r.get('success'))
+        total_success += success_count
+        total_processed += len(results)
+        
+        print(f"✓ Task '{task}' complete: {success_count}/{len(results)} successful")
+        
+    print(f"\n✓ All tasks complete: {total_success}/{total_processed} successful total")
 
 
 def run_download_single(args):
@@ -172,7 +186,7 @@ Examples:
     
     # Download
     download_parser = subparsers.add_parser('download', help='Download from youtube_url.txt')
-    download_parser.add_argument('--task', '-t', required=True, choices=VALID_TASKS)
+    download_parser.add_argument('--task', '-t', required=True, nargs='+', choices=VALID_TASKS, help='One or more tasks (e.g. face tattoo)')
     download_parser.add_argument('--count', '-n', type=int, help='Max videos to collect (default: 1000)')
     download_parser.add_argument('--fast', action='store_true', help='Enable fast parallel downloading')
     
@@ -185,12 +199,12 @@ Examples:
     
     # Verify
     verify_parser = subparsers.add_parser('verify', help='Verify with YOLO-World')
-    verify_parser.add_argument('--task', '-t', required=True, choices=VALID_TASKS)
+    verify_parser.add_argument('--task', '-t', required=True, nargs='+', choices=VALID_TASKS)
     verify_parser.add_argument('--video', '-v', help='Specific video file')
     
     # Pipeline
     pipeline_parser = subparsers.add_parser('pipeline', help='Full pipeline')
-    pipeline_parser.add_argument('--task', '-t', required=True, choices=VALID_TASKS)
+    pipeline_parser.add_argument('--task', '-t', required=True, nargs='+', choices=VALID_TASKS)
     pipeline_parser.add_argument('--verify', action='store_true')
     
     # List tasks

{ytcollector-1.0.4 → ytcollector-1.0.5}/ytcollector.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ytcollector
-Version: 1.0.4
+Version: 1.0.5
 Summary: SBS 데이터셋 수집기
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown