mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-01-31 11:46:16 +08:00
069527a7f1
- Add vulnerability snapshot DTO, handler, repository, and service layer with comprehensive test coverage - Add screenshot snapshot DTO, handler, repository, and service layer for snapshot management - Reorganize directories tab from secondary assets navigation to primary navigation in scan history and target layouts - Update frontend navigation to include FolderSearch icon for directories tab with badge count display - Add i18n translations for directories tab in English and Chinese messages - Implement seed data generation tools with Python API client for testing and data population - Add data generator, error handler, and progress tracking utilities for seed API - Update target validator to support new snapshot-related validations - Refactor organization and vulnerability handlers to support snapshot operations - Add integration tests and property-based tests for vulnerability snapshot functionality - Update Go module dependencies to support new snapshot features
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
"""
|
|
Progress Tracker Module
|
|
|
|
Displays progress and statistics during data generation.
|
|
"""
|
|
|
|
import time
|
|
from typing import List
|
|
|
|
|
|
class ProgressTracker:
|
|
"""Tracks and displays progress during data generation."""
|
|
|
|
def __init__(self):
|
|
"""Initialize progress tracker."""
|
|
self.current_phase = ""
|
|
self.current_count = 0
|
|
self.total_count = 0
|
|
self.success_count = 0
|
|
self.error_count = 0
|
|
self.errors: List[str] = []
|
|
self.start_time = time.time()
|
|
self.phase_start_time = 0.0
|
|
|
|
def start_phase(self, phase_name: str, total: int, emoji: str = ""):
|
|
"""
|
|
Start a new phase.
|
|
|
|
Args:
|
|
phase_name: Name of the phase (e.g., "Creating organizations")
|
|
total: Total number of items to process
|
|
emoji: Emoji icon for the phase
|
|
"""
|
|
self.current_phase = phase_name
|
|
self.current_count = 0
|
|
self.total_count = total
|
|
self.success_count = 0
|
|
self.error_count = 0
|
|
self.phase_start_time = time.time()
|
|
|
|
prefix = f"{emoji} " if emoji else ""
|
|
print(f"{prefix}{phase_name}...", end=" ", flush=True)
|
|
|
|
def update(self, count: int):
|
|
"""
|
|
Update progress.
|
|
|
|
Args:
|
|
count: Current count
|
|
"""
|
|
self.current_count = count
|
|
|
|
|
|
def add_success(self, count: int):
|
|
"""
|
|
Record successful operations.
|
|
|
|
Args:
|
|
count: Number of successful operations
|
|
"""
|
|
self.success_count += count
|
|
|
|
def add_error(self, error: str):
|
|
"""
|
|
Record an error.
|
|
|
|
Args:
|
|
error: Error message
|
|
"""
|
|
self.error_count += 1
|
|
self.errors.append(error)
|
|
|
|
def finish_phase(self):
|
|
"""Complete current phase and display summary."""
|
|
elapsed = time.time() - self.phase_start_time
|
|
print(f"[{self.current_count}/{self.total_count}] ✓ {self.success_count} created", end="")
|
|
|
|
if self.error_count > 0:
|
|
print(f" ({self.error_count} errors)", end="")
|
|
|
|
print() # New line
|
|
|
|
|
|
def print_summary(self):
|
|
"""Print final summary."""
|
|
total_time = time.time() - self.start_time
|
|
|
|
print()
|
|
print("✅ Test data generation completed!")
|
|
print(f" Total time: {total_time:.1f}s")
|
|
print(f" Success: {self.success_count:,} records")
|
|
|
|
if self.error_count > 0:
|
|
print(f" Errors: {self.error_count} records")
|