""" Notebooks command for building Jupyter notebooks from Python files using Jupytext. """ import subprocess import sys from argparse import ArgumentParser, Namespace from pathlib import Path from typing import List, Tuple from rich.panel import Panel from rich.text import Text from .base import BaseCommand from ..core.exceptions import ExecutionError, ModuleNotFoundError class NotebooksCommand(BaseCommand): """Command to build Jupyter notebooks from Python files using Jupytext.""" @property def name(self) -> str: return "notebooks" @property def description(self) -> str: return "Build notebooks from Python files" def add_arguments(self, parser: ArgumentParser) -> None: """Add notebooks command arguments.""" parser.add_argument( '--module', help='Build notebook for specific module' ) parser.add_argument( '--force', action='store_true', help='Force rebuild even if notebook exists' ) parser.add_argument( '--dry-run', action='store_true', help='Show what would be built without actually building' ) def validate_args(self, args: Namespace) -> None: """Validate notebooks command arguments.""" if args.module: module_dir = self.config.modules_dir / args.module if not module_dir.exists(): raise ModuleNotFoundError(f"Module directory '{args.module}' not found") # Find module Python file in the module directory # Extract short name from module directory name if args.module.startswith(tuple(f"{i:02d}_" for i in range(100))): short_name = args.module[3:] # Remove "00_" prefix else: short_name = args.module dev_file = module_dir / f"{short_name}.py" if not dev_file.exists(): raise ModuleNotFoundError( f"No module file found in module '{args.module}'. Expected: {dev_file.name}" ) def _find_dev_files(self) -> List[Path]: """Find all module Python files in modules directory.""" dev_files = [] # Look in modules/ directory modules_dir = self.config.modules_dir for module_dir in modules_dir.iterdir(): if module_dir.is_dir() and not module_dir.name.startswith('.'): # Extract short name from module directory name module_name = module_dir.name if module_name.startswith(tuple(f"{i:02d}_" for i in range(100))): short_name = module_name[3:] # Remove "00_" prefix else: short_name = module_name # Look for module Python file (without _dev suffix) py_file = module_dir / f"{short_name}.py" if py_file.exists(): dev_files.append(py_file) return sorted(dev_files) def _convert_file(self, dev_file: Path) -> Tuple[bool, str]: """Convert a single Python file to notebook using Jupytext.""" try: # Use Jupytext from venv to convert Python file to notebook import sys venv_python = Path(sys.executable) jupytext_cmd = venv_python.parent / "jupytext" result = subprocess.run([ str(jupytext_cmd), "--to", "notebook", str(dev_file) ], capture_output=True, text=True, timeout=30, cwd=dev_file.parent) if result.returncode == 0: notebook_file = dev_file.with_suffix('.ipynb') return True, f"{dev_file.name} ā†’ {notebook_file.name}" else: error_msg = result.stderr.strip() if result.stderr.strip() else "Conversion failed" return False, error_msg except subprocess.TimeoutExpired: return False, "Conversion timed out" except FileNotFoundError: return False, "Jupytext not found. Install with: pip install jupytext" except Exception as e: return False, f"Error: {str(e)}" def run(self, args: Namespace) -> int: """Execute the notebooks command.""" self.console.print(Panel( "šŸ““ Building Notebooks from Python Files (using Jupytext)", title="Notebook Generation", border_style="bright_cyan" )) # Find files to convert if args.module: module_dir = self.config.modules_dir / args.module # Extract short name from module directory name module_name = args.module if module_name.startswith(tuple(f"{i:02d}_" for i in range(100))): short_name = module_name[3:] # Remove "00_" prefix else: short_name = module_name dev_file = module_dir / f"{short_name}.py" if dev_file.exists(): dev_files = [dev_file] else: dev_files = [] self.console.print(f"šŸ”„ Building notebook for module: {args.module}") else: dev_files = self._find_dev_files() if not dev_files: self.console.print(Panel( "[yellow]āš ļø No *.py files found in modules/[/yellow]", title="Nothing to Convert", border_style="yellow" )) return 0 self.console.print(f"šŸ”„ Building notebooks for {len(dev_files)} modules...") # Dry run mode if args.dry_run: self.console.print("\n[cyan]Dry run mode - would convert:[/cyan]") for dev_file in dev_files: module_name = dev_file.parent.name self.console.print(f" ā€¢ {module_name}: {dev_file.name}") return 0 # Convert files success_count = 0 error_count = 0 for dev_file in dev_files: success, message = self._convert_file(dev_file) module_name = dev_file.parent.name if success: success_count += 1 self.console.print(f" āœ… {module_name}: {message}") else: error_count += 1 self.console.print(f" āŒ {module_name}: {message}") # Summary self._print_summary(success_count, error_count) return 0 if error_count == 0 else 1 def _print_summary(self, success_count: int, error_count: int) -> None: """Print command execution summary.""" summary_text = Text() if success_count > 0: summary_text.append(f"āœ… Successfully built {success_count} notebook(s)\n", style="bold green") if error_count > 0: summary_text.append(f"āŒ Failed to build {error_count} notebook(s)\n", style="bold red") if success_count > 0: summary_text.append("\nšŸ’” Next steps:\n", style="bold yellow") summary_text.append(" ā€¢ Open notebooks with: jupyter lab\n", style="white") summary_text.append(" ā€¢ Work interactively in the notebooks\n", style="white") summary_text.append(" ā€¢ Export code with: tito package export\n", style="white") summary_text.append(" ā€¢ Run tests with: tito module test\n", style="white") border_style = "green" if error_count == 0 else "yellow" self.console.print(Panel( summary_text, title="Notebook Generation Complete", border_style=border_style ))