a cOj@s`dZddlZddlmZddlZddlmZmZdddZeddd Z d d Z ed d Z dS)uf 数据库连接工具。 提供 PostgreSQL 和 SQLite 的连接函数，以及上下文管理器。 N)contextmanager)DB_PATH PG_CONFIGcCs|pt}tjfi|S)u4获取 PostgreSQL 连接。默认连接 41 号 PG。)rpsycopg2connect)configZcfgr %/Users/admin/quant_ocf/ocf/core/db.py get_pg_connsr ccsJt|}|}z"|V|W||n||0dS)u&PostgreSQL 游标上下文管理器。N)r cursorcommitclose)rconncurr r r pg_cursors  rcCs ttS)u获取本地 SQLite 连接。)sqlite3rrr r r r get_sqlite_conn"srccsHt}|}z"|V|W||n||0dS)u"SQLite 游标上下文管理器。N)rr r r)rrr r r sqlite_cursor's  r)N)N) __doc__rZ contextlibrrsettingsrrr rrrr r r r s