import psycopg2
from psycopg2 import OperationalError


# Neon database connection string

def create_connection(CONN_STRING):
    """Create a connection to the Neon PostgreSQL database."""
    try:
        connection = psycopg2.connect(CONN_STRING, sslmode="require")
        print("✅ Connection to Neon PostgreSQL DB successful")
        return connection
    except OperationalError as e:
        print(f"❌ The error '{e}' occurred")
        return None

def close_connection(connection):
    """Close the database connection."""
    if connection:
        connection.close()
        print("🔒 Connection closed")