ytcollector 1.0.4__py3-none-any.whl → 1.0.5__py3-none-any.whl

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/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.dist-info → ytcollector-1.0.5.dist-info}/METADATA

@@ -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.dist-info → ytcollector-1.0.5.dist-info}/RECORD

@@ -1,12 +1,12 @@
 config/settings.py,sha256=RcK41kaUC0zam5SsdXfb7u_qjM_TlJDa0a8hC_MGacQ,1197
-ytcollector/__init__.py,sha256=6gfthb-C32RDhgM7eJeMql7HSQjxcgRFQesqoRjV6gg,365
-ytcollector/cli.py,sha256=bp9DnVD0bgSHD2sf8no_6s9uZHA7SjYQWdPA-3HaRBY,7720
+ytcollector/__init__.py,sha256=JRX6gn5SOiHpAiuuD7x8Ist8ZWI8zNujw78pNwGsmK8,365
+ytcollector/cli.py,sha256=meCnT3cMBF15AkHpPeUdCFjL6WHZI-UB_F1echeU6is,8328
 ytcollector/config.py,sha256=SPhdw78A9v6ZEI7ykMi_37--BGdWg5hmgZkawFxfxjg,2436
 ytcollector/downloader.py,sha256=3Y54mudqQ1LqfFtKrU3uloNp2Oz-hcf28GwEI6eDUa0,14944
 ytcollector/utils.py,sha256=gInDx6adV-SfQ2SH5_i8w1gvYL-Nsmz1e1W__gCdVH8,4654
 ytcollector/verifier.py,sha256=8Nn3b6fTQYxCGPt01kJMDSZ2hy8gk54deSayOpBuY48,6286
-ytcollector-1.0.4.dist-info/METADATA,sha256=s-AAzo8l9x0baBMRcHQnIryHFmyjw3RbFqPn6EiZo6Y,3543
-ytcollector-1.0.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ytcollector-1.0.4.dist-info/entry_points.txt,sha256=PoanZbxogGnV4tLcZZkla0Yh7OvPtqcukDYr563w5RA,53
-ytcollector-1.0.4.dist-info/top_level.txt,sha256=TVfBZHJgYRfSSTgLJELvOoMA55qR8kWuxtiIaItwzIQ,19
-ytcollector-1.0.4.dist-info/RECORD,,
+ytcollector-1.0.5.dist-info/METADATA,sha256=VRpphUKlM4HelokurDjF6w3cSzYKfIzBVC7JwLO3rLY,3543
+ytcollector-1.0.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ytcollector-1.0.5.dist-info/entry_points.txt,sha256=PoanZbxogGnV4tLcZZkla0Yh7OvPtqcukDYr563w5RA,53
+ytcollector-1.0.5.dist-info/top_level.txt,sha256=TVfBZHJgYRfSSTgLJELvOoMA55qR8kWuxtiIaItwzIQ,19
+ytcollector-1.0.5.dist-info/RECORD,,