27 lines
651 B
Python
27 lines
651 B
Python
"""
|
||
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,
|
||
)
|