import os

import psycopg


DATABASE_URL = os.environ.get("KUNDENPLATTFORM_DATABASE_URL")


def get_database_connection():
    if not DATABASE_URL:
        raise RuntimeError("KUNDENPLATTFORM_DATABASE_URL ist nicht gesetzt.")

    return psycopg.connect(DATABASE_URL)


async def get_async_database_connection():
    if not DATABASE_URL:
        raise RuntimeError("KUNDENPLATTFORM_DATABASE_URL ist nicht gesetzt.")

    return await psycopg.AsyncConnection.connect(DATABASE_URL)