pomera-ai-commander 0.1.0 → 1.2.1

package/tools/email_extraction_tool.py

@@ -1,249 +1,249 @@
-"""
-Email Extraction Tool Module - Advanced email extraction utility
-
-This module provides comprehensive email extraction functionality with UI components
-for the Promera AI Commander application.
-"""
-
-import tkinter as tk
-from tkinter import ttk
-import re
-from collections import Counter
-
-
-class EmailExtractionProcessor:
-    """Email extraction processor with advanced filtering and formatting options."""
-    
-    @staticmethod
-    def extract_emails_advanced(text, omit_duplicates, hide_counts, sort_emails, only_domain):
-        """Advanced email extraction with options for deduplication, counting, sorting, and domain-only extraction."""
-        # Extract all email addresses using improved regex pattern
-        pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
-        emails = re.findall(pattern, text)
-        
-        if not emails:
-            return "No email addresses found in the text."
-        
-        # Extract domains if only_domain is True
-        if only_domain:
-            emails = [email.split('@')[1] for email in emails]
-        
-        # Count occurrences
-        email_counts = Counter(emails)
-        
-        # Get unique emails if omit_duplicates is True
-        if omit_duplicates:
-            unique_emails = list(email_counts.keys())
-            if sort_emails:
-                unique_emails.sort()
-            
-            # Format output
-            if hide_counts:
-                return '\n'.join(unique_emails)
-            else:
-                # When omit_duplicates=True, show count as (1) for all
-                return '\n'.join([f"{email} (1)" for email in unique_emails])
-        else:
-            # Keep all emails including duplicates
-            if sort_emails:
-                emails.sort()
-            
-            if hide_counts:
-                return '\n'.join(emails)
-            else:
-                # Show actual counts for each unique email
-                result = []
-                processed = set()
-                for email in emails:
-                    if email not in processed:
-                        result.append(f"{email} ({email_counts[email]})")
-                        processed.add(email)
-                
-                if sort_emails:
-                    result.sort()
-                
-                return '\n'.join(result)
-
-    @staticmethod
-    def process_text(input_text, settings):
-        """Process text using the current settings."""
-        return EmailExtractionProcessor.extract_emails_advanced(
-            input_text,
-            settings.get("omit_duplicates", False),
-            settings.get("hide_counts", True),
-            settings.get("sort_emails", False),
-            settings.get("only_domain", False)
-        )
-
-
-class EmailExtractionUI:
-    """UI components for the Email Extraction Tool."""
-    
-    def __init__(self, parent, settings, on_setting_change_callback=None, apply_tool_callback=None):
-        """
-        Initialize the Email Extraction Tool UI.
-        
-        Args:
-            parent: Parent widget
-            settings: Dictionary containing tool settings
-            on_setting_change_callback: Callback function for setting changes
-            apply_tool_callback: Callback function for applying the tool
-        """
-        self.parent = parent
-        self.settings = settings
-        self.on_setting_change_callback = on_setting_change_callback
-        self.apply_tool_callback = apply_tool_callback
-        
-        # Initialize UI variables
-        self.email_omit_duplicates_var = tk.BooleanVar(value=settings.get("omit_duplicates", False))
-        self.email_hide_counts_var = tk.BooleanVar(value=settings.get("hide_counts", True))
-        self.email_sort_emails_var = tk.BooleanVar(value=settings.get("sort_emails", False))
-        self.email_only_domain_var = tk.BooleanVar(value=settings.get("only_domain", False))
-        
-        self.create_widgets()
-
-    def create_widgets(self):
-        """Creates the UI widgets for the Email Extraction Tool."""
-        # Checkboxes for various options
-        ttk.Checkbutton(
-            self.parent, 
-            text="Omit duplicates", 
-            variable=self.email_omit_duplicates_var, 
-            command=self._on_setting_change
-        ).pack(side=tk.LEFT, padx=5)
-        
-        ttk.Checkbutton(
-            self.parent, 
-            text="Hide counts", 
-            variable=self.email_hide_counts_var, 
-            command=self._on_setting_change
-        ).pack(side=tk.LEFT, padx=5)
-        
-        ttk.Checkbutton(
-            self.parent, 
-            text="Sort emails", 
-            variable=self.email_sort_emails_var, 
-            command=self._on_setting_change
-        ).pack(side=tk.LEFT, padx=5)
-        
-        ttk.Checkbutton(
-            self.parent, 
-            text="Only domain", 
-            variable=self.email_only_domain_var, 
-            command=self._on_setting_change
-        ).pack(side=tk.LEFT, padx=5)
-        
-        # Extract button
-        if self.apply_tool_callback:
-            ttk.Button(
-                self.parent, 
-                text="Extract", 
-                command=self.apply_tool_callback
-            ).pack(side=tk.LEFT, padx=10)
-
-    def _on_setting_change(self):
-        """Handle setting changes."""
-        if self.on_setting_change_callback:
-            self.on_setting_change_callback()
-
-    def get_current_settings(self):
-        """Get the current settings from the UI."""
-        return {
-            "omit_duplicates": self.email_omit_duplicates_var.get(),
-            "hide_counts": self.email_hide_counts_var.get(),
-            "sort_emails": self.email_sort_emails_var.get(),
-            "only_domain": self.email_only_domain_var.get()
-        }
-
-    def update_settings(self, settings):
-        """Update the UI with new settings."""
-        self.email_omit_duplicates_var.set(settings.get("omit_duplicates", False))
-        self.email_hide_counts_var.set(settings.get("hide_counts", True))
-        self.email_sort_emails_var.set(settings.get("sort_emails", False))
-        self.email_only_domain_var.set(settings.get("only_domain", False))
-
-
-class EmailExtractionTool:
-    """Main Email Extraction Tool class that combines processor and UI functionality."""
-    
-    def __init__(self):
-        self.processor = EmailExtractionProcessor()
-        self.ui = None
-        
-    def create_ui(self, parent, settings, on_setting_change_callback=None, apply_tool_callback=None):
-        """Create and return the UI component."""
-        self.ui = EmailExtractionUI(parent, settings, on_setting_change_callback, apply_tool_callback)
-        return self.ui
-        
-    def process_text(self, input_text, settings):
-        """Process text using the current settings."""
-        return self.processor.process_text(input_text, settings)
-        
-    def get_default_settings(self):
-        """Get default settings for the Email Extraction Tool."""
-        return {
-            "omit_duplicates": False,
-            "hide_counts": True,
-            "sort_emails": False,
-            "only_domain": False
-        }
-
-
-# Convenience functions for backward compatibility
-def extract_emails_advanced(text, omit_duplicates, hide_counts, sort_emails, only_domain):
-    """Extract emails with advanced options."""
-    return EmailExtractionProcessor.extract_emails_advanced(
-        text, omit_duplicates, hide_counts, sort_emails, only_domain
-    )
-
-
-def process_email_extraction(input_text, settings):
-    """Process email extraction with the specified settings."""
-    return EmailExtractionProcessor.process_text(input_text, settings)
-
-
-# BaseTool-compatible wrapper
-try:
-    from tools.base_tool import BaseTool
-    from typing import Dict, Any
-    import tkinter as tk
-    from tkinter import ttk
-    
-    class EmailExtractionToolV2(BaseTool):
-        """
-        BaseTool-compatible version of EmailExtractionTool.
-        """
-        
-        TOOL_NAME = "Email Extraction Tool"
-        TOOL_DESCRIPTION = "Extract email addresses from text"
-        TOOL_VERSION = "2.0.0"
-        
-        def process_text(self, input_text: str, settings: Dict[str, Any]) -> str:
-            """Extract emails from text."""
-            return EmailExtractionProcessor.extract_emails_advanced(
-                input_text,
-                settings.get("omit_duplicates", False),
-                settings.get("hide_counts", True),
-                settings.get("sort_emails", False),
-                settings.get("only_domain", False)
-            )
-        
-        def get_default_settings(self) -> Dict[str, Any]:
-            return {
-                "omit_duplicates": False,
-                "hide_counts": True,
-                "sort_emails": False,
-                "only_domain": False
-            }
-        
-        def create_ui(self, parent: tk.Widget, settings: Dict[str, Any], 
-                     on_change=None, on_apply=None) -> tk.Widget:
-            """Create UI for Email Extraction Tool."""
-            frame = ttk.Frame(parent)
-            ttk.Label(frame, text="Extract email addresses").pack(side=tk.LEFT, padx=5)
-            if on_apply:
-                ttk.Button(frame, text="Extract", command=on_apply).pack(side=tk.LEFT, padx=5)
-            return frame
-
-except ImportError:
+"""
+Email Extraction Tool Module - Advanced email extraction utility
+
+This module provides comprehensive email extraction functionality with UI components
+for the Promera AI Commander application.
+"""
+
+import tkinter as tk
+from tkinter import ttk
+import re
+from collections import Counter
+
+
+class EmailExtractionProcessor:
+    """Email extraction processor with advanced filtering and formatting options."""
+    
+    @staticmethod
+    def extract_emails_advanced(text, omit_duplicates, hide_counts, sort_emails, only_domain):
+        """Advanced email extraction with options for deduplication, counting, sorting, and domain-only extraction."""
+        # Extract all email addresses using improved regex pattern
+        pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
+        emails = re.findall(pattern, text)
+        
+        if not emails:
+            return "No email addresses found in the text."
+        
+        # Extract domains if only_domain is True
+        if only_domain:
+            emails = [email.split('@')[1] for email in emails]
+        
+        # Count occurrences
+        email_counts = Counter(emails)
+        
+        # Get unique emails if omit_duplicates is True
+        if omit_duplicates:
+            unique_emails = list(email_counts.keys())
+            if sort_emails:
+                unique_emails.sort()
+            
+            # Format output
+            if hide_counts:
+                return '\n'.join(unique_emails)
+            else:
+                # When omit_duplicates=True, show count as (1) for all
+                return '\n'.join([f"{email} (1)" for email in unique_emails])
+        else:
+            # Keep all emails including duplicates
+            if sort_emails:
+                emails.sort()
+            
+            if hide_counts:
+                return '\n'.join(emails)
+            else:
+                # Show actual counts for each unique email
+                result = []
+                processed = set()
+                for email in emails:
+                    if email not in processed:
+                        result.append(f"{email} ({email_counts[email]})")
+                        processed.add(email)
+                
+                if sort_emails:
+                    result.sort()
+                
+                return '\n'.join(result)
+
+    @staticmethod
+    def process_text(input_text, settings):
+        """Process text using the current settings."""
+        return EmailExtractionProcessor.extract_emails_advanced(
+            input_text,
+            settings.get("omit_duplicates", False),
+            settings.get("hide_counts", True),
+            settings.get("sort_emails", False),
+            settings.get("only_domain", False)
+        )
+
+
+class EmailExtractionUI:
+    """UI components for the Email Extraction Tool."""
+    
+    def __init__(self, parent, settings, on_setting_change_callback=None, apply_tool_callback=None):
+        """
+        Initialize the Email Extraction Tool UI.
+        
+        Args:
+            parent: Parent widget
+            settings: Dictionary containing tool settings
+            on_setting_change_callback: Callback function for setting changes
+            apply_tool_callback: Callback function for applying the tool
+        """
+        self.parent = parent
+        self.settings = settings
+        self.on_setting_change_callback = on_setting_change_callback
+        self.apply_tool_callback = apply_tool_callback
+        
+        # Initialize UI variables
+        self.email_omit_duplicates_var = tk.BooleanVar(value=settings.get("omit_duplicates", False))
+        self.email_hide_counts_var = tk.BooleanVar(value=settings.get("hide_counts", True))
+        self.email_sort_emails_var = tk.BooleanVar(value=settings.get("sort_emails", False))
+        self.email_only_domain_var = tk.BooleanVar(value=settings.get("only_domain", False))
+        
+        self.create_widgets()
+
+    def create_widgets(self):
+        """Creates the UI widgets for the Email Extraction Tool."""
+        # Checkboxes for various options
+        ttk.Checkbutton(
+            self.parent, 
+            text="Omit duplicates", 
+            variable=self.email_omit_duplicates_var, 
+            command=self._on_setting_change
+        ).pack(side=tk.LEFT, padx=5)
+        
+        ttk.Checkbutton(
+            self.parent, 
+            text="Hide counts", 
+            variable=self.email_hide_counts_var, 
+            command=self._on_setting_change
+        ).pack(side=tk.LEFT, padx=5)
+        
+        ttk.Checkbutton(
+            self.parent, 
+            text="Sort emails", 
+            variable=self.email_sort_emails_var, 
+            command=self._on_setting_change
+        ).pack(side=tk.LEFT, padx=5)
+        
+        ttk.Checkbutton(
+            self.parent, 
+            text="Only domain", 
+            variable=self.email_only_domain_var, 
+            command=self._on_setting_change
+        ).pack(side=tk.LEFT, padx=5)
+        
+        # Extract button
+        if self.apply_tool_callback:
+            ttk.Button(
+                self.parent, 
+                text="Extract", 
+                command=self.apply_tool_callback
+            ).pack(side=tk.LEFT, padx=10)
+
+    def _on_setting_change(self):
+        """Handle setting changes."""
+        if self.on_setting_change_callback:
+            self.on_setting_change_callback()
+
+    def get_current_settings(self):
+        """Get the current settings from the UI."""
+        return {
+            "omit_duplicates": self.email_omit_duplicates_var.get(),
+            "hide_counts": self.email_hide_counts_var.get(),
+            "sort_emails": self.email_sort_emails_var.get(),
+            "only_domain": self.email_only_domain_var.get()
+        }
+
+    def update_settings(self, settings):
+        """Update the UI with new settings."""
+        self.email_omit_duplicates_var.set(settings.get("omit_duplicates", False))
+        self.email_hide_counts_var.set(settings.get("hide_counts", True))
+        self.email_sort_emails_var.set(settings.get("sort_emails", False))
+        self.email_only_domain_var.set(settings.get("only_domain", False))
+
+
+class EmailExtractionTool:
+    """Main Email Extraction Tool class that combines processor and UI functionality."""
+    
+    def __init__(self):
+        self.processor = EmailExtractionProcessor()
+        self.ui = None
+        
+    def create_ui(self, parent, settings, on_setting_change_callback=None, apply_tool_callback=None):
+        """Create and return the UI component."""
+        self.ui = EmailExtractionUI(parent, settings, on_setting_change_callback, apply_tool_callback)
+        return self.ui
+        
+    def process_text(self, input_text, settings):
+        """Process text using the current settings."""
+        return self.processor.process_text(input_text, settings)
+        
+    def get_default_settings(self):
+        """Get default settings for the Email Extraction Tool."""
+        return {
+            "omit_duplicates": False,
+            "hide_counts": True,
+            "sort_emails": False,
+            "only_domain": False
+        }
+
+
+# Convenience functions for backward compatibility
+def extract_emails_advanced(text, omit_duplicates, hide_counts, sort_emails, only_domain):
+    """Extract emails with advanced options."""
+    return EmailExtractionProcessor.extract_emails_advanced(
+        text, omit_duplicates, hide_counts, sort_emails, only_domain
+    )
+
+
+def process_email_extraction(input_text, settings):
+    """Process email extraction with the specified settings."""
+    return EmailExtractionProcessor.process_text(input_text, settings)
+
+
+# BaseTool-compatible wrapper
+try:
+    from tools.base_tool import BaseTool
+    from typing import Dict, Any
+    import tkinter as tk
+    from tkinter import ttk
+    
+    class EmailExtractionToolV2(BaseTool):
+        """
+        BaseTool-compatible version of EmailExtractionTool.
+        """
+        
+        TOOL_NAME = "Email Extraction Tool"
+        TOOL_DESCRIPTION = "Extract email addresses from text"
+        TOOL_VERSION = "2.0.0"
+        
+        def process_text(self, input_text: str, settings: Dict[str, Any]) -> str:
+            """Extract emails from text."""
+            return EmailExtractionProcessor.extract_emails_advanced(
+                input_text,
+                settings.get("omit_duplicates", False),
+                settings.get("hide_counts", True),
+                settings.get("sort_emails", False),
+                settings.get("only_domain", False)
+            )
+        
+        def get_default_settings(self) -> Dict[str, Any]:
+            return {
+                "omit_duplicates": False,
+                "hide_counts": True,
+                "sort_emails": False,
+                "only_domain": False
+            }
+        
+        def create_ui(self, parent: tk.Widget, settings: Dict[str, Any], 
+                     on_change=None, on_apply=None) -> tk.Widget:
+            """Create UI for Email Extraction Tool."""
+            frame = ttk.Frame(parent)
+            ttk.Label(frame, text="Extract email addresses").pack(side=tk.LEFT, padx=5)
+            if on_apply:
+                ttk.Button(frame, text="Extract", command=on_apply).pack(side=tk.LEFT, padx=5)
+            return frame
+
+except ImportError:
     pass