wip add org traces

Author: maxkorp

organize_traces.py

@@ -0,0 +1,170 @@
+#!/usr/bin/env python3
+"""
+Script to organize Playwright trace folders by framework, feature, and test name.
+"""
+
+import os
+import re
+import shutil
+import argparse
+from pathlib import Path
+from collections import defaultdict
+
+def extract_test_info(error_context_path):
+    """Extract framework, feature, and test name from error-context.md file."""
+    try:
+        with open(error_context_path, 'r', encoding='utf-8') as f:
+            content = f.read()
+
+                # Extract test name
+        # Handle both formats: "[Framework] Test Name" and "Feature >> [Framework] Test Name"
+        name_match = re.search(r'- Name: (?:.*? >> )?\[([^\]]+)\]\s*(.+)', content)
+        if not name_match:
+            return None, None, None
+
+        framework_bracket = name_match.group(1)
+        test_name = name_match.group(2).strip()
+
+        # Extract location to get framework and feature
+        location_match = re.search(r'- Location: .*?/tests/([^/]+)/([^:]+)\.spec\.ts', content)
+        if not location_match:
+            return None, None, None
+
+        framework_from_path = location_match.group(1)
+        feature_file = location_match.group(2)
+
+        # Clean up framework name (remove 'Tests' suffix if present)
+        framework = framework_from_path.replace('Tests', '').lower()
+
+        # Extract feature name from filename
+        feature = feature_file
+
+        return framework, feature, test_name
+
+    except Exception as e:
+        print(f"Error reading {error_context_path}: {e}")
+        return None, None, None
+
+def analyze_traces_folder(traces_path):
+    """Analyze the traces folder and return the planned folder structure."""
+    traces_path = Path(traces_path)
+
+    if not traces_path.exists():
+        print(f"Error: {traces_path} does not exist")
+        return None
+
+    structure = defaultdict(lambda: defaultdict(list))
+    unknown_folders = []
+
+    for folder in traces_path.iterdir():
+        if not folder.is_dir():
+            continue
+
+        error_context_path = folder / "error-context.md"
+
+        if error_context_path.exists():
+            framework, feature, test_name = extract_test_info(error_context_path)
+
+            if framework and feature and test_name:
+                structure[framework][feature].append((test_name, folder.name))
+            else:
+                unknown_folders.append(folder.name)
+        else:
+            unknown_folders.append(folder.name)
+
+    return structure, unknown_folders
+
+def print_planned_structure(structure, unknown_folders):
+    """Print the planned folder structure."""
+    print("Planned folder structure:")
+    print("=" * 50)
+
+    for framework in sorted(structure.keys()):
+        print(f"\n{framework}/")
+        for feature in sorted(structure[framework].keys()):
+            print(f"  {feature}/")
+            for test_name, folder_name in structure[framework][feature]:
+                print(f"    {test_name}/")
+                print(f"      -> {folder_name}")
+
+    if unknown_folders:
+        print(f"\nunknown/")
+        for folder_name in sorted(unknown_folders):
+            print(f"  -> {folder_name}")
+
+    print("\n" + "=" * 50)
+
+def execute_reorganization(traces_path, structure, unknown_folders):
+    """Execute the folder reorganization."""
+    traces_path = Path(traces_path)
+
+    # Create framework folders and move test folders
+    for framework, features in structure.items():
+        framework_path = traces_path / framework
+        framework_path.mkdir(exist_ok=True)
+
+        for feature, tests in features.items():
+            feature_path = framework_path / feature
+            feature_path.mkdir(exist_ok=True)
+
+            for test_name, folder_name in tests:
+                test_path = feature_path / test_name
+                test_path.mkdir(exist_ok=True)
+
+                source_folder = traces_path / folder_name
+                if source_folder.exists():
+                    # Move the entire folder
+                    shutil.move(str(source_folder), str(test_path / folder_name))
+                    print(f"Moved {folder_name} to {framework}/{feature}/{test_name}/")
+
+    # Move unknown folders
+    if unknown_folders:
+        unknown_path = traces_path / "unknown"
+        unknown_path.mkdir(exist_ok=True)
+
+        for folder_name in unknown_folders:
+            source_folder = traces_path / folder_name
+            if source_folder.exists():
+                shutil.move(str(source_folder), str(unknown_path / folder_name))
+                print(f"Moved {folder_name} to unknown/")
+
+def main():
+    parser = argparse.ArgumentParser(description='Organize Playwright trace folders by framework, feature, and test name.')
+    parser.add_argument('--dry-run', action='store_true', help='Show planned structure without making changes')
+    args = parser.parse_args()
+
+    traces_path = Path.home() / "Downloads" / "playwright-traces-8"
+
+    print(f"Analyzing traces folder: {traces_path}")
+
+    # Analyze the current structure
+    result = analyze_traces_folder(traces_path)
+    if result is None:
+        return
+
+    structure, unknown_folders = result
+
+    if not structure and not unknown_folders:
+        print("No folders found to organize.")
+        return
+
+    # Print planned structure
+    print_planned_structure(structure, unknown_folders)
+
+    if args.dry_run:
+        print("\nDRY RUN MODE: No changes were made.")
+        return
+
+    # Ask for confirmation
+    response = input("\nDo you want to proceed with the reorganization? (y/N): ")
+    if response.lower() not in ['y', 'yes']:
+        print("Operation cancelled.")
+        return
+
+    # Execute the reorganization
+    print("\nExecuting reorganization...")
+    execute_reorganization(traces_path, structure, unknown_folders)
+    print("\nReorganization complete!")
+
+if __name__ == "__main__":
+    main()