"""
zqyy_app 数据库连接

使用 psycopg2 直连 PostgreSQL，不引入 ORM。
连接参数从环境变量读取（经 config 模块加载）。
"""

import psycopg2
from psycopg2.extensions import connection as PgConnection

from app.config import APP_DB_NAME, DB_HOST, DB_PASSWORD, DB_PORT, DB_USER


def get_connection() -> PgConnection:
    """
    获取 zqyy_app 数据库连接。

    调用方负责关闭连接（推荐配合 contextmanager 或 try/finally 使用）。
    """
    return psycopg2.connect(
        host=DB_HOST,
        port=DB_PORT,
        user=DB_USER,
        password=DB_PASSWORD,
        dbname=APP_DB_NAME,
    )