# database.py
import psycopg2
from psycopg2.extras import RealDictCursor

def connect_db():
    conn = psycopg2.connect(
        dbname="solana_bot_db",
        user="your_user",
        password="your_password",
        host="localhost",
        port="5432"
    )
    return conn

def insert_token_data(token_data):
    query = """
    INSERT INTO tokens (id, name, launch_time, liquidity, volume)
    VALUES (%s, %s, %s, %s, %s)
    ON CONFLICT (id) DO NOTHING;
    """
    conn = connect_db()
    with conn.cursor() as cur:
        for token in token_data:
            cur.execute(query, (token['id'], token['name'], token['launch_time'],
                                token['liquidity'], token['volume']))
    conn.commit()
    conn.close()