""" Progress handlers for PyUCC GUI. Handles progress callbacks with batch processing to avoid GUI blocking. """ from pathlib import Path class ProgressHandlers: """Handles progress callbacks for all analysis types with batch processing.""" BATCH_SIZE = 50 # Insert items in batches of 50 to reduce GUI blocking UPDATE_FREQUENCY = 10 # Update progress UI every N files def __init__(self, gui_app): """ Args: gui_app: Reference to the main GUI application instance """ self.app = gui_app self._count_batch_buffer = [] self._metrics_batch_buffer = [] def on_count_progress(self, res): """Handle progress for countings analysis. Args: res: Dictionary with file counting results """ try: if not isinstance(res, dict): return file = res.get("file") name = Path(file).name if file else "" code = res.get("code_lines") or res.get("code") or 0 comment = res.get("comment_lines") or 0 blank = res.get("blank_lines") or 0 total = res.get("physical_lines") or 0 lang = res.get("language") or "" # Add to batch buffer self._count_batch_buffer.append( (name, file, int(code), int(comment), int(blank), int(total), lang) ) # Insert batch when buffer is full if len(self._count_batch_buffer) >= self.BATCH_SIZE: self._flush_count_batch() # Update progress less frequently self._update_progress_counter() except Exception: pass def on_count_done(self, results): """Handle completion of countings analysis.""" # Flush remaining batch buffer self._flush_count_batch() self.app._current_task_id = None try: self.app._set_phase("Idle") except Exception: pass self.app._enable_action_buttons() def on_metrics_progress(self, res): """Handle progress for metrics analysis. Args: res: Dictionary with file metrics results """ try: if not isinstance(res, dict): return file = res.get("file") name = Path(file).name if file else "" avg = res.get("avg_cc") or res.get("avg") or 0 maxcc = res.get("max_cc") or 0 func = res.get("func_count") or 0 mi = res.get("mi") or 0 lang = res.get("language") or "" # Add to batch buffer self._metrics_batch_buffer.append((name, file, avg, maxcc, func, mi, lang)) # Insert batch when buffer is full if len(self._metrics_batch_buffer) >= self.BATCH_SIZE: self._flush_metrics_batch() # Update progress self._update_progress_counter() except Exception: pass def on_metrics_done(self, results): """Handle completion of metrics analysis.""" # Flush remaining batch buffer self._flush_metrics_batch() self.app._current_task_id = None try: self.app._set_phase("Idle") except Exception: pass self.app._enable_action_buttons() def _flush_count_batch(self): """Flush countings batch buffer to results tree.""" if self._count_batch_buffer: for row_values in self._count_batch_buffer: self.app.results_tree.insert("", "end", values=row_values) self._count_batch_buffer.clear() def _flush_metrics_batch(self): """Flush metrics batch buffer to results tree.""" if self._metrics_batch_buffer: for row_values in self._metrics_batch_buffer: self.app.results_tree.insert("", "end", values=row_values) self._metrics_batch_buffer.clear() def _update_progress_counter(self): """Update progress counter and UI (with reduced frequency).""" try: self.app._processed_files = getattr(self.app, "_processed_files", 0) + 1 # Update UI only every N files or at completion if ( self.app._processed_files % self.UPDATE_FREQUENCY == 0 or self.app._processed_files == self.app._total_files ): self.app._lbl_files.config( text=f"Files: {self.app._processed_files}/{self.app._total_files}" ) self.app.progress["value"] = self.app._processed_files except Exception: pass