88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
import sqlite3
|
|
import json
|
|
import os
|
|
from datetime import datetime
|
|
|
|
# Robust path handling:
|
|
# 1. Get the directory where this script (db.py) lives
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
# 2. Define data directory and db path relative to it
|
|
DATA_DIR = os.path.join(BASE_DIR, 'data')
|
|
DB_PATH = os.path.join(DATA_DIR, 'leads.db')
|
|
|
|
def init_db():
|
|
# Ensure data directory exists
|
|
if not os.path.exists(DATA_DIR):
|
|
os.makedirs(DATA_DIR)
|
|
|
|
conn = sqlite3.connect(DB_PATH)
|
|
c = conn.cursor()
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS leads (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
source_id TEXT UNIQUE,
|
|
received_at TIMESTAMP,
|
|
company_name TEXT,
|
|
contact_name TEXT,
|
|
email TEXT,
|
|
phone TEXT,
|
|
raw_body TEXT,
|
|
enrichment_data TEXT,
|
|
status TEXT DEFAULT 'new',
|
|
response_draft TEXT,
|
|
sent_at TIMESTAMP
|
|
)
|
|
''')
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def insert_lead(lead_data):
|
|
# Ensure DB exists before inserting (if ingest runs before init)
|
|
if not os.path.exists(DB_PATH):
|
|
init_db()
|
|
|
|
conn = sqlite3.connect(DB_PATH)
|
|
c = conn.cursor()
|
|
try:
|
|
c.execute('''
|
|
INSERT INTO leads (source_id, received_at, company_name, contact_name, email, phone, raw_body, status)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
''', (
|
|
lead_data.get('id'),
|
|
datetime.now(),
|
|
lead_data.get('company'),
|
|
lead_data.get('contact'),
|
|
lead_data.get('email'),
|
|
lead_data.get('phone'),
|
|
lead_data.get('raw_body'),
|
|
'new'
|
|
))
|
|
conn.commit()
|
|
return True
|
|
except sqlite3.IntegrityError:
|
|
return False
|
|
finally:
|
|
conn.close()
|
|
|
|
def get_leads():
|
|
if not os.path.exists(DB_PATH):
|
|
init_db()
|
|
|
|
conn = sqlite3.connect(DB_PATH)
|
|
conn.row_factory = sqlite3.Row
|
|
c = conn.cursor()
|
|
c.execute('SELECT * FROM leads ORDER BY received_at DESC')
|
|
rows = c.fetchall()
|
|
conn.close()
|
|
return [dict(row) for row in rows]
|
|
|
|
def update_lead_status(lead_id, status, response_draft=None):
|
|
conn = sqlite3.connect(DB_PATH)
|
|
c = conn.cursor()
|
|
if response_draft:
|
|
c.execute('UPDATE leads SET status = ?, response_draft = ? WHERE id = ?', (status, response_draft, lead_id))
|
|
else:
|
|
c.execute('UPDATE leads SET status = ? WHERE id = ?', (status, lead_id))
|
|
conn.commit()
|
|
conn.close()
|