U ڲg@sdZddlmZddlZddlmZddlmZddlmZddlm Z dd lm Z dd l m Z m Z e jZe jZd Zd"d dd dddZd#d dddddZd ddddZedZedZd d dddZdddd d!ZdS)$z* Functions to manipulate conninfo strings ) annotationsN)pq)errors)_conninfo_utils)_conninfo_attempts)_conninfo_attempts_async) ConnParamConnDictstrr )conninfokwargsreturncKsp|s |s dS|s t|t|Sdd|D}|rLt|}|||}ddd|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 cSsi|]\}}|dk r||qSN.0kvrr4/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/conninfo.py 6sz!make_conninfo.. css(|] \}}|dtt|VqdS)=N) _param_escaper rrrr =sz make_conninfo..)_parse_conninfor itemsconninfo_to_dictupdatejoin)rrtmprrr make_conninfos r#r cKs<t|}dd|D}|D]\}}|dk r|||<q|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 cSs(i|] }|jdk r|j|jqSr)valkeyworddecode)roptrrrrUs z$conninfo_to_dict..N)rr)rroptsrvrrrrrrEs rzlist[pq.ConninfoOption])rrc CsLztj|WStjk rF}ztt|dW5d}~XYnXdS)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)rZConninfoparseencodeeZOperationalErrorProgrammingErrorr )rexrrrr^srz([\\'])z\s)srcCs.|sdStd|}t|r*d|d}|S)z9 Apply the escaping rule required by PQconnectdb z''z\\\1') re_escapesubre_spacesearch)r/rrrrps    rint)paramsrcCsnt|d}|dkrt}ztt|}Wn&tk rNtd|dYnX|dkr^t}n |dkrjd}|S)zG Return the timeout in seconds from the connection parameters. connect_timeoutNzbad value for connect_timeout: r)r get_param_DEFAULT_CONNECT_TIMEOUTr5float ValueErrorr,r-)r6valuetimeoutrrrtimeout_from_conninfo~s r?)r )r )__doc__ __future__rrer rrr,rrrabcr r Zconninfo_attemptsZconninfo_attempts_asyncr:r#rrcompiler1r3rr?rrrrs$      )