Complete data_enrichment.py - add missing methods and fix syntax

data_enrichment.py

@@ -338,4 +338,84 @@ class DataEnrichmentManager:
         if contact_data.get('domain'):
             website_contacts = self.scrape_website_contacts(contact_data['domain'])
             if website_contacts:
-                enriched['enrichments
+                enriched['enrichments']['website_contacts'] = website_contacts
+                enriched['sources'].append('website_contacts')
+        
+        # Local business data (if company name and location provided)
+        if contact_data.get('company') and contact_data.get('location'):
+            business_data = self.scrape_local_business(
+                query=contact_data['company'],
+                location=contact_data['location']
+            )
+            if business_data:
+                enriched['enrichments']['local_business'] = business_data
+                enriched['sources'].append('local_business')
+        
+        # Twitter enrichment
+        if contact_data.get('twitter_username'):
+            twitter_data = self.enrich_twitter_profile(contact_data['twitter_username'])
+            if twitter_data:
+                enriched['enrichments']['twitter'] = twitter_data
+                enriched['sources'].append('twitter')
+        
+        # Facebook enrichment
+        if contact_data.get('facebook_id'):
+            facebook_data = self.enrich_facebook_profile(contact_data['facebook_id'])
+            if facebook_data:
+                enriched['enrichments']['facebook'] = facebook_data
+                enriched['sources'].append('facebook')
+        
+        # Google search for additional info
+        if contact_data.get('name'):
+            search_query = f"{contact_data['name']}"
+            if contact_data.get('company'):
+                search_query += f" {contact_data['company']}"
+            google_results = self.search_google(search_query, num_results=5)
+            if google_results:
+                enriched['enrichments']['google_search'] = google_results
+                enriched['sources'].append('google')
+        
+        enriched['enriched_at'] = datetime.now().isoformat()
+        enriched['total_sources'] = len(enriched['sources'])
+        
+        return enriched
+    
+    def get_enrichment_stats(self) -> Dict:
+        """Get cache statistics and API usage"""
+        stats = {
+            'cache_directory': str(self.cache_dir),
+            'cached_profiles': 0,
+            'cache_types': {},
+            'total_api_calls': 0,
+            'cache_hits': 0,
+            'cache_misses': 0,
+            'total_enrichments': 0,
+            'successful_enrichments': 0,
+            'failed_enrichments': 0
+        }
+        
+        if self.cache_dir.exists():
+            cache_files = list(self.cache_dir.glob('*.json'))
+            stats['cached_profiles'] = len(cache_files)
+            
+            for cache_file in cache_files:
+                cache_type = cache_file.stem.split('_')[0]
+                stats['cache_types'][cache_type] = stats['cache_types'].get(cache_type, 0) + 1
+        
+        stats['configured_apis'] = [k for k, v in self.api_keys.items() if v]
+        stats['total_apis'] = len(stats['configured_apis'])
+        
+        return stats
+
+
+# Convenience functions
+def create_enrichment_manager(username: str) -> DataEnrichmentManager:
+    """Create a new enrichment manager instance"""
+    return DataEnrichmentManager(username)
+
+
+def enrich_contact(username: str, contact_data: Dict) -> Dict:
+    """Quick enrichment for a single contact"""
+    manager = DataEnrichmentManager(username)
+    return manager.enrich_contact_full(contact_data)
+