ҪiUdZddlmZddlZddlmZddlmZmZddl Z ddl m Z ddl m Z ddlmZdd lmZeeZdad ed <ej,Zdd Zdd ZddZeddZedddZdddZdddZdddZy)u utils/database.py — PostgreSQL connection pool + helpers ======================================================== Centralized, thread-safe Postgres access using a psycopg2 ThreadedConnectionPool. All routes / services should go through `fetch_one`, `fetch_all`, or `execute` — never open their own connections. Lifecycle: init_pool() – called once at FastAPI startup (fail-fast). close_pool() – called once at FastAPI shutdown. get_conn() – context manager that checks out a connection from the pool and returns it on exit. Credentials are loaded from `.env` via `config.settings`. Nothing is hardcoded. ) annotationsN)contextmanager)AnyIterator)pool)RealDictCursor)settings) get_loggerz%pg_pool.ThreadedConnectionPool | None_poolc t5ttjd dddytj rtj sKBH& I:0I."I"#I.+I:I$)I KIK"I+ 'I..I7 3I:: J>1J94K9J>>KK ct5t, tjtj ddadddy#t $r }tj d|Yd}~/d}~wwxYw#dawxYw#1swYyxYw)z6Close all connections in the pool. Called at shutdown.NzPostgreSQL pool closedz!Error closing PostgreSQL pool: %s)rr closeallrr r0warning)r3s r6r1r1esm     12   F ?EE F s?A6)AA6 A, A'"A/'A,,A//A33A66A?ctduSN)r r6is_readyr?ss  r>c#RKt tdtj} ||js|j tj|y#t $r" |j #t $rYwxYwwxYw#tj|wxYww)z{ Check out a connection from the pool, return it on exit. Auto-commits on success and rolls back on exception. Nu7DB pool not initialized — call init_pool() at startup)r rgetconnclosedcommitr0rollbackputconn)r4s r6r,r,ys  }TUU ==?D  {{ KKM  d   MMO       dsL&B' A B' B )A:9B : BB BB  B B$$B'c#Kt5}|rtnd}|j|5}|ddddddy#1swYxYw#1swYyxYww)zo Convenience: get_conn() + cursor() in one go. Returns dict-like rows by default (RealDictCursor). N)cursor_factory)r,rr-) dict_rowsr4rGr5s r6 get_cursorrIsG t+4$ [[[ 73I8  7 7 s1 AA AA  AA A  AAct5}|j||xsd|j}|r t|ndcdddS#1swYyxYw)z7Run SELECT and return the first row as a dict, or None.r=N)rIr.r/dict)sqlparamsr5rows r6 fetch_onerOs;  C2&llntCyT s 5A  Act5}|j||xsd|j}|Dcgc] }t|c}cdddScc}w#1swYyxYw)z2Run SELECT and return all rows as a list of dicts.r=N)rIr.fetchallrK)rLrMr5rowsrs r6 fetch_allrTsP  C2&||~!%&AQ& ' s+AAAAA"ctd5}|j||xsd|jcdddS#1swYyxYw)z5Run INSERT / UPDATE / DELETE and return the rowcount.F)rHr=N)rIr.rowcount)rLrMr5s r6r.r.s1 e $ C2&|| % $ $s "9A)returnNone)rWbool)rWz(Iterator[psycopg2.extensions.connection])T)rHrYrWz$Iterator[psycopg2.extensions.cursor]r<)rLstrrMtuple | dict | NonerWz dict | None)rLrZrMr[rWz list[dict])rLrZrMr[rWint) __doc__ __future__r threading contextlibrtypingrrr*rr&psycopg2.extrasrconfigr utils.loggerr __name__rr __annotations__Lockrr7r1r?r,rIrOrTr.r=r>r6rhs"#% $*#04,3 Y^^  :z  .*'r>