fetch-data.py

import sqlite3
import json


def get_posts_by_persona(persona_name):
    """Fetch all posts for a given persona."""
    conn = sqlite3.connect("personas.db")
    cursor = conn.cursor()

    # Fetch persona ID
    cursor.execute("SELECT persona_id FROM personas WHERE name = ?", (persona_name,))
    persona = cursor.fetchone()

    if not persona:
        print(f"Persona '{persona_name}' not found.")
        return []

    persona_id = persona[0]

    # Fetch posts for this persona
    cursor.execute("SELECT text_blocks FROM posts WHERE persona_id = ?", (persona_id,))
    posts = [row[0] for row in cursor.fetchall()]

    conn.close()
    return posts


# Example: Fetch posts for "Marketer"
selected_persona = "Marketer"
persona_posts = get_posts_by_persona(selected_persona)
print(f"Posts for {selected_persona}:")
print(json.dumps(persona_posts, indent=2, ensure_ascii=False))