]jVUddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d lmZd d lmZdd lZdd lmZd aded<GddeeZd S)) annotations)Callable)Any)connect) Connection)Cursor)dict_row) get_settings)BasePostgresHelpersMixin)BasePostgresPilotMixinN)ConnectionPoolzConnectionPool | None _GLOBAL_POOLcTeZdZdZdddZdd Zdd Zdd ZddZddZ ddZ ddZ dS)BasePostgresClientzJMinimal PostgreSQL client using psycopg_pool and env-driven configuration.N database_url str | NonereturnNonecVt}|p|j|_|j|_dSN)r r _database_url database_connect_timeout_seconds_connect_timeout)selfrsettingss >)BX-B ( Iboolc*t|jS)z.Return True when a database URL is configured.)r rrs r is_enabledzBasePostgresClient.is_enabled!sD&'''rc4|sdS|5}|5}|d|dddn #1swxYwYdddn #1swxYwYdS)zGCheck if database connection is alive by executing a lightweight query.FzSELECT 1NT)r#get_connectioncursorexecutefetchone)r connectionr&s rpingzBasePostgresClient.ping%s    5  " " "j""$$ "z***!!! " " " " " " " " " " " " " " " " " " " " " " " " " " " " " "ts5B *A6* B 6A: :B =A: >B  BBrc|jstdtDt|jdd|jt ddat jtjtS)NzDATABASE_URL is not configured.connect_timeout row_factoryT)conninfomin_sizemax_sizekwargsopen) r ValueErrorrrrr atexitregistercloser"s r _get_poolzBasePostgresClient._get_pool/sv! @>?? ?  )+'+'<#+   L OL. / / /rrcN|S)z6Get a managed connection context from the global pool.)r:r)r"s rr%z!BasePostgresClient.get_connectionBs~~**,,,rcDt|j|jtS)z7Backward compatibility for direct raw connection usage.r.)rrrr r"s r_open_connectionz#BasePostgresClient._open_connectionFs*   1     r operation Callable[[Connection[Any]], Any]c|5} ||}||cdddS#t$r|wxYw#1swxYwYdS)zOExecute one callback inside a single transaction with explicit commit/rollback.N)r%commit Exceptionrollback)rr>r)results rrun_in_transactionz%BasePostgresClient.run_in_transactionOs  " " j ":..!!###             ##%%%            s"A' A A$$A''A+.A+r)Connection[Any] | NoneCallable[[Cursor[Any]], Any]cf|7|5}||cdddS#1swxYwY|5}|5}||}dddn #1swxYwY|dddn #1swxYwY|S)zQExecute cursor-bound work either inside an existing transaction or a private one.N)r&r%rA)rr)r>r&owned_connectionrDs r!_execute_with_optional_connectionz4BasePostgresClient._execute_with_optional_connectionZss  !""$$ ) y(( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) " " &&6!((** +f"6** + + + + + + + + + + + + + + +  # # % % % & & & & & & & & & & & & & & & sB /33B&# A;/ B&;A? ?B&A? B&&B*-B*r)rrrr)rr )rr)rr)r>r?rr)r)rFr>rGrr) __name__ __module__ __qualname____doc__rr#r*r:r%r=rErJrrrrsUTJJJJJ ((((&----        rr) __future__rcollections.abcrtypingrpsycopgrpsycopg.connectionrpsycopg.cursorr psycopg.rowsr app.core.settingsr helpersr pilot_queriesr r7 psycopg_poolrr__annotations__rrOrrr\s,"""""""$$$$$$))))))!!!!!!!!!!!!******------111111 ''''''&* ****RRRRRRRRRRr