135 lines
3.9 KiB
Python
135 lines
3.9 KiB
Python
"""
|
|
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.export_btn.config(state='normal')
|
|
self.app.cancel_btn.config(state='disabled')
|
|
self.app._current_task_id = None
|
|
try:
|
|
self.app._set_phase("Idle")
|
|
except Exception:
|
|
pass
|
|
|
|
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.export_btn.config(state='normal')
|
|
self.app.cancel_btn.config(state='disabled')
|
|
self.app._current_task_id = None
|
|
try:
|
|
self.app._set_phase("Idle")
|
|
except Exception:
|
|
pass
|
|
|
|
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
|