Update src/core/services/database_service.py

src/core/services/database_service.py

@@ -29,11 +29,60 @@ class DatabaseService:
             conn.close()
 
     def setup_database(self):
-        """Create database tables with relationships"""
         with self.get_db() as conn:
             c = conn.cursor()
             c.executescript('''
-                -- SQL to create tables
+                CREATE TABLE IF NOT EXISTS users (
+                    id TEXT PRIMARY KEY,
+                    email TEXT UNIQUE,
+                    name TEXT,
+                    role TEXT,
+                    department TEXT,
+                    title TEXT,
+                    region TEXT,
+                    quota REAL,
+                    created_at TEXT,
+                    last_login TEXT
+                );
+                
+                CREATE TABLE IF NOT EXISTS accounts (
+                    id TEXT PRIMARY KEY,
+                    name TEXT,
+                    parent_account_id TEXT,
+                    industry TEXT,
+                    status TEXT,
+                    website TEXT,
+                    annual_revenue REAL,
+                    employee_count INTEGER,
+                    technology_stack TEXT,
+                    region TEXT,
+                    address TEXT,
+                    account_owner_id TEXT,
+                    engagement_score REAL,
+                    created_at TEXT,
+                    last_activity_at TEXT,
+                    FOREIGN KEY (parent_account_id) REFERENCES accounts (id),
+                    FOREIGN KEY (account_owner_id) REFERENCES users (id)
+                );
+                
+                CREATE TABLE IF NOT EXISTS contacts (
+                    id TEXT PRIMARY KEY,
+                    account_id TEXT,
+                    first_name TEXT,
+                    last_name TEXT,
+                    email TEXT,
+                    phone TEXT,
+                    title TEXT,
+                    department TEXT,
+                    reports_to_id TEXT,
+                    influence_level TEXT,
+                    engagement_score REAL,
+                    preferences TEXT,
+                    created_at TEXT,
+                    last_contacted TEXT,
+                    FOREIGN KEY (account_id) REFERENCES accounts (id),
+                    FOREIGN KEY (reports_to_id) REFERENCES contacts (id)
+                );
             ''')
 
     def generate_synthetic_data(self):
@@ -216,7 +265,17 @@ class DatabaseService:
                 :title, :department, :reports_to_id, :influence_level,
                 :engagement_score, :preferences, :created_at, :last_contacted
             )''', contacts)
-
+    def get_user_by_email(self, email: str) -> Optional[Dict]:
+        with self.get_db() as conn:
+            cursor = conn.execute(
+                "SELECT * FROM users WHERE email = ?", 
+                (email,)
+            )
+            row = cursor.fetchone()
+            if row:
+                return dict(row)
+            return None
+            
     def get_user_accounts(self, user_id: str) -> List[Dict]:
         """Get accounts associated with user"""
         with self.get_db() as conn:
@@ -227,3 +286,15 @@ class DatabaseService:
                 ORDER BY a.name
             """, (user_id,))
             return [dict(row) for row in cursor.fetchall()]
+    def get_account_metrics(self, account_id: str) -> Dict:
+        with self.get_db() as conn:
+            contacts_count = conn.execute(
+                "SELECT COUNT(*) FROM contacts WHERE account_id = ?",
+                (account_id,)
+            ).fetchone()[0]
+            
+            return {
+                'contact_count': contacts_count,
+                'interaction_count': 0,  # Default for now
+                'avg_sentiment': 0  # Default for now
+            }