o "4g@sdZddlmZddlZddlmZddlmZddlmZddlm Z dd lm Z dd l m Z m Z e jZe jZd Zd#d$ddZd#d%ddZd&ddZedZedZd'ddZd(d!d"ZdS))z* Functions to manipulate conninfo strings ) annotationsN)pq)errors)_conninfo_utils)_conninfo_attempts)_conninfo_attempts_async) ConnParamConnDictconninfostrkwargsr returncKsp|s|sdS|st|t|Sdd|D}|r&t|}|||}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|] \}}|dur||qSN.0kvrrY/var/www/html/authentication-server/venv/lib/python3.10/site-packages/psycopg/conninfo.py 6sz!make_conninfo.. css*|]\}}|dtt|VqdS)=N) _param_escaperrrrr =s(z make_conninfo..)_parse_conninforitemsconninfo_to_dictupdatejoin)r rtmprrr make_conninfos r#r cKs<t|}dd|D}|D] \}}|dur|||<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|]}|jdur|j|jqSr)valkeyworddecode)roptrrrrUs"z$conninfo_to_dict..N)rr)r roptsrvrrrrrrEsrlist[pq.ConninfoOption]c Cs>z tj|WStjy}ztt|dd}~ww)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 exrrrr^s rz([\\'])z\sscCs.|sdStd|}t|rd|d}|S)z9 Apply the escaping rule required by PQconnectdb z''z\\\1') re_escapesubre_spacesearch)r2rrrrps    rparamsintcCsjt|d}|dur t}ztt|}Wnty$td|dw|dkr-t}|S|dkr3d}|S)zG Return the timeout in seconds from the connection parameters. connect_timeoutNzbad value for connect_timeout: r)r get_param_DEFAULT_CONNECT_TIMEOUTr9float ValueErrorr.r0)r8valuetimeoutrrrtimeout_from_conninfo~s  rB)r )r rrr rr)r rrr rr )r rrr*)r2rrr)r8r rr9)__doc__ __future__rrer rrr.rrrabcr r conninfo_attemptsconninfo_attempts_asyncr=r#rrcompiler4r6rrBrrrrs&        )