*JfdZddlmZddlZddlmZddlmZddlmZ ddlm Z dd lm Z dd lm Z e j Z e jZe jZd ZdddZdd dZd!dZejdZejdZd"dZd#dZdS)$z* Functions to manipulate conninfo strings ) annotationsN)Any)pq)errors)_conninfo_utils)_conninfo_attempts)_conninfo_attempts_asyncconninfostrkwargsrreturnc ^|s|sdS|st|t|Sd|D}|r&t|}|||}dd|D}t||S)a Merge a string and keyword params into a single conninfo string. :param conninfo: A `connection string`__ as accepted by PostgreSQL. :param kwargs: Parameters overriding the ones specified in `!conninfo`. :return: A connection string valid for PostgreSQL, with the `!kwargs` parameters merged. Raise `~psycopg.ProgrammingError` if the input doesn't make a valid conninfo string. .. __: https://www.postgresql.org/docs/current/libpq-connect.html #LIBPQ-CONNSTRING r ci|] \}}||| SN.0kvs S/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/psycopg/conninfo.py z!make_conninfo..8s A A Av11=a=== c3^K|](\}}|dtt|V)dS)=N) _param_escaperrs r z make_conninfo..?sBTT6Aq166}SVV4466TTTTTTr)_parse_conninforitemsconninfo_to_dictupdatejoin)r rtmps r make_conninfor's Fr !!!8}}B A A A AFx(( 6xxTTV\\^^TTTTTHH OrConnDictc ~t|}d|D}|D] \}}||||< |S)a Convert the `!conninfo` string into a dictionary of parameters. :param conninfo: A `connection string`__ as accepted by PostgreSQL. :param kwargs: Parameters overriding the ones specified in `!conninfo`. :return: Dictionary with the parameters parsed from `!conninfo` and `!kwargs`. Raise `~psycopg.ProgrammingError` if `!conninfo` is not a a valid connection string. .. __: https://www.postgresql.org/docs/current/libpq-connect.html #LIBPQ-CONNSTRING c~i|]:}|j |j|j;Sr)valkeyworddecode)ropts rrz$conninfo_to_dict..Ws; X X XSCGDW#+     0 0DWDWDWr)r!r")r roptsrvrrs rr#r#GsS 8 $ $D X XD X X XB 1 =BqE Irlist[pq.ConninfoOption]c tj|S#tj$r'}t jt|dd}~wwxYw)z Verify that `!conninfo` is a valid connection string. Raise ProgrammingError if the string is not valid. Return the result of pq.Conninfo.parse() on success. N)rConninfoparseencodeeOperationalErrorProgrammingErrorr)r exs rr!r!^sb4{  !2!2333 444 R))t34s03A)"A$$A)z([\\'])z\ssc|sdStd|}t|rd|zdz}|S)z9 Apply the escaping rule required by PQconnectdb z''z\\\1') re_escapesubre_spacesearch)r:s rrrpsJ t gq!!Aq !GcM Hrparamsintctj|d}|t} tt |}n&#t $rt jd|dwxYw|dkrt}n|dkrd}|S)zG Return the timeout in seconds from the connection parameters. connect_timeoutNzbad value for connect_timeout: r)r get_param_DEFAULT_CONNECT_TIMEOUTrBfloat ValueErrorr6r8)rAvaluetimeouts rtimeout_from_conninforL~s.7@QRRE }(XeEll## XXX !L5!L!LMMSWWX!||+ 1 Ns =#A )r )r rrrrr)r rrrrr()r rrr1)r:rrr)rAr(rrB)__doc__ __future__rretypingrr rrr6rr r r(conninfo_attemptsconninfo_attempts_asyncrGr'r#r!compiler=r?rrLrrrrTsY #"""""  &&&&&&  #&82J &&&&&R. 4 4 4 4 BJz " " 2:e       r