Class ModuleProgressBar

codewiki/cli/utils/progress.py:176–221

Progress bar for module-by-module generation.

class ModuleProgressBar:
    """Progress bar for module-by-module generation."""
    
    def __init__(self, total_modules: int, verbose: bool = False):
        """
        Initialize module progress bar.
        
        Args:
            total_modules: Total number of modules to process
            verbose: Enable verbose output
        """
        self.total_modules = total_modules
        self.current_module = 0
        self.verbose = verbose
        self.bar = None
        
        if not verbose:
            self.bar = click.progressbar(
                length=total_modules,
                label="Generating modules",
                show_eta=True,
                show_percent=True,
            )
            self.bar.__enter__()
    
    def update(self, module_name: str, cached: bool = False):
        """
        Update progress for a module.
        
        Args:
            module_name: Name of the module
            cached: Whether the module was loaded from cache
        """
        self.current_module += 1
        
        if self.verbose:
            status = "✓ (cached)" if cached else "⟳ (generating)"
            click.echo(f"  [{self.current_module}/{self.total_modules}] {module_name}... {status}")
        elif self.bar:
            self.bar.update(1)
    
    def finish(self):
        """Finish progress bar."""
        if self.bar:
            self.bar.__exit__(None, None, None)
            self.bar = None